[RIC-196] Refactor LostConnection....
[ric-plt/e2mgr.git] / 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 }