E2Manager/mocks/rnibWriterMock.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         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  25         "github.com/stretchr/testify/mock"
  26 )
  27
  28 type RnibWriterMock struct {
  29         mock.Mock
  30 }
  31
  32 func (rnibWriterMock *RnibWriterMock) SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error {
  33         args := rnibWriterMock.Called(nbIdentity, nb)
  34
  35         errArg := args.Get(0)
  36
  37         if errArg != nil {
  38                 return errArg.(error)
  39         }
  40
  41         return nil
  42 }
  43
  44 func (rnibWriterMock *RnibWriterMock) UpdateNodebInfo(nodebInfo *entities.NodebInfo) error {
  45         args := rnibWriterMock.Called(nodebInfo)
  46
  47         errArg := args.Get(0)
  48
  49         if errArg != nil {
  50                 return errArg.(error)
  51         }
  52
  53         return nil
  54 }
  55
  56 func (rnibWriterMock *RnibWriterMock) SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) error {
  57         args := rnibWriterMock.Called(inventoryName, ranLoadInformation)
  58
  59         errArg := args.Get(0)
  60
  61         if errArg != nil {
  62                 return errArg.(error)
  63         }
  64
  65         return nil
  66 }
  67
  68 func (rnibWriterMock *RnibWriterMock) SaveE2TInstance(e2tInstance *entities.E2TInstance) error {
  69         args := rnibWriterMock.Called(e2tInstance)
  70
  71         return args.Error(0)
  72 }
  73
  74 func (rnibWriterMock *RnibWriterMock) SaveE2TAddresses(addresses []string) error {
  75         args := rnibWriterMock.Called(addresses)
  76
  77         return args.Error(0)
  78 }
  79
  80 func (rnibWriterMock *RnibWriterMock) RemoveE2TInstance(address string) error {
  81         args := rnibWriterMock.Called(address)
  82
  83         return args.Error(0)
  84 }