2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
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
9 // http://www.apache.org/licenses/LICENSE-2.0
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.
18 // This source code is part of the near-RT RIC (RAN Intelligent Controller)
19 // platform project (RICP).
29 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
30 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
31 "github.com/pkg/errors"
32 "github.com/stretchr/testify/mock"
38 func initE2TKeepAliveTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.E2TShutdownManagerMock, *E2TKeepAliveWorker) {
40 logger, err := logger.InitLogger(DebugLevel)
42 t.Errorf("#... - failed to initialize logger, error: %s", err)
44 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3, KeepAliveResponseTimeoutMs: 400, KeepAliveDelayMs: 100}
46 readerMock := &mocks.RnibReaderMock{}
47 writerMock := &mocks.RnibWriterMock{}
48 e2tShutdownManagerMock := &mocks.E2TShutdownManagerMock{}
50 rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
51 e2tInstancesManager := NewE2TInstancesManager(rnibDataService, logger)
53 rmrMessengerMock := &mocks.RmrMessengerMock{}
54 rmrSender := initRmrSender(rmrMessengerMock, logger)
56 e2tKeepAliveWorker := NewE2TKeepAliveWorker(logger, rmrSender, e2tInstancesManager, e2tShutdownManagerMock, config)
58 return rmrMessengerMock, readerMock, writerMock, e2tShutdownManagerMock, &e2tKeepAliveWorker
61 func TestSendKeepAliveRequest(t *testing.T) {
62 rmrMessengerMock, _, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
64 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
66 e2tKeepAliveWorker.SendKeepAliveRequest()
68 var payload, xAction []byte
69 var msgSrc unsafe.Pointer
70 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
72 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
75 func TestShutdownExpiredE2T_InternalError(t *testing.T) {
76 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
78 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
80 e2tKeepAliveWorker.E2TKeepAliveExpired()
82 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
85 func TestShutdownExpiredE2T_NoAddresses(t *testing.T) {
86 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
88 addresses := []string{}
90 readerMock.On("GetE2TAddresses").Return(addresses, nil)
92 e2tKeepAliveWorker.E2TKeepAliveExpired()
94 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
97 func TestShutdownExpiredE2T_NotExpired_InternalError(t *testing.T) {
98 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
100 addresses := []string{E2TAddress,E2TAddress2}
101 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
102 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
103 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
104 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
106 readerMock.On("GetE2TAddresses").Return(addresses, nil)
107 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
109 e2tKeepAliveWorker.E2TKeepAliveExpired()
111 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
114 func TestShutdownExpiredE2T_NoE2T(t *testing.T) {
115 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
117 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewResourceNotFoundError("not found"))
119 e2tKeepAliveWorker.E2TKeepAliveExpired()
121 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
124 func TestShutdownExpiredE2T_NotExpired(t *testing.T) {
125 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
127 addresses := []string{E2TAddress,E2TAddress2}
128 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
129 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
130 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
131 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
133 readerMock.On("GetE2TAddresses").Return(addresses, nil)
134 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
136 e2tKeepAliveWorker.E2TKeepAliveExpired()
138 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
141 func TestShutdownExpiredE2T_One_E2TExpired(t *testing.T) {
142 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
144 addresses := []string{E2TAddress,E2TAddress2}
145 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
146 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
148 time.Sleep(time.Duration(400) * time.Millisecond)
150 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
151 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
153 readerMock.On("GetE2TAddresses").Return(addresses, nil)
154 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
155 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
157 e2tKeepAliveWorker.E2TKeepAliveExpired()
159 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 1)
162 func TestShutdownExpiredE2T_Two_E2TExpired(t *testing.T) {
163 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
165 addresses := []string{E2TAddress,E2TAddress2}
166 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
167 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
169 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
170 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
172 time.Sleep(time.Duration(400) * time.Millisecond)
174 readerMock.On("GetE2TAddresses").Return(addresses, nil)
175 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
176 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
177 e2tShutdownManagerMock.On("Shutdown", e2tInstance2).Return(nil)
179 e2tKeepAliveWorker.E2TKeepAliveExpired()
181 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 2)
184 func TestExecute_Two_E2TExpired(t *testing.T) {
185 rmrMessengerMock, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
187 addresses := []string{E2TAddress,E2TAddress2}
188 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
189 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
191 readerMock.On("GetE2TAddresses").Return(addresses, nil)
192 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1}, nil)
193 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
194 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
196 go e2tKeepAliveWorker.Execute()
198 time.Sleep(time.Duration(500) * time.Millisecond)
200 var payload, xAction []byte
201 var msgSrc unsafe.Pointer
202 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
204 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
205 e2tShutdownManagerMock.AssertCalled(t, "Shutdown", e2tInstance1)