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"
39 func initRanSetupManagerTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibWriterMock, *RanSetupManager) {
40 logger, err := logger.InitLogger(logger.DebugLevel)
42 t.Errorf("#... - failed to initialize logger, error: %s", err)
44 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
46 rmrMessengerMock := &mocks.RmrMessengerMock{}
47 rmrSender := initRmrSender(rmrMessengerMock, logger)
49 readerMock := &mocks.RnibReaderMock{}
51 writerMock := &mocks.RnibWriterMock{}
53 rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
54 ranSetupManager := NewRanSetupManager(logger, rmrSender, rnibDataService)
55 return rmrMessengerMock, writerMock, ranSetupManager
58 func TestExecuteSetupConnectingX2Setup(t *testing.T) {
59 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
63 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
64 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
66 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
68 payload := e2pdus.PackedX2setupRequest
69 xaction := []byte(ranName)
70 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
71 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
73 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
74 t.Errorf("want: success, got: error: %s", err)
77 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
78 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
81 func TestExecuteSetupConnectingEndcX2Setup(t *testing.T) {
82 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
86 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
87 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
89 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
91 payload := e2pdus.PackedEndcX2setupRequest
92 xaction := []byte(ranName)
93 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
94 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
96 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
97 t.Errorf("want: success, got: error: %s", err)
100 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
101 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
104 func TestExecuteSetupDisconnected(t *testing.T) {
105 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
109 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
110 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
111 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
113 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
114 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
117 xaction := []byte(ranName)
118 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
119 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
121 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
122 t.Errorf("want: failure, got: success")
125 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
126 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
129 func TestExecuteSetupConnectingRnibError(t *testing.T) {
130 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
134 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
135 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
136 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
137 var rnibErr = common.NewInternalError(fmt.Errorf("DB error"))
138 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
139 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
142 xaction := []byte(ranName)
143 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
144 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
146 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
147 t.Errorf("want: failure, got: success")
149 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
152 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
153 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
156 func TestExecuteSetupDisconnectedRnibError(t *testing.T) {
157 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
161 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
162 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
163 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
165 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
166 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(common.NewInternalError(fmt.Errorf("DB error")))
169 xaction := []byte(ranName)
170 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
171 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
173 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
174 t.Errorf("want: failure, got: success")
176 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
179 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
180 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
183 func TestExecuteSetupUnsupportedProtocol(t *testing.T) {
184 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
188 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL}
189 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL, ConnectionAttempts: 1}
191 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
193 payload := e2pdus.PackedX2setupRequest
194 xaction := []byte(ranName)
195 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
196 rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
198 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
199 t.Errorf("want: error, got: success")
202 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
203 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
206 func initLog(t *testing.T) *logger.Logger {
207 log, err := logger.InitLogger(logger.InfoLevel)
209 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)