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.
23 "e2mgr/e2managererrors"
29 "e2mgr/providers/httpmsghandlerprovider"
35 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
36 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
37 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
38 "github.com/gorilla/mux"
39 "github.com/stretchr/testify/assert"
40 "github.com/stretchr/testify/mock"
49 func TestX2SetupInvalidBody(t *testing.T) {
51 readerMock, writerMock, rmrMessengerMock, ranSetupManager := initTest(t)
54 readerProvider := func() reader.RNibReader {
57 writerProvider := func() rNibWriter.RNibWriter {
61 header := http.Header{}
62 header.Set("Content-Type", "application/json")
63 httpRequest, _ := http.NewRequest("POST", "http://localhost:3800/v1/nodeb/x2-setup", strings.NewReader("{}{}"))
64 httpRequest.Header = header
66 writer := httptest.NewRecorder()
67 config := configuration.ParseConfiguration()
68 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
69 controller.X2SetupHandler(writer, httpRequest)
71 var errorResponse = parseJsonRequest(t, writer.Body)
73 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
74 assert.Equal(t, e2managererrors.NewInvalidJsonError().Code, errorResponse.Code)
77 func TestX2SetupSuccess(t *testing.T) {
79 readerMock, writerMock, rmrMessengerMock, ranSetupManager := initTest(t)
82 readerProvider := func() reader.RNibReader {
85 writerProvider := func() rNibWriter.RNibWriter {
90 nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
91 readerMock.On("GetNodeb", ranName).Return(nb, nil)
93 var nbUpdated = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
94 writerMock.On("UpdateNodebInfo", nbUpdated).Return(nil)
96 payload := e2pdus.PackedX2setupRequest
97 xaction := []byte(ranName)
98 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
100 rmrMessengerMock.On("SendMsg", mock.Anything, mock.Anything).Return(msg, nil)
102 header := http.Header{}
103 header.Set("Content-Type", "application/json")
104 httpRequest := tests.GetHttpRequest()
105 httpRequest.Header = header
107 writer := httptest.NewRecorder()
108 config := configuration.ParseConfiguration()
109 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
110 controller.X2SetupHandler(writer, httpRequest)
112 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
115 func TestEndcSetupSuccess(t *testing.T) {
117 readerMock, writerMock, rmrMessengerMock, ranSetupManager := initTest(t)
120 readerProvider := func() reader.RNibReader {
123 writerProvider := func() rNibWriter.RNibWriter {
128 nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
129 readerMock.On("GetNodeb", ranName).Return(nb, nil)
131 var nbUpdated = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
132 writerMock.On("UpdateNodebInfo", nbUpdated).Return(nil)
134 payload := e2pdus.PackedEndcX2setupRequest
135 xaction := []byte(ranName)
136 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
138 rmrMessengerMock.On("SendMsg", mock.Anything, mock.Anything).Return(msg, nil)
140 header := http.Header{}
141 header.Set("Content-Type", "application/json")
142 httpRequest := tests.GetHttpRequest()
143 httpRequest.Header = header
145 writer := httptest.NewRecorder()
146 config := configuration.ParseConfiguration()
147 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
148 controller.EndcSetupHandler(writer, httpRequest)
150 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
153 func TestShutdownHandlerRnibError(t *testing.T) {
155 config := configuration.ParseConfiguration()
157 rmrMessengerMock := &mocks.RmrMessengerMock{}
158 readerMock := &mocks.RnibReaderMock{}
159 readerProvider := func() reader.RNibReader {
162 writerMock := &mocks.RnibWriterMock{}
163 writerProvider := func() rNibWriter.RNibWriter {
167 rnibErr := &common.ResourceNotFoundError{}
168 var nbIdentityList []*entities.NbIdentity
169 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
171 writer := httptest.NewRecorder()
172 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
173 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
174 controller.ShutdownHandler(writer, tests.GetHttpRequest())
176 var errorResponse = parseJsonRequest(t, writer.Body)
178 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
179 assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Code)
182 func TestHeaderValidationFailed(t *testing.T) {
184 config := configuration.ParseConfiguration()
186 rmrMessengerMock := &mocks.RmrMessengerMock{}
187 readerMock := &mocks.RnibReaderMock{}
188 readerProvider := func() reader.RNibReader {
191 writerMock := &mocks.RnibWriterMock{}
192 writerProvider := func() rNibWriter.RNibWriter {
196 writer := httptest.NewRecorder()
197 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
198 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
200 header := &http.Header{}
202 controller.handleRequest(writer, header, httpmsghandlerprovider.ShutdownRequest, nil, true)
204 var errorResponse = parseJsonRequest(t, writer.Body)
205 err := e2managererrors.NewHeaderValidationError()
207 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
208 assert.Equal(t, errorResponse.Code, err.Code)
209 assert.Equal(t, errorResponse.Message, err.Message)
212 func TestShutdownStatusNoContent(t *testing.T) {
215 rmrMessengerMock := &mocks.RmrMessengerMock{}
216 readerMock := &mocks.RnibReaderMock{}
217 readerProvider := func() reader.RNibReader {
220 writerMock := &mocks.RnibWriterMock{}
221 writerProvider := func() rNibWriter.RNibWriter {
224 config := configuration.ParseConfiguration()
227 nbIdentityList := []*entities.NbIdentity{}
228 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
230 writer := httptest.NewRecorder()
231 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
232 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
233 controller.ShutdownHandler(writer, tests.GetHttpRequest())
235 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
238 func TestHandleInternalError(t *testing.T) {
240 config := configuration.ParseConfiguration()
242 rmrMessengerMock := &mocks.RmrMessengerMock{}
243 readerMock := &mocks.RnibReaderMock{}
244 readerProvider := func() reader.RNibReader {
247 writerMock := &mocks.RnibWriterMock{}
248 writerProvider := func() rNibWriter.RNibWriter {
252 writer := httptest.NewRecorder()
253 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
254 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
255 err := e2managererrors.NewInternalError()
257 controller.handleErrorResponse(err, writer)
258 var errorResponse = parseJsonRequest(t, writer.Body)
260 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
261 assert.Equal(t, errorResponse.Code, err.Code)
262 assert.Equal(t, errorResponse.Message, err.Message)
265 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
267 config := configuration.ParseConfiguration()
269 rmrMessengerMock := &mocks.RmrMessengerMock{}
270 readerMock := &mocks.RnibReaderMock{}
271 readerProvider := func() reader.RNibReader {
274 writerMock := &mocks.RnibWriterMock{}
275 writerProvider := func() rNibWriter.RNibWriter {
278 writer := httptest.NewRecorder()
279 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
280 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
281 err := e2managererrors.NewCommandAlreadyInProgressError()
283 controller.handleErrorResponse(err, writer)
284 var errorResponse = parseJsonRequest(t, writer.Body)
286 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
287 assert.Equal(t, errorResponse.Code, err.Code)
288 assert.Equal(t, errorResponse.Message, err.Message)
291 func TestValidateHeaders(t *testing.T) {
294 rmrMessengerMock := &mocks.RmrMessengerMock{}
295 readerMock := &mocks.RnibReaderMock{}
296 readerProvider := func() reader.RNibReader {
299 writerMock := &mocks.RnibWriterMock{}
300 writerProvider := func() rNibWriter.RNibWriter {
303 config := configuration.ParseConfiguration()
304 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
305 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
307 header := http.Header{}
308 header.Set("Content-Type", "application/json")
309 result := controller.validateRequestHeader(&header)
311 assert.Nil(t, result)
314 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
316 var errorResponse models.ErrorResponse
317 body, err := ioutil.ReadAll(r)
319 t.Errorf("Error cannot deserialize json request")
321 json.Unmarshal(body, &errorResponse)
326 func initLog(t *testing.T) *logger.Logger {
327 log, err := logger.InitLogger(logger.InfoLevel)
329 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
334 func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T) {
339 readerMock := &mocks.RnibReaderMock{}
340 readerProvider := func() reader.RNibReader {
343 writerMock := &mocks.RnibWriterMock{}
344 writerProvider := func() rNibWriter.RNibWriter {
347 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
348 xaction := []byte(ranName)
349 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
350 rmrMessengerMock := &mocks.RmrMessengerMock{}
351 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
353 config := configuration.ParseConfiguration()
354 rmrService := getRmrService(rmrMessengerMock, log)
356 writer := httptest.NewRecorder()
357 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
358 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
360 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
361 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
363 data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"}
364 b := new(bytes.Buffer)
365 _ = json.NewEncoder(b).Encode(data4Req)
366 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
367 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
369 controller.X2ResetHandler(writer, req)
370 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
374 func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T) {
379 readerMock := &mocks.RnibReaderMock{}
380 readerProvider := func() reader.RNibReader {
383 writerMock := &mocks.RnibWriterMock{}
384 writerProvider := func() rNibWriter.RNibWriter {
388 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
389 xaction := []byte(ranName)
390 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
391 rmrMessengerMock := &mocks.RmrMessengerMock{}
392 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
394 config := configuration.ParseConfiguration()
395 rmrService := getRmrService(rmrMessengerMock, log)
397 writer := httptest.NewRecorder()
398 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
399 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
401 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
402 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
405 b := new(bytes.Buffer)
406 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
407 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
409 controller.X2ResetHandler(writer, req)
410 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
414 func TestX2ResetHandleFailureInvalidBody(t *testing.T) {
419 readerMock := &mocks.RnibReaderMock{}
420 readerProvider := func() reader.RNibReader {
423 writerMock := &mocks.RnibWriterMock{}
424 writerProvider := func() rNibWriter.RNibWriter {
427 rmrMessengerMock := &mocks.RmrMessengerMock{}
429 config := configuration.ParseConfiguration()
430 rmrService := getRmrService(rmrMessengerMock, log)
432 writer := httptest.NewRecorder()
433 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
434 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
436 // Invalid json: attribute name without quotes (should be "cause":).
437 b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"")
438 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
439 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
441 controller.X2ResetHandler(writer, req)
442 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
446 func TestHandleErrorResponse(t *testing.T) {
449 readerMock := &mocks.RnibReaderMock{}
450 readerProvider := func() reader.RNibReader {
453 writerMock := &mocks.RnibWriterMock{}
454 writerProvider := func() rNibWriter.RNibWriter {
457 rmrMessengerMock := &mocks.RmrMessengerMock{}
459 config := configuration.ParseConfiguration()
460 rmrService := getRmrService(rmrMessengerMock, log)
461 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
462 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
464 writer := httptest.NewRecorder()
465 controller.handleErrorResponse(e2managererrors.NewRnibDbError(), writer)
466 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
468 writer = httptest.NewRecorder()
469 controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(), writer)
470 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
472 writer = httptest.NewRecorder()
473 controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(), writer)
474 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
476 writer = httptest.NewRecorder()
477 controller.handleErrorResponse(e2managererrors.NewWrongStateError("", ""), writer)
478 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
480 writer = httptest.NewRecorder()
481 controller.handleErrorResponse(e2managererrors.NewRequestValidationError(), writer)
482 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
484 writer = httptest.NewRecorder()
485 controller.handleErrorResponse(e2managererrors.NewRmrError(), writer)
486 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
488 writer = httptest.NewRecorder()
489 controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(), writer)
490 assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode)
492 writer = httptest.NewRecorder()
493 controller.handleErrorResponse(fmt.Errorf("ErrorError"), writer)
494 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
497 func initTest(t *testing.T)(*mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock, *managers.RanSetupManager) {
500 readerMock := &mocks.RnibReaderMock{}
501 writerMock := &mocks.RnibWriterMock{}
502 writerProvider := func() rNibWriter.RNibWriter {
506 rmrMessengerMock := &mocks.RmrMessengerMock{}
507 rmrService := getRmrService(rmrMessengerMock, log)
509 ranSetupManager := managers.NewRanSetupManager(log, rmrService, writerProvider)
511 return readerMock, writerMock, rmrMessengerMock, ranSetupManager