From 64e65d70d238928daf8080bb82636cd193b41766 Mon Sep 17 00:00:00 2001
From: andronache <andrei.andronache@itsyscom.com>
Date: Mon, 11 Jan 2021 16:29:46 +0200
Subject: [PATCH] Added tests to statmetrics

---
 engine/statmetrics_test.go | 40 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/engine/statmetrics_test.go b/engine/statmetrics_test.go
index a318f31fe..d522706af 100644
--- a/engine/statmetrics_test.go
+++ b/engine/statmetrics_test.go
@@ -3222,3 +3222,43 @@ func TestStatMetricsStatDistinctRemEvent(t *testing.T) {
 		t.Errorf("\nExpecting <%+v>,\n Recevied <%+v>", expected, dst)
 	}
 }
+
+func TestStatMetricsStatDistinctRemEvent2(t *testing.T) {
+	dst := StatDistinct{
+		FilterIDs: []string{"Test_Filter_ID"},
+		FieldValues: map[string]utils.StringSet{
+			"FieldValue1": {},
+		},
+		Events: map[string]map[string]int64{
+			"Event1": {
+				"FieldValue1": 2,
+			},
+			"Event2": {},
+		},
+		MinItems:  3,
+		FieldName: "Test_Field_Name",
+		Count:     3,
+	}
+	expected := StatDistinct{
+		FilterIDs: []string{"Test_Filter_ID"},
+		FieldValues: map[string]utils.StringSet{
+			"FieldValue1": {},
+		},
+		Events: map[string]map[string]int64{
+			"Event1": {
+				"FieldValue1": 1,
+			},
+			"Event2": {},
+		},
+		MinItems:  3,
+		FieldName: "Test_Field_Name",
+		Count:     2,
+	}
+	err := dst.RemEvent("Event1")
+	if err != nil {
+		t.Errorf("\nExpecting <nil>,\n Recevied <%+v>", err)
+	}
+	if !reflect.DeepEqual(expected, dst) {
+		t.Errorf("\nExpecting <%+v>,\n Recevied <%+v>", expected, dst)
+	}
+}