X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fservices%2Frmrsender%2Frmr_sender_test.go;h=0c6dc28304edda224522b7dc7ca9cdee9ea9199d;hb=78f7c411678d2756ad076a71e6c093cd38cfca99;hp=c4d0c38bd0f01860ba8ce66e14369e2156aad317;hpb=bcb124908ffd1de0c00868838bbac733b881fcb2;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/services/rmrsender/rmr_sender_test.go b/E2Manager/services/rmrsender/rmr_sender_test.go
index c4d0c38..0c6dc28 100644
--- a/E2Manager/services/rmrsender/rmr_sender_test.go
+++ b/E2Manager/services/rmrsender/rmr_sender_test.go
@@ -52,13 +52,13 @@ func TestRmrSenderSendSuccess(t *testing.T) {
 	payload := []byte("some payload")
 	var xAction []byte
 	mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction)
-	rmrMessengerMock.On("SendMsg", mbuf).Return(&rmrCgo.MBuf{}, nil)
+	rmrMessengerMock.On("SendMsg", mbuf, true).Return(&rmrCgo.MBuf{}, nil)
 	rmrMsg := models.NewRmrMessage(123, ranName, payload, xAction)
 	rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
 	rmrSender := NewRmrSender(logger, rmrMessenger)
 	err := rmrSender.Send(rmrMsg)
 	assert.Nil(t, err)
-	rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf)
+	rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf, true)
 
 }
 
@@ -69,12 +69,12 @@ func TestRmrSenderSendFailure(t *testing.T) {
 	payload := []byte("some payload")
 	var xAction []byte
 	mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction)
-	rmrMessengerMock.On("SendMsg", mbuf).Return(mbuf, fmt.Errorf("rmr send failure"))
+	rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, fmt.Errorf("rmr send failure"))
 	rmrMsg := models.NewRmrMessage(123, ranName, payload, xAction)
 	rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
 	rmrSender := NewRmrSender(logger, rmrMessenger)
 	err := rmrSender.Send(rmrMsg)
-	rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf)
+	rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf, true)
 	assert.NotNil(t, err)
 }