X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmocks%2FrmrMessengerMock.go;h=732f69631d77f3725a0e4688ba830e14be6c6fc5;hb=071419a702d7642a4d20b662c0219fdda0c80d1c;hp=ff6c71fabe8894aa3a9bf4004aa3750517cfe9d8;hpb=07ef76dd471a0892a893c90e0ab06713aee34be1;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/mocks/rmrMessengerMock.go b/E2Manager/mocks/rmrMessengerMock.go
index ff6c71f..732f696 100644
--- a/E2Manager/mocks/rmrMessengerMock.go
+++ b/E2Manager/mocks/rmrMessengerMock.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 mocks
 
@@ -27,18 +30,23 @@ 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) WhSendMsg(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)
 }