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.
17 // This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 // platform project (RICP).
25 "e2mgr/e2managererrors"
32 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
33 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
34 "github.com/stretchr/testify/assert"
35 "github.com/stretchr/testify/mock"
40 func initRanSetupManagerTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibWriterMock, *RanSetupManager) {
41 logger, err := logger.InitLogger(logger.DebugLevel)
43 t.Errorf("#... - failed to initialize logger, error: %s", err)
45 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
47 rmrMessengerMock := &mocks.RmrMessengerMock{}
48 rmrSender := initRmrSender(rmrMessengerMock, logger)
50 readerMock := &mocks.RnibReaderMock{}
52 writerMock := &mocks.RnibWriterMock{}
54 rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
55 ranSetupManager := NewRanSetupManager(logger, rmrSender, rnibDataService)
56 return rmrMessengerMock, writerMock, ranSetupManager
59 func TestExecuteSetupConnectingX2Setup(t *testing.T) {
60 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
64 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
65 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
67 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
69 payload := e2pdus.PackedX2setupRequest
70 xAction := []byte(ranName)
71 var msgSrc unsafe.Pointer
72 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
73 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
75 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
76 t.Errorf("want: success, got: error: %s", err)
79 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
80 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
83 func TestExecuteSetupConnectingEndcX2Setup(t *testing.T) {
84 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
88 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
89 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
91 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
93 payload := e2pdus.PackedEndcX2setupRequest
94 xAction := []byte(ranName)
95 var msgSrc unsafe.Pointer
96 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
97 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
99 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
100 t.Errorf("want: success, got: error: %s", err)
103 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
104 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
107 func TestExecuteSetupDisconnected(t *testing.T) {
108 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
112 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
113 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
114 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
116 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
117 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
120 xAction := []byte(ranName)
121 var msgSrc unsafe.Pointer
122 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
123 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
125 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
126 t.Errorf("want: failure, got: success")
129 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
130 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
133 func TestExecuteSetupConnectingRnibError(t *testing.T) {
134 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
138 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
139 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
140 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
141 var rnibErr = common.NewInternalError(fmt.Errorf("DB error"))
142 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
143 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
146 xAction := []byte(ranName)
147 var msgSrc unsafe.Pointer
148 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
149 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
151 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
152 t.Errorf("want: failure, got: success")
154 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
157 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
158 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
161 func TestExecuteSetupDisconnectedRnibError(t *testing.T) {
162 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
166 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
167 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
168 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
170 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
171 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(common.NewInternalError(fmt.Errorf("DB error")))
174 xAction := []byte(ranName)
175 var msgSrc unsafe.Pointer
176 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
177 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
179 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
180 t.Errorf("want: failure, got: success")
182 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
185 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
186 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
189 func TestExecuteSetupUnsupportedProtocol(t *testing.T) {
190 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
194 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL}
195 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL}
197 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
199 payload := e2pdus.PackedX2setupRequest
200 xAction := []byte(ranName)
201 var msgSrc unsafe.Pointer
202 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
203 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
205 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
206 t.Errorf("want: error, got: success")
209 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
210 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
213 func initLog(t *testing.T) *logger.Logger {
214 log, err := logger.InitLogger(logger.InfoLevel)
216 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)