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 }