RIC-11565:Add support for Multiple E2 Nodes: CU/DU for the case having same GNBId
[ric-plt/e2mgr.git] / E2Manager / handlers / rmrmsghandlers / endc_configuration_update_handler_test.go
index db847ad..6400cf3 100644 (file)
@@ -26,6 +26,7 @@ import (
        "fmt"
        "testing"
        "time"
+       "unsafe"
 )
 
 const PackedEndcConfigurationUpdateAck = "2025000a00000100f70003000000"
@@ -43,31 +44,35 @@ func TestHandleEndcConfigUpdateSuccess(t *testing.T) {
        h, rmrMessengerMock := initEndcConfigurationUpdateHandlerTest(t)
 
        ranName := "test"
-       xaction := []byte(ranName)
+       xAction := []byte("123456aa")
 
        var payload []byte
        _, _ = fmt.Sscanf(PackedEndcConfigurationUpdateAck, "%x", &payload)
+       var msgSrc unsafe.Pointer
 
-       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_CONF_UPDATE_ACK, len(payload), ranName, &payload, &xaction)
-       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: mBuf.Len, Payload: *mBuf.Payload, StartTime: time.Now()}
+       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_CONF_UPDATE_ACK, len(payload), ranName, &payload, &xAction, msgSrc)
+       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: mBuf.Len, Payload: *mBuf.Payload, StartTime: time.Now(),
+               TransactionId: *mBuf.XAction}
        var err error
-       rmrMessengerMock.On("SendMsg", mBuf).Return(&rmrCgo.MBuf{}, err)
+       rmrMessengerMock.On("SendMsg", mBuf, true).Return(&rmrCgo.MBuf{}, err)
        h.Handle(&notificationRequest)
-       rmrMessengerMock.AssertCalled(t, "SendMsg", mBuf)
+       rmrMessengerMock.AssertCalled(t, "SendMsg", mBuf, true)
 }
 
 func TestHandleEndcConfigUpdateFailure(t *testing.T) {
        h, rmrMessengerMock := initEndcConfigurationUpdateHandlerTest(t)
 
        ranName := "test"
-       xaction := []byte(ranName)
+       xAction := []byte("123456aa")
 
        var payload []byte
        _, _ = fmt.Sscanf(PackedEndcConfigurationUpdateFailure, "%x", &payload)
+       var msgSrc unsafe.Pointer
 
-       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_CONF_UPDATE_FAILURE, len(payload), ranName, &payload, &xaction)
-       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: 0, Payload: []byte{0}, StartTime: time.Now()}
-       rmrMessengerMock.On("SendMsg", mBuf).Return(&rmrCgo.MBuf{}, fmt.Errorf("send failure"))
+       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_CONF_UPDATE_FAILURE, len(payload), ranName, &payload, &xAction, msgSrc)
+       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: 0, Payload: []byte{0}, StartTime: time.Now(),
+               TransactionId: *mBuf.XAction}
+       rmrMessengerMock.On("SendMsg", mBuf, true).Return(&rmrCgo.MBuf{}, fmt.Errorf("send failure"))
        h.Handle(&notificationRequest)
-       rmrMessengerMock.AssertCalled(t, "SendMsg", mBuf)
+       rmrMessengerMock.AssertCalled(t, "SendMsg", mBuf, true)
 }