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