X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fservices%2Frmrsender%2Frmr_sender_test.go;h=0c0129c69c6cf786eb293cdbc71576c76563a491;hb=27feca449ac9a0693b37ff22a900f2534c244387;hp=26ab0b4696d0ef875501ddfd3f76799a61fc1d1a;hpb=de19068aaa1f3d2b415cd960106121ceb167aaa9;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/services/rmrsender/rmr_sender_test.go b/E2Manager/services/rmrsender/rmr_sender_test.go
index 26ab0b4..0c0129c 100644
--- a/E2Manager/services/rmrsender/rmr_sender_test.go
+++ b/E2Manager/services/rmrsender/rmr_sender_test.go
@@ -13,7 +13,10 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
-//
+
+//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//  platform project (RICP).
+
 
 package rmrsender
 
@@ -25,6 +28,7 @@ import (
 	"fmt"
 	"github.com/stretchr/testify/assert"
 	"testing"
+	"unsafe"
 )
 
 func initRmrSenderTest(t *testing.T) (*logger.Logger, *mocks.RmrMessengerMock) {
@@ -50,15 +54,16 @@ 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
+	var msgSrc unsafe.Pointer
+	mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction, msgSrc)
+	rmrMessengerMock.On("SendMsg", mbuf, true).Return(&rmrCgo.MBuf{}, nil)
+	rmrMsg := models.NewRmrMessage(123, ranName, payload, xAction, nil)
 	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)
 
 }
 
@@ -67,14 +72,15 @@ 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
+	var msgSrc unsafe.Pointer
+	mbuf := rmrCgo.NewMBuf(123, len(payload), ranName, &payload, &xAction, msgSrc)
+	rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, fmt.Errorf("rmr send failure"))
+	rmrMsg := models.NewRmrMessage(123, ranName, payload, xAction, nil)
 	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)
 }