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/common"
  22         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  23         "github.com/stretchr/testify/mock"
  24 )
  25
  26 type RnibWriterMock struct {
  27         mock.Mock
  28 }
  29
  30 func (rnibWriterMock *RnibWriterMock) SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) common.IRNibError {
  31         args := rnibWriterMock.Called(nbIdentity, nb)
  32
  33         errArg := args.Get(0)
  34
  35         if errArg != nil {
  36                 return errArg.(common.IRNibError)
  37         }
  38
  39         return nil
  40 }
  41
  42 func (rnibWriterMock *RnibWriterMock) UpdateNodebInfo(nodebInfo *entities.NodebInfo) common.IRNibError {
  43         args := rnibWriterMock.Called(nodebInfo)
  44
  45         errArg := args.Get(0)
  46
  47         if errArg != nil {
  48                 return errArg.(common.IRNibError)
  49         }
  50
  51         return nil
  52 }
  53
  54 func (rnibWriterMock *RnibWriterMock) SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) common.IRNibError {
  55         args := rnibWriterMock.Called(inventoryName, ranLoadInformation)
  56
  57         errArg := args.Get(0)
  58
  59         if errArg != nil {
  60                 return errArg.(common.IRNibError)
  61         }
  62
  63         return nil
  64 }