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"
37 func initE2TKeepAliveTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.E2TShutdownManagerMock, *E2TKeepAliveWorker) {
38 logger, err := logger.InitLogger(logger.DebugLevel)
40 t.Errorf("#... - failed to initialize logger, error: %s", err)
42 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3, KeepAliveResponseTimeoutMs: 400, KeepAliveDelayMs: 100}
44 readerMock := &mocks.RnibReaderMock{}
45 writerMock := &mocks.RnibWriterMock{}
46 e2tShutdownManagerMock := &mocks.E2TShutdownManagerMock{}
48 rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
49 e2tInstancesManager := NewE2TInstancesManager(rnibDataService, logger)
51 rmrMessengerMock := &mocks.RmrMessengerMock{}
52 rmrSender := initRmrSender(rmrMessengerMock, logger)
54 e2tKeepAliveWorker := NewE2TKeepAliveWorker(logger, rmrSender, e2tInstancesManager, e2tShutdownManagerMock, config)
56 return rmrMessengerMock, readerMock, writerMock, e2tShutdownManagerMock, &e2tKeepAliveWorker
59 func TestSendKeepAliveRequest(t *testing.T) {
60 rmrMessengerMock, _, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
62 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
64 e2tKeepAliveWorker.SendKeepAliveRequest()
66 var payload, xAction []byte
67 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction)
69 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
72 func TestShutdownExpiredE2T_InternalError(t *testing.T) {
73 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
75 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
77 e2tKeepAliveWorker.E2TKeepAliveExpired()
79 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
82 func TestShutdownExpiredE2T_NoAddresses(t *testing.T) {
83 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
85 addresses := []string{}
87 readerMock.On("GetE2TAddresses").Return(addresses, nil)
89 e2tKeepAliveWorker.E2TKeepAliveExpired()
91 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
94 func TestShutdownExpiredE2T_NotExpired_InternalError(t *testing.T) {
95 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
97 addresses := []string{E2TAddress,E2TAddress2}
98 e2tInstance1 := entities.NewE2TInstance(E2TAddress)
99 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
100 e2tInstance2 := entities.NewE2TInstance(E2TAddress2)
101 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
103 readerMock.On("GetE2TAddresses").Return(addresses, nil)
104 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")))
106 e2tKeepAliveWorker.E2TKeepAliveExpired()
108 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
111 func TestShutdownExpiredE2T_NoE2T(t *testing.T) {
112 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
114 readerMock.On("GetE2TAddresses").Return([]string{}, common.NewResourceNotFoundError("not found"))
116 e2tKeepAliveWorker.E2TKeepAliveExpired()
118 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
121 func TestShutdownExpiredE2T_NotExpired(t *testing.T) {
122 rmrMessengerMock, readerMock, _, _, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
124 addresses := []string{E2TAddress,E2TAddress2}
125 e2tInstance1 := entities.NewE2TInstance(E2TAddress)
126 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
127 e2tInstance2 := entities.NewE2TInstance(E2TAddress2)
128 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
130 readerMock.On("GetE2TAddresses").Return(addresses, nil)
131 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
133 e2tKeepAliveWorker.E2TKeepAliveExpired()
135 rmrMessengerMock.AssertNotCalled(t, "Shutdown")
138 func TestShutdownExpiredE2T_One_E2TExpired(t *testing.T) {
139 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
141 addresses := []string{E2TAddress,E2TAddress2}
142 e2tInstance1 := entities.NewE2TInstance(E2TAddress)
143 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
145 time.Sleep(time.Duration(400) * time.Millisecond)
147 e2tInstance2 := entities.NewE2TInstance(E2TAddress2)
148 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
150 readerMock.On("GetE2TAddresses").Return(addresses, nil)
151 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
152 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
154 e2tKeepAliveWorker.E2TKeepAliveExpired()
156 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 1)
159 func TestShutdownExpiredE2T_Two_E2TExpired(t *testing.T) {
160 _, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
162 addresses := []string{E2TAddress,E2TAddress2}
163 e2tInstance1 := entities.NewE2TInstance(E2TAddress)
164 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
166 e2tInstance2 := entities.NewE2TInstance(E2TAddress2)
167 e2tInstance2.AssociatedRanList = []string{"test4","test5","test6", "test7"}
169 time.Sleep(time.Duration(400) * time.Millisecond)
171 readerMock.On("GetE2TAddresses").Return(addresses, nil)
172 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1, e2tInstance2}, nil)
173 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
174 e2tShutdownManagerMock.On("Shutdown", e2tInstance2).Return(nil)
176 e2tKeepAliveWorker.E2TKeepAliveExpired()
178 e2tShutdownManagerMock.AssertNumberOfCalls(t, "Shutdown", 2)
181 func TestExecute_Two_E2TExpired(t *testing.T) {
182 rmrMessengerMock, readerMock, _, e2tShutdownManagerMock, e2tKeepAliveWorker := initE2TKeepAliveTest(t)
184 addresses := []string{E2TAddress,E2TAddress2}
185 e2tInstance1 := entities.NewE2TInstance(E2TAddress)
186 e2tInstance1.AssociatedRanList = []string{"test1","test2","test3"}
188 readerMock.On("GetE2TAddresses").Return(addresses, nil)
189 readerMock.On("GetE2TInstances",addresses).Return([]*entities.E2TInstance{e2tInstance1}, nil)
190 e2tShutdownManagerMock.On("Shutdown", e2tInstance1).Return(nil)
191 rmrMessengerMock.On("SendMsg", mock.Anything, false).Return(&rmrCgo.MBuf{}, nil)
193 go e2tKeepAliveWorker.Execute()
195 time.Sleep(time.Duration(500) * time.Millisecond)
197 var payload, xAction []byte
198 req := rmrCgo.NewMBuf(rmrCgo.E2_TERM_KEEP_ALIVE_REQ, 0, "", &payload, &xAction)
200 rmrMessengerMock.AssertCalled(t, "SendMsg", req, false)
201 e2tShutdownManagerMock.AssertCalled(t, "Shutdown", e2tInstance1)