[RIC-621] - Fix Automation
[ric-plt/e2mgr.git] / E2Manager / mocks / rmrMessengerMock.go
1 //
2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 //      http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16
17 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 //  platform project (RICP).
19
20
21 package mocks
22
23 import (
24         "e2mgr/logger"
25         "e2mgr/rmrCgo"
26         "github.com/stretchr/testify/mock"
27 )
28
29 type RmrMessengerMock struct {
30         mock.Mock
31 }
32
33 func (m *RmrMessengerMock) Init(port string, maxMsgSize int, flags int, logger *logger.Logger) rmrCgo.RmrMessenger{
34         args := m.Called(port, maxMsgSize, flags, logger)
35         return args.Get(0).(rmrCgo.RmrMessenger)
36 }
37
38 func (m *RmrMessengerMock) SendMsg(msg *rmrCgo.MBuf, printLogs bool) (*rmrCgo.MBuf, error){
39         args := m.Called(msg, printLogs)
40         return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
41 }
42
43 func (m *RmrMessengerMock) WhSendMsg(msg *rmrCgo.MBuf, printLogs bool) (*rmrCgo.MBuf, error){
44         args := m.Called(msg, printLogs)
45         return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
46 }
47
48 func (m *RmrMessengerMock) RecvMsg() (*rmrCgo.MBuf, error){
49         args := m.Called()
50         return args.Get(0).(*rmrCgo.MBuf), args.Error(1)
51 }
52
53 func (m *RmrMessengerMock) RtsMsg(msg *rmrCgo.MBuf){
54         m.Called( )
55 }
56
57 func (m *RmrMessengerMock) FreeMsg(){
58         m.Called( )
59 }
60
61 func (m *RmrMessengerMock) IsReady() bool{
62         args := m.Called( )
63         return args.Bool(0)
64 }
65
66 func (m *RmrMessengerMock) Close(){
67         m.Called( )
68 }