From 09e9939482de7a652fbdb189579c14a4f3547120 Mon Sep 17 00:00:00 2001
From: armirveliaj <armir.veliaj@itsyscom.com>
Date: Thu, 30 May 2024 10:59:06 -0400
Subject: [PATCH] Add unit tests on ees

---
 ees/virtualee_test.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/ees/virtualee_test.go b/ees/virtualee_test.go
index 71d90ee1c..0c8507e4d 100644
--- a/ees/virtualee_test.go
+++ b/ees/virtualee_test.go
@@ -50,3 +50,32 @@ func TestVirtualEeExportEvent(t *testing.T) {
 	}
 	vEe.Close()
 }
+
+func TestVirtualEeConnect(t *testing.T) {
+	vEe := &VirtualEE{}
+	err := vEe.Connect()
+	if err != nil {
+		t.Errorf("Connect() returned an error: %v, expected nil", err)
+	}
+}
+
+func TestVirtualEE_PrepareMap(t *testing.T) {
+	vEe := &VirtualEE{}
+	event := "test event"
+	cgrEv := &utils.CGREvent{
+		Tenant: "event",
+	}
+
+	// Test case when PrepareMap is called
+	result, err := vEe.PrepareMap(cgrEv)
+
+	// Check that the returned error is nil
+	if err != nil {
+		t.Errorf("PrepareMap() returned an error: %v, expected nil", err)
+	}
+
+	// Check that the returned result is the expected event
+	if result != event {
+		t.Errorf("PrepareMap() returned %v, expected %v", result, event)
+	}
+}