RIC-1044: Add array size checks
[ric-plt/e2mgr.git] / E2Manager / managers / e2t_keep_alive_worker_test.go
index 21d26b7..bd77572 100644 (file)
@@ -32,10 +32,12 @@ import (
        "github.com/stretchr/testify/mock"
        "testing"
        "time"
+       "unsafe"
 )
 
 func initE2TKeepAliveTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.E2TShutdownManagerMock, *E2TKeepAliveWorker) {
-       logger, err := logger.InitLogger(logger.DebugLevel)
+       DebugLevel := int8(4)
+       logger, err := logger.InitLogger(DebugLevel)
        if err != nil {
                t.Errorf("#... - failed to initialize logger, error: %s", err)
        }
@@ -64,7 +66,8 @@ func TestSendKeepAliveRequest(t *testing.T) {
        e2tKeepAliveWorker.SendKeepAliveRequest()
 
        var payload, xAction []byte
-       req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction)
+       var msgSrc unsafe.Pointer
+       req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
 
        rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
 }
@@ -195,8 +198,9 @@ func TestExecute_Two_E2TExpired(t *testing.T) {
        time.Sleep(time.Duration(500) * time.Millisecond)
 
        var payload, xAction []byte
-       req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction)
+       var msgSrc unsafe.Pointer
+       req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
 
        rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
        e2tShutdownManagerMock.AssertCalled(t, "Shutdown", e2tInstance1)
-}
\ No newline at end of file
+}