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

diff --git a/E2Manager/mocks/rmrMessengerMock.go b/E2Manager/mocks/rmrMessengerMock.go
index dc4fa15..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,13 +30,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, printLogs bool) (*rmrCgo.MBuf, error){
+	args := m.Called(msg, printLogs)
+	return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
 }
 
-func (m *RmrMessengerMock) SendMsg(msg *rmrCgo.MBuf) (*rmrCgo.MBuf, error){
-	args := m.Called(msg)
+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)
 }