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 ,\n Recevied <%+v>", err) + } + if !reflect.DeepEqual(expected, dst) { + t.Errorf("\nExpecting <%+v>,\n Recevied <%+v>", expected, dst) + } +}