Unit test cases improvements

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

diff --git a/pkg/control/ut_ctrl_submgr_test.go b/pkg/control/ut_ctrl_submgr_test.go
index 31eed31..4e2c471 100644 (file)
--- a/pkg/control/ut_ctrl_submgr_test.go
+++ b/pkg/control/ut_ctrl_submgr_test.go
@@ -298,7 +298,7 @@ func (mc *testingSubmgrControl) VerifyAllClean(t *testing.T) {
                        }
                }
                <-time.After(time.Millisecond * 100)
-               mc.TestLog(t, "VerifyAllClean delay plus 100ms")
+               xapp.Logger.Debug("VerifyAllClean delay plus 100ms")
        }
 
        assert.Equal(t, 0, len(mainCtrl.c.registry.register))
@@ -313,6 +313,15 @@ func (mc *testingSubmgrControl) VerifyAllClean(t *testing.T) {
        verifyE2KeyCount(t, 0)
 }
 
+func (mc *testingSubmgrControl) WaitOngoingRequestMapEmpty() {
+       for i := 0; i < 100; i++ {
+               if len(mainCtrl.c.restDuplicateCtrl.ongoingRequestMap) != 0 {
+                       <-time.After(time.Millisecond * 100)
+                       xapp.Logger.Debug("WaitOngoingRequestMapEmpty delay plus 100ms")
+               }
+       }
+}
+
 func (mc *testingSubmgrControl) GetMetrics(t *testing.T) (string, error) {
        req, err := http.NewRequest("GET", "http://localhost:8080/ric/v1/metrics", nil)
        if err != nil {