[RICPLT-2526] KeepAlive worker and response.
[ric-plt/e2mgr.git] / E2Manager / mocks / e2t_instances_manager_mock.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 package mocks
18
19 import (
20         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
21         "github.com/stretchr/testify/mock"
22 )
23
24 type E2TInstancesManagerMock struct {
25         mock.Mock
26 }
27
28 func (m *E2TInstancesManagerMock) GetE2TInstance(e2tAddress string) (*entities.E2TInstance, error) {
29         args := m.Called(e2tAddress)
30
31         return args.Get(0).(*entities.E2TInstance), args.Error(1)
32 }
33
34 func (m *E2TInstancesManagerMock) AddE2TInstance(e2tInstanceAddress string) error {
35         args := m.Called(e2tInstanceAddress)
36         return args.Error(0)
37 }
38
39 func (m *E2TInstancesManagerMock) RemoveE2TInstance(e2tInstance *entities.E2TInstance) error {
40         args := m.Called(e2tInstance)
41         return args.Error(0)
42 }
43
44 func (m *E2TInstancesManagerMock) SelectE2TInstance() (string, error) {
45         args := m.Called()
46         return args.String(0), args.Error(1)
47 }
48
49 func (m *E2TInstancesManagerMock) AssociateRan(ranName string, e2tAddress string) error {
50         args := m.Called(ranName, e2tAddress)
51         return args.Error(0)
52
53 }
54 func (m *E2TInstancesManagerMock) DissociateRan(ranName string, e2tAddress string) error {
55         args := m.Called(ranName, e2tAddress)
56         return args.Error(0)
57
58 }
59
60 func (m *E2TInstancesManagerMock) GetE2TInstances() ([]*entities.E2TInstance, error) {
61         args := m.Called()
62
63         return args.Get(0).([]*entities.E2TInstance), args.Error(1)
64 }
65
66 func (m *E2TInstancesManagerMock) GetE2TInstancesNoLogs() ([]*entities.E2TInstance, error) {
67         args := m.Called()
68
69         return args.Get(0).([]*entities.E2TInstance), args.Error(1)
70 }
71
72 func (m *E2TInstancesManagerMock) ResetKeepAliveTimestamp(e2tAddress string) error {
73         args := m.Called(e2tAddress)
74         return args.Error(0)
75
76 }