Verify all clean function added for unit tests

[ric-plt/submgr.git] / pkg / control / sdl_e2SubsDb_test.go

diff --git a/pkg/control/sdl_e2SubsDb_test.go b/pkg/control/sdl_e2SubsDb_test.go
index 3175cc9..df0d060 100644 (file)
--- a/pkg/control/sdl_e2SubsDb_test.go
+++ b/pkg/control/sdl_e2SubsDb_test.go
@@ -198,6 +198,17 @@ func TestWriteSubscriptionToSdl(t *testing.T) {
        if err != nil {
                t.Errorf("TEST: %s", err.Error())
        }
+       verifyE2KeyCount(t, 1)
+}
+
+func verifyE2KeyCount(t *testing.T, expectedCount int) {
+
+       count, err := mainCtrl.c.GetE2KeyCount()
+       if err != nil {
+               t.Errorf("TEST: %s", err.Error())
+       } else {
+               assert.Equal(t, expectedCount, count)
+       }
 }
 
 func TestReadSubscriptionFromSdl(t *testing.T) {