[RIC-621] - Fix Automation
[ric-plt/e2mgr.git] / E2Manager / mocks / rnibReaderMock.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 package mocks
21
22 import (
23         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
24         "github.com/stretchr/testify/mock"
25 )
26
27 type RnibReaderMock struct {
28         mock.Mock
29 }
30
31 func (m *RnibReaderMock) GetNodeb(inventoryName string) (*entities.NodebInfo, error) {
32         args := m.Called(inventoryName)
33
34         errArg := args.Get(1);
35         if errArg != nil {
36                 return args.Get(0).(*entities.NodebInfo), errArg.(error);
37         }
38
39         return args.Get(0).(*entities.NodebInfo), nil
40 }
41
42 func (m *RnibReaderMock) GetNodebByGlobalNbId(nodeType entities.Node_Type, globalNbId *entities.GlobalNbId) (*entities.NodebInfo, error) {
43         args := m.Called(nodeType, globalNbId)
44
45         errArg := args.Get(1);
46         if errArg != nil {
47                 return args.Get(0).(*entities.NodebInfo), errArg.(error);
48         }
49
50         return args.Get(0).(*entities.NodebInfo), nil
51 }
52
53 func (m *RnibReaderMock) GetCellList(inventoryName string) (*entities.Cells, error) {
54         args := m.Called(inventoryName)
55
56         errArg := args.Get(1);
57         if errArg != nil {
58                 return args.Get(0).(*entities.Cells), errArg.(error);
59         }
60
61         return args.Get(0).(*entities.Cells), nil
62 }
63
64 func (m *RnibReaderMock) GetListGnbIds() ([]*entities.NbIdentity, error) {
65         args := m.Called()
66
67         errArg := args.Get(1);
68         if errArg != nil {
69                 return args.Get(0).([]*entities.NbIdentity), errArg.(error);
70         }
71
72         return args.Get(0).([]*entities.NbIdentity), nil
73 }
74
75 func (m *RnibReaderMock) GetListEnbIds() ([]*entities.NbIdentity, error) {
76         args := m.Called()
77
78         errArg := args.Get(1);
79         if errArg != nil {
80                 return args.Get(0).([]*entities.NbIdentity), errArg.(error);
81         }
82
83         return args.Get(0).([]*entities.NbIdentity), nil
84
85 }
86
87 func (m *RnibReaderMock) GetCountGnbList() (int, error) {
88         args := m.Called()
89
90         errArg := args.Get(1);
91         if errArg != nil {
92                 return args.Int(0), errArg.(error);
93         }
94
95         return args.Int(0), nil
96
97 }
98
99 func (m *RnibReaderMock) GetCell(inventoryName string, pci uint32) (*entities.Cell, error) {
100         args := m.Called(inventoryName, pci)
101
102         errArg := args.Get(1);
103         if errArg != nil {
104                 return args.Get(0).(*entities.Cell), errArg.(error);
105         }
106
107         return args.Get(0).(*entities.Cell), nil
108 }
109
110 func (m *RnibReaderMock) GetCellById(cellType entities.Cell_Type, cellId string) (*entities.Cell, error) {
111         args := m.Called(cellType, cellId)
112
113         errArg := args.Get(1);
114         if errArg != nil {
115                 return args.Get(0).(*entities.Cell), errArg.(error);
116         }
117
118         return args.Get(0).(*entities.Cell), nil
119 }
120
121 func (m *RnibReaderMock) GetListNodebIds() ([]*entities.NbIdentity, error) {
122         args := m.Called()
123
124         errArg := args.Get(1)
125
126         if errArg != nil {
127                 return args.Get(0).([]*entities.NbIdentity), errArg.(error)
128         }
129
130         return args.Get(0).([]*entities.NbIdentity), nil
131 }
132
133 func (m *RnibReaderMock) GetRanLoadInformation(inventoryName string) (*entities.RanLoadInformation, error) {
134         args := m.Called()
135
136         errArg := args.Get(1)
137
138         if errArg != nil {
139                 return args.Get(0).(*entities.RanLoadInformation), errArg.(error)
140         }
141
142         return args.Get(0).(*entities.RanLoadInformation), nil
143 }
144
145 func (m *RnibReaderMock) GetE2TInstance(e2taddress string) (*entities.E2TInstance, error) {
146         args := m.Called(e2taddress)
147         return args.Get(0).(*entities.E2TInstance), args.Error(1)
148 }
149
150 func (m *RnibReaderMock) GetE2TInstances(addresses []string) ([]*entities.E2TInstance, error) {
151         args := m.Called(addresses)
152         return args.Get(0).([]*entities.E2TInstance), args.Error(1)
153 }
154
155 func (m *RnibReaderMock) GetE2TAddresses() ([]string, error) {
156         args := m.Called()
157         return args.Get(0).([]string), args.Error(1)
158 }
159
160 func (m *RnibReaderMock) GetGeneralConfiguration() (*entities.GeneralConfiguration, error) {
161         args := m.Called()
162         return args.Get(0).(*entities.GeneralConfiguration), args.Error(1)
163 }