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"
33 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
34 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
35 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
36 "github.com/stretchr/testify/assert"
37 "github.com/stretchr/testify/mock"
41 func initRanSetupManagerTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibWriterMock, *RanSetupManager) {
42 logger, err := logger.InitLogger(logger.DebugLevel)
44 t.Errorf("#... - failed to initialize logger, error: %s", err)
46 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
48 rmrMessengerMock := &mocks.RmrMessengerMock{}
49 rmrSender := initRmrSender(rmrMessengerMock, logger)
51 readerMock := &mocks.RnibReaderMock{}
52 rnibReaderProvider := func() reader.RNibReader {
55 writerMock := &mocks.RnibWriterMock{}
56 rnibWriterProvider := func() rNibWriter.RNibWriter {
59 rnibDataService := services.NewRnibDataService(logger, config, rnibReaderProvider, rnibWriterProvider)
60 ranSetupManager := NewRanSetupManager(logger, rmrSender, rnibDataService)
61 return rmrMessengerMock, writerMock, ranSetupManager
64 func TestExecuteSetupConnectingX2Setup(t *testing.T) {
65 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
69 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
70 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
72 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
74 payload := e2pdus.PackedX2setupRequest
75 xaction := []byte(ranName)
76 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
77 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
79 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
80 t.Errorf("want: success, got: error: %s", err)
83 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
84 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
87 func TestExecuteSetupConnectingEndcX2Setup(t *testing.T) {
88 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
92 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
93 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
95 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
97 payload := e2pdus.PackedEndcX2setupRequest
98 xaction := []byte(ranName)
99 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
100 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
102 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
103 t.Errorf("want: success, got: error: %s", err)
106 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
107 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
110 func TestExecuteSetupDisconnected(t *testing.T) {
111 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
115 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
116 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
117 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
119 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
120 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
123 xaction := []byte(ranName)
124 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
125 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, fmt.Errorf("send failure"))
127 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
128 t.Errorf("want: failure, got: success")
131 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
132 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
135 func TestExecuteSetupConnectingRnibError(t *testing.T) {
136 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
140 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
141 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
142 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
143 var rnibErr = common.NewInternalError(fmt.Errorf("DB error"))
144 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
145 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
148 xaction := []byte(ranName)
149 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
150 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, fmt.Errorf("send failure"))
152 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
153 t.Errorf("want: failure, got: success")
155 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
158 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
159 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
162 func TestExecuteSetupDisconnectedRnibError(t *testing.T) {
163 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
167 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
168 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
169 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
171 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
172 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(common.NewInternalError(fmt.Errorf("DB error")))
175 xaction := []byte(ranName)
176 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
177 rmrMessengerMock.On("SendMsg", mock.Anything).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, ConnectionAttempts: 1}
197 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
199 payload := e2pdus.PackedX2setupRequest
200 xaction := []byte(ranName)
201 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
202 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
204 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
205 t.Errorf("want: error, got: success")
208 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
209 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
212 func initLog(t *testing.T) *logger.Logger {
213 log, err := logger.InitLogger(logger.InfoLevel)
215 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)