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) {
39 logger, err := logger.InitLogger(logger.DebugLevel)
41 t.Errorf("#... - failed to initialize logger, error: %s", err)
43 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3, KeepAliveResponseTimeoutMs: 400, KeepAliveDelayMs: 100}
45 readerMock := &mocks.RnibReaderMock{}
46 writerMock := &mocks.RnibWriterMock{}
47 e2tShutdownManagerMock := &mocks.E2TShutdownManagerMock{}
49 rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
50 e2tInstancesManager := NewE2TInstancesManager(rnibDataService, logger)
52 rmrMessengerMock := &mocks.RmrMessengerMock{}
53 rmrSender := initRmrSender(rmrMessengerMock, logger)
55 e2tKeepAliveWorker := NewE2TKeepAliveWorker(logger, rmrSender, e2tInstancesManager, e2tShutdownManagerMock, config)
57 return rmrMessengerMock, readerMock, writerMock, e2tShutdownManagerMock, &e2tKeepAliveWorker
60 func TestSendKeepAliveRequest(t *testing.T) {
61 rmrMessengerMock, _, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
63 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
65 e2tKeepAliveWorker.SendKeepAliveRequest()
67 var payload, xAction []byte
68 var msgSrc unsafe.Pointer
69 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
71 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
74 func TestShutdownExpiredE2T_InternalError(t *testing.T) {
75 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
77 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
79 e2tKeepAliveWorker.E2TKeepAliveExpired()
81 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
84 func TestShutdownExpiredE2T_NoAddresses(t *testing.T) {
85 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
87 addresses := []string{}
89 readerMock.On("GetE2TAddresses").Return(addresses, nil)
91 e2tKeepAliveWorker.E2TKeepAliveExpired()
93 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
96 func TestShutdownExpiredE2T_NotExpired_InternalError(t *testing.T) {
97 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
99 addresses := []string{E2TAddress,E2TAddress2}
100 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
101 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
102 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
103 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
105 readerMock.On("GetE2TAddresses").Return(addresses, nil)
106 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
108 e2tKeepAliveWorker.E2TKeepAliveExpired()
110 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
113 func TestShutdownExpiredE2T_NoE2T(t *testing.T) {
114 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
116 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewResourceNotFoundError("not found"))
118 e2tKeepAliveWorker.E2TKeepAliveExpired()
120 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
123 func TestShutdownExpiredE2T_NotExpired(t *testing.T) {
124 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
126 addresses := []string{E2TAddress,E2TAddress2}
127 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
128 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
129 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
130 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
132 readerMock.On("GetE2TAddresses").Return(addresses, nil)
133 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
135 e2tKeepAliveWorker.E2TKeepAliveExpired()
137 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
140 func TestShutdownExpiredE2T_One_E2TExpired(t *testing.T) {
141 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
143 addresses := []string{E2TAddress,E2TAddress2}
144 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
145 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
147 time.Sleep(time.Duration(400) * time.Millisecond)
149 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
150 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
152 readerMock.On("GetE2TAddresses").Return(addresses, nil)
153 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
154 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
156 e2tKeepAliveWorker.E2TKeepAliveExpired()
158 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 1)
161 func TestShutdownExpiredE2T_Two_E2TExpired(t *testing.T) {
162 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
164 addresses := []string{E2TAddress,E2TAddress2}
165 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
166 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
168 e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
169 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
171 time.Sleep(time.Duration(400) * time.Millisecond)
173 readerMock.On("GetE2TAddresses").Return(addresses, nil)
174 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
175 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
176 e2tShutdownManagerMock.On("Shutdown", e2tInstance2).Return(nil)
178 e2tKeepAliveWorker.E2TKeepAliveExpired()
180 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 2)
183 func TestExecute_Two_E2TExpired(t *testing.T) {
184 rmrMessengerMock, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
186 addresses := []string{E2TAddress,E2TAddress2}
187 e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
188 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
190 readerMock.On("GetE2TAddresses").Return(addresses, nil)
191 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1}, nil)
192 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
193 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
195 go e2tKeepAliveWorker.Execute()
197 time.Sleep(time.Duration(500) * time.Millisecond)
199 var payload, xAction []byte
200 var msgSrc unsafe.Pointer
201 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction, msgSrc)
203 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
204 e2tShutdownManagerMock.AssertCalled(t, "Shutdown", e2tInstance1)