Copy latest code

[ric-plt/e2mgr.git] / E2Manager / mocks / rmrMessengerMock.go

diff --git a/E2Manager/mocks/rmrMessengerMock.go b/E2Manager/mocks/rmrMessengerMock.go
index dc4fa15..b517fe9 100644 (file)
--- 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,13 @@ 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) (*rmrCgo.MBuf, error){
-       args := m.Called(msg)
+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)
 }