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.
22 "e2mgr/e2managererrors"
30 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
31 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
32 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
33 "github.com/stretchr/testify/assert"
34 "github.com/stretchr/testify/mock"
38 func initRanSetupManagerTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibWriterMock, *RanSetupManager) {
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}
45 rmrMessengerMock := &mocks.RmrMessengerMock{}
46 rmrSender := initRmrSender(rmrMessengerMock, logger)
48 readerMock := &mocks.RnibReaderMock{}
49 rnibReaderProvider := func() reader.RNibReader {
52 writerMock := &mocks.RnibWriterMock{}
53 rnibWriterProvider := func() rNibWriter.RNibWriter {
56 rnibDataService := services.NewRnibDataService(logger, config, rnibReaderProvider, rnibWriterProvider)
57 ranSetupManager := NewRanSetupManager(logger, rmrSender, rnibDataService)
58 return rmrMessengerMock, writerMock, ranSetupManager
61 func TestExecuteSetupConnectingX2Setup(t *testing.T) {
62 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
66 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
67 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
69 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
71 payload := e2pdus.PackedX2setupRequest
72 xaction := []byte(ranName)
73 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
74 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
76 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
77 t.Errorf("want: success, got: error: %s", err)
80 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
81 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
84 func TestExecuteSetupConnectingEndcX2Setup(t *testing.T) {
85 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
89 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
90 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
92 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
94 payload := e2pdus.PackedEndcX2setupRequest
95 xaction := []byte(ranName)
96 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
97 rmrMessengerMock.On("SendMsg", mock.Anything).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, ConnectionAttempts: 1}
114 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
116 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
117 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
120 xaction := []byte(ranName)
121 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
122 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, fmt.Errorf("send failure"))
124 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
125 t.Errorf("want: failure, got: success")
128 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
129 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
132 func TestExecuteSetupConnectingRnibError(t *testing.T) {
133 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
137 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
138 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
139 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
140 var rnibErr = common.NewInternalError(fmt.Errorf("DB error"))
141 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
142 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
145 xaction := []byte(ranName)
146 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
147 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, fmt.Errorf("send failure"))
149 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
150 t.Errorf("want: failure, got: success")
152 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
155 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
156 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
159 func TestExecuteSetupDisconnectedRnibError(t *testing.T) {
160 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
164 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
165 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
166 var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
168 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
169 writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(common.NewInternalError(fmt.Errorf("DB error")))
172 xaction := []byte(ranName)
173 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
174 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, fmt.Errorf("send failure"))
176 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
177 t.Errorf("want: failure, got: success")
179 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
182 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
183 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
186 func TestExecuteSetupUnsupportedProtocol(t *testing.T) {
187 rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
191 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL}
192 var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL, ConnectionAttempts: 1}
194 writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
196 payload := e2pdus.PackedX2setupRequest
197 xaction := []byte(ranName)
198 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
199 rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
201 if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
202 t.Errorf("want: error, got: success")
205 writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
206 rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
209 func initLog(t *testing.T) *logger.Logger {
210 log, err := logger.InitLogger(logger.InfoLevel)
212 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)