[RICPLT-2526] KeepAlive worker and response.
[ric-plt/e2mgr.git] / E2Manager / mocks / rmrMessengerMock.go
index ff6c71f..c8d32f2 100644 (file)
@@ -27,18 +27,18 @@ type RmrMessengerMock struct {
        mock.Mock
 }
 
-func (m *RmrMessengerMock) Init(port string, maxMsgSize int, flags int, logger *logger.Logger) *rmrCgo.RmrMessenger{
+func (m *RmrMessengerMock) Init(port string, maxMsgSize int, flags int, logger *logger.Logger) rmrCgo.RmrMessenger{
        args := m.Called(port, maxMsgSize, flags, logger)
-       return args.Get(0).(*rmrCgo.RmrMessenger)
+       return args.Get(0).(rmrCgo.RmrMessenger)
 }
 
-func (m *RmrMessengerMock) SendMsg(msg *rmrCgo.MBuf, maxMsgSize int) (*rmrCgo.MBuf, error){
-       args := m.Called(msg, maxMsgSize)
+func (m *RmrMessengerMock) SendMsg(msg *rmrCgo.MBuf, printLogs bool) (*rmrCgo.MBuf, error){
+       args := m.Called(msg, printLogs)
        return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
 }
 
 func (m *RmrMessengerMock) RecvMsg() (*rmrCgo.MBuf, error){
-       args := m.Called( )
+       args := m.Called()
        return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
 }