RIC-11565:Add support for Multiple E2 Nodes: CU/DU for the case having same GNBId

[ric-plt/e2mgr.git] / E2Manager / handlers / rmrmsghandlers / x2enb_configuration_update_handler_test.go

diff --git a/E2Manager/handlers/rmrmsghandlers/x2enb_configuration_update_handler_test.go b/E2Manager/handlers/rmrmsghandlers/x2enb_configuration_update_handler_test.go
index 30ba645..5a48a8d 100644 (file)
--- a/E2Manager/handlers/rmrmsghandlers/x2enb_configuration_update_handler_test.go
+++ b/E2Manager/handlers/rmrmsghandlers/x2enb_configuration_update_handler_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 rmrmsghandlers
 
 import (
@@ -23,6 +26,7 @@ import (
        "fmt"
        "testing"
        "time"
+       "unsafe"
 )
 
 const PackedX2EnbConfigurationUpdateAck = "200800080000010011400100"
@@ -40,31 +44,34 @@ func TestHandleX2EnbConfigUpdateSuccess(t *testing.T) {
        h, rmrMessengerMock := initX2EnbConfigurationUpdateHandlerTest(t)
 
        ranName := "test"
-       xaction := []byte(ranName)
-
+       xAction := []byte("123456aa")
        var payload []byte
        _, _ = fmt.Sscanf(PackedX2EnbConfigurationUpdateAck, "%x", &payload)
+       var msgSrc unsafe.Pointer
 
-       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENB_CONFIGURATION_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_ENB_CONFIGURATION_UPDATE_ACK, len(payload), ranName, &payload, &xAction, msgSrc)
+       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: mBuf.Len, Payload: *mBuf.Payload,
+               StartTime: time.Now(), TransactionId: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 TestHandleX2EnbConfigUpdateFailure(t *testing.T) {
        h, rmrMessengerMock := initX2EnbConfigurationUpdateHandlerTest(t)
 
        ranName := "test"
-       xaction := []byte(ranName)
+       xAction := []byte("123456aa")
 
        var payload []byte
        _, _ = fmt.Sscanf(PackedX2EnbConfigurationUpdateFailure, "%x", &payload)
+       var msgSrc unsafe.Pointer
 
-       mBuf := rmrCgo.NewMBuf(rmrCgo.RIC_ENB_CONFIGURATION_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_ENB_CONFIGURATION_UPDATE_FAILURE, len(payload), ranName, &payload, &xAction, msgSrc)
+       notificationRequest := models.NotificationRequest{RanName: mBuf.Meid, Len: 0, Payload: []byte{0},
+               StartTime: time.Now(), TransactionId: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)
 }