Copy latest code

[ric-plt/e2mgr.git] / E2Manager / services / rmrsender / rmr_sender_test.go

diff --git a/E2Manager/services/rmrsender/rmr_sender_test.go b/E2Manager/services/rmrsender/rmr_sender_test.go
index df53959..bc45307 100644 (file)
--- a/E2Manager/services/rmrsender/rmr_sender_test.go
+++ b/E2Manager/services/rmrsender/rmr_sender_test.go
@@ -53,15 +53,15 @@ func TestRmrSenderSendSuccess(t *testing.T) {
 
        ranName := "test"
        payload := []byte("some payload")
-       xaction := []byte(ranName)
-       mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xaction)
-       rmrMessengerMock.On("SendMsg", mbuf).Return(&rmrCgo.MBuf{}, nil)
-       rmrMsg := models.NewRmrMessage(123, ranName, payload)
+       var xAction []byte
+       mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction)
+       rmrMessengerMock.On("SendMsg", mbuf, true).Return(&rmrCgo.MBuf{}, nil)
+       rmrMsg := models.NewRmrMessage(123, ranName, payload, xAction)
        rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
-       rmrSender := NewRmrSender(logger, &rmrMessenger)
+       rmrSender := NewRmrSender(logger, rmrMessenger)
        err := rmrSender.Send(rmrMsg)
        assert.Nil(t, err)
-       rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf)
+       rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf, true)
 
 }
 
@@ -70,14 +70,14 @@ func TestRmrSenderSendFailure(t *testing.T) {
 
        ranName := "test"
        payload := []byte("some payload")
-       xaction := []byte(ranName)
-       mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xaction)
-       rmrMessengerMock.On("SendMsg", mbuf).Return(mbuf, fmt.Errorf("rmr send failure"))
-       rmrMsg := models.NewRmrMessage(123, ranName, payload)
+       var xAction []byte
+       mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction)
+       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)
+       rmrSender := NewRmrSender(logger, rmrMessenger)
        err := rmrSender.Send(rmrMsg)
-       rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf)
+       rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf, true)
        assert.NotNil(t, err)
 }