X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Fut_messaging_test.go;fp=pkg%2Fcontrol%2Fut_messaging_test.go;h=2a3ad96216cf9436c55adb3ec1c589d673f704aa;hb=fc20a5c2486129b14808898a01f3a18437749d25;hp=10e10b71c19b8f57ac880734185d5ab4605b5795;hpb=e5fdcd295671f86f1cbda6e5571c6f2075796af4;p=ric-plt%2Fsubmgr.git

diff --git a/pkg/control/ut_messaging_test.go b/pkg/control/ut_messaging_test.go
index 10e10b7..2a3ad96 100644
--- a/pkg/control/ut_messaging_test.go
+++ b/pkg/control/ut_messaging_test.go
@@ -5259,12 +5259,9 @@ func TestRESTSubReqAndSubDelOkSameActionWithRestartsInMiddle(t *testing.T) {
 	<-time.After(time.Second * 1)
 
 	// Delete subscription 1, and wait until it has removed the first endpoint
-	subepcnt := mainCtrl.get_subs_entrypoint_cnt(t, e2SubsId1)
 	xappConn1.SendRESTSubsDelReq(t, &restSubId1)
-	mainCtrl.wait_subs_entrypoint_cnt_change(t, e2SubsId1, subepcnt, 10)
-
+	mainCtrl.WaitRESTSubscriptionDelete(restSubId1)
 	// Above wait does not work correctly anymore as this delay makes this test case work
-	//<-time.After(time.Second * 1)
 
 	mainCtrl.SimulateRestart(t)
 	xapp.Logger.Debug("mainCtrl.SimulateRestart done 2")