// // Copyright 2019 AT&T Intellectual Property // Copyright 2019 Nokia // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // This source code is part of the near-RT RIC (RAN Intelligent Controller) // platform project (RICP). package controllers import ( "bytes" "e2mgr/clients" "e2mgr/configuration" "e2mgr/e2managererrors" "e2mgr/logger" "e2mgr/managers" "e2mgr/mocks" "e2mgr/models" "e2mgr/providers/httpmsghandlerprovider" "e2mgr/rmrCgo" "e2mgr/services" "e2mgr/services/rmrsender" "e2mgr/tests" "encoding/json" "fmt" "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common" "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities" "github.com/golang/protobuf/jsonpb" "github.com/gorilla/mux" "github.com/pkg/errors" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/mock" "io" "io/ioutil" "net/http" "net/http/httptest" "strings" "testing" "unsafe" ) const ( RanName = "test" AssociatedE2TInstanceAddress = "10.0.2.15:38000" CorruptedJson = "{\"errorCode\":401,\"errorMessage\":\"corrupted json\"}" ValidationFailureJson = "{\"errorCode\":402,\"errorMessage\":\"Validation error\"}" ResourceNotFoundJson = "{\"errorCode\":404,\"errorMessage\":\"Resource not found\"}" NodebExistsJson = "{\"errorCode\":406,\"errorMessage\":\"Nodeb already exists\"}" RnibErrorJson = "{\"errorCode\":500,\"errorMessage\":\"RNIB error\"}" InternalErrorJson = "{\"errorCode\":501,\"errorMessage\":\"Internal Server Error. Please try again later\"}" AddEnbUrl = "/nodeb/enb" ) var ( ServedNrCellInformationRequiredFields = []string{"cellId", "choiceNrMode", "nrMode", "servedPlmns"} NrNeighbourInformationRequiredFields = []string{"nrCgi", "choiceNrMode", "nrMode"} AddEnbRequestRequiredFields = []string{"ranName", "enb", "globalNbId"} UpdateEnbRequestRequiredFields = []string{"enb"} GlobalIdRequiredFields = []string{"plmnId", "nbId"} EnbRequiredFields = []string{"enbType", "servedCells"} ServedCellRequiredFields = []string{"broadcastPlmns", "cellId", "choiceEutraMode", "eutraMode", "tac"} ) type controllerGetNodebTestContext struct { ranName string nodebInfo *entities.NodebInfo rnibError error expectedStatusCode int expectedJsonResponse string } type controllerGetNodebIdListTestContext struct { nodebIdList []*entities.NbIdentity rnibError error expectedStatusCode int expectedJsonResponse string } type getNodebInfoResult struct { nodebInfo *entities.NodebInfo rnibError error } type updateGnbCellsParams struct { err error } type updateEnbCellsParams struct { err error } type addEnbParams struct { err error } type addNbIdentityParams struct { err error } type removeServedCellsParams struct { servedCellInfo []*entities.ServedCellInfo err error } type removeServedNrCellsParams struct { servedNrCells []*entities.ServedNRCell err error } type controllerUpdateEnbTestContext struct { getNodebInfoResult *getNodebInfoResult removeServedCellsParams *removeServedCellsParams updateEnbCellsParams *updateEnbCellsParams requestBody map[string]interface{} expectedStatusCode int expectedJsonResponse string } type controllerUpdateGnbTestContext struct { getNodebInfoResult *getNodebInfoResult removeServedNrCellsParams *removeServedNrCellsParams updateGnbCellsParams *updateGnbCellsParams requestBody map[string]interface{} expectedStatusCode int expectedJsonResponse string } type controllerAddEnbTestContext struct { getNodebInfoResult *getNodebInfoResult addEnbParams *addEnbParams addNbIdentityParams *addNbIdentityParams requestBody map[string]interface{} expectedStatusCode int expectedJsonResponse string } type controllerDeleteEnbTestContext struct { getNodebInfoResult *getNodebInfoResult expectedStatusCode int expectedJsonResponse string } func generateServedNrCells(cellIds ...string) []*entities.ServedNRCell { servedNrCells := []*entities.ServedNRCell{} for _, v := range cellIds { servedNrCells = append(servedNrCells, &entities.ServedNRCell{ServedNrCellInformation: &entities.ServedNRCellInformation{ CellId: v, ChoiceNrMode: &entities.ServedNRCellInformation_ChoiceNRMode{ Fdd: &entities.ServedNRCellInformation_ChoiceNRMode_FddInfo{ }, }, NrMode: entities.Nr_FDD, NrPci: 5, ServedPlmns: []string{"whatever"}, }}) } return servedNrCells } func generateServedCells(cellIds ...string) []*entities.ServedCellInfo { var servedCells []*entities.ServedCellInfo for i, v := range cellIds { servedCells = append(servedCells, &entities.ServedCellInfo{ CellId: v, ChoiceEutraMode: &entities.ChoiceEUTRAMode{ Fdd: &entities.FddInfo{}, }, Pci: uint32(i + 1), BroadcastPlmns: []string{"whatever"}, }) } return servedCells } func buildNrNeighbourInformation(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "nrCgi": "whatever", "choiceNrMode": map[string]interface{}{ "tdd": map[string]interface{}{}, }, "nrMode": 1, "nrPci": 1, } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func buildServedNrCellInformation(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "cellId": "whatever", "choiceNrMode": map[string]interface{}{ "fdd": map[string]interface{}{}, }, "nrMode": 1, "nrPci": 1, "servedPlmns": []interface{}{ "whatever", }, } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func buildServedCell(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "cellId": "whatever", "choiceEutraMode": map[string]interface{}{ "fdd": map[string]interface{}{}, }, "eutraMode": 1, "pci": 1, "tac": "whatever3", "broadcastPlmns": []interface{}{ "whatever", }, } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func getUpdateEnbRequest(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "enb": buildEnb(propToOmit), } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func getAddEnbRequest(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "ranName": RanName, "globalNbId": buildGlobalNbId(""), "enb": buildEnb(""), } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func buildEnb(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "enbType": 1, "servedCells": []interface{}{ buildServedCell(""), }} if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func buildGlobalNbId(propToOmit string) map[string]interface{} { ret := map[string]interface{}{ "plmnId": "whatever", "nbId": "whatever2", } if len(propToOmit) != 0 { delete(ret, propToOmit) } return ret } func setupControllerTest(t *testing.T) (*NodebController, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock, *mocks.E2TInstancesManagerMock, managers.RanListManager) { log := initLog(t) config := configuration.ParseConfiguration() rmrMessengerMock := &mocks.RmrMessengerMock{} readerMock := &mocks.RnibReaderMock{} writerMock := &mocks.RnibWriterMock{} rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock) rmrSender := getRmrSender(rmrMessengerMock, log) e2tInstancesManager := &mocks.E2TInstancesManagerMock{} httpClientMock := &mocks.HttpClientMock{} rmClient := clients.NewRoutingManagerClient(log, config, httpClientMock) ranListManager := managers.NewRanListManager(log, rnibDataService) ranAlarmService := &mocks.RanAlarmServiceMock{} ranConnectStatusChangeManager := managers.NewRanConnectStatusChangeManager(log, rnibDataService, ranListManager, ranAlarmService) nodebValidator := managers.NewNodebValidator() updateEnbManager := managers.NewUpdateEnbManager(log, rnibDataService, nodebValidator) updateGnbManager := managers.NewUpdateGnbManager(log, rnibDataService, nodebValidator) handlerProvider := httpmsghandlerprovider.NewIncomingRequestHandlerProvider(log, rmrSender, config, rnibDataService, e2tInstancesManager, rmClient, ranConnectStatusChangeManager, nodebValidator, updateEnbManager, updateGnbManager, ranListManager) controller := NewNodebController(log, handlerProvider) return controller, readerMock, writerMock, rmrMessengerMock, e2tInstancesManager, ranListManager } func setupDeleteEnbControllerTest(t *testing.T, preAddNbIdentity bool) (*NodebController, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *entities.NbIdentity) { log := initLog(t) config := configuration.ParseConfiguration() rmrMessengerMock := &mocks.RmrMessengerMock{} readerMock := &mocks.RnibReaderMock{} writerMock := &mocks.RnibWriterMock{} rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock) rmrSender := getRmrSender(rmrMessengerMock, log) e2tInstancesManager := &mocks.E2TInstancesManagerMock{} httpClientMock := &mocks.HttpClientMock{} rmClient := clients.NewRoutingManagerClient(log, config, httpClientMock) ranListManager := managers.NewRanListManager(log, rnibDataService) var nbIdentity *entities.NbIdentity if preAddNbIdentity { nbIdentity = &entities.NbIdentity{InventoryName: RanName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}} writerMock.On("AddNbIdentity", entities.Node_ENB, nbIdentity).Return(nil) ranListManager.AddNbIdentity(entities.Node_ENB, nbIdentity) } ranAlarmService := &mocks.RanAlarmServiceMock{} ranConnectStatusChangeManager := managers.NewRanConnectStatusChangeManager(log, rnibDataService, ranListManager, ranAlarmService) nodebValidator := managers.NewNodebValidator() updateEnbManager := managers.NewUpdateEnbManager(log, rnibDataService, nodebValidator) updateGnbManager := managers.NewUpdateGnbManager(log, rnibDataService, nodebValidator) handlerProvider := httpmsghandlerprovider.NewIncomingRequestHandlerProvider(log, rmrSender, config, rnibDataService, e2tInstancesManager, rmClient, ranConnectStatusChangeManager, nodebValidator, updateEnbManager, updateGnbManager, ranListManager) controller := NewNodebController(log, handlerProvider) return controller, readerMock, writerMock, nbIdentity } func TestShutdownHandlerRnibError(t *testing.T) { controller, _, _, _, e2tInstancesManagerMock, _ := setupControllerTest(t) e2tInstancesManagerMock.On("GetE2TAddresses").Return([]string{}, e2managererrors.NewRnibDbError()) writer := httptest.NewRecorder() controller.Shutdown(writer, tests.GetHttpRequest()) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Code) } func TestSetGeneralConfigurationHandlerRnibError(t *testing.T) { controller, readerMock, _, _, _, _ := setupControllerTest(t) configuration := &entities.GeneralConfiguration{} readerMock.On("GetGeneralConfiguration").Return(configuration, e2managererrors.NewRnibDbError()) writer := httptest.NewRecorder() httpRequest, _ := http.NewRequest("PUT", "https://localhost:3800/v1/nodeb/parameters", strings.NewReader("{\"enableRic\":false}")) controller.SetGeneralConfiguration(writer, httpRequest) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) assert.Equal(t, e2managererrors.NewRnibDbError().Code, errorResponse.Code) } func TestSetGeneralConfigurationInvalidJson(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() httpRequest, _ := http.NewRequest("PUT", "https://localhost:3800/v1/nodeb/parameters", strings.NewReader("{\"enableRic\":false, \"someValue\":false}")) controller.SetGeneralConfiguration(writer, httpRequest) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) assert.Equal(t, e2managererrors.NewInvalidJsonError().Code, errorResponse.Code) } func controllerGetNodebTestExecuter(t *testing.T, context *controllerGetNodebTestContext) { controller, readerMock, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() readerMock.On("GetNodeb", context.ranName).Return(context.nodebInfo, context.rnibError) req, _ := http.NewRequest(http.MethodGet, "/nodeb", nil) req = mux.SetURLVars(req, map[string]string{"ranName": context.ranName}) controller.GetNodeb(writer, req) assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, context.expectedJsonResponse, string(bodyBytes)) } func controllerGetNodebIdListTestExecuter(t *testing.T, context *controllerGetNodebIdListTestContext) { controller, readerMock, _, _, _, ranListManager := setupControllerTest(t) writer := httptest.NewRecorder() readerMock.On("GetListNodebIds").Return(context.nodebIdList, context.rnibError) err := ranListManager.InitNbIdentityMap() if err != nil { t.Errorf("Error cannot init identity") } req, _ := http.NewRequest(http.MethodGet, "/nodeb/states", nil) controller.GetNodebIdList(writer, req) assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Contains(t, context.expectedJsonResponse, string(bodyBytes)) } func activateControllerUpdateEnbMocks(context *controllerUpdateEnbTestContext, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock, updateEnbRequest *models.UpdateEnbRequest) { if context.getNodebInfoResult != nil { readerMock.On("GetNodeb", RanName).Return(context.getNodebInfoResult.nodebInfo, context.getNodebInfoResult.rnibError) } if context.removeServedCellsParams != nil { writerMock.On("RemoveServedCells", RanName, context.removeServedCellsParams.servedCellInfo).Return(context.removeServedCellsParams.err) } if context.updateEnbCellsParams != nil { updatedNodebInfo := *context.getNodebInfoResult.nodebInfo if context.getNodebInfoResult.nodebInfo.SetupFromNetwork { updateEnbRequest.Enb.EnbType = context.getNodebInfoResult.nodebInfo.GetEnb().EnbType } updatedNodebInfo.Configuration = &entities.NodebInfo_Enb{Enb: updateEnbRequest.Enb} writerMock.On("UpdateEnb", &updatedNodebInfo, updateEnbRequest.Enb.ServedCells).Return(context.updateEnbCellsParams.err) } } func activateControllerUpdateGnbMocks(context *controllerUpdateGnbTestContext, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock) { if context.getNodebInfoResult != nil { readerMock.On("GetNodeb", RanName).Return(context.getNodebInfoResult.nodebInfo, context.getNodebInfoResult.rnibError) } if context.removeServedNrCellsParams != nil { writerMock.On("RemoveServedNrCells", RanName, context.removeServedNrCellsParams.servedNrCells).Return(context.removeServedNrCellsParams.err) } if context.updateGnbCellsParams != nil { updatedNodebInfo := *context.getNodebInfoResult.nodebInfo gnb := entities.Gnb{} _ = jsonpb.Unmarshal(getJsonRequestAsBuffer(context.requestBody), &gnb) updatedGnb := *updatedNodebInfo.GetGnb() updatedGnb.ServedNrCells = gnb.ServedNrCells writerMock.On("UpdateGnbCells", &updatedNodebInfo, gnb.ServedNrCells).Return(context.updateGnbCellsParams.err) } } func assertControllerUpdateGnb(t *testing.T, context *controllerUpdateGnbTestContext, writer *httptest.ResponseRecorder, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock) { assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, context.expectedJsonResponse, string(bodyBytes)) readerMock.AssertExpectations(t) writerMock.AssertExpectations(t) } func assertControllerUpdateEnb(t *testing.T, context *controllerUpdateEnbTestContext, writer *httptest.ResponseRecorder, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock) { assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, context.expectedJsonResponse, string(bodyBytes)) readerMock.AssertExpectations(t) writerMock.AssertExpectations(t) } func assertControllerAddEnb(t *testing.T, context *controllerAddEnbTestContext, writer *httptest.ResponseRecorder, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock) { assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, context.expectedJsonResponse, string(bodyBytes)) readerMock.AssertExpectations(t) writerMock.AssertExpectations(t) } func assertControllerDeleteEnb(t *testing.T, context *controllerDeleteEnbTestContext, writer *httptest.ResponseRecorder, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock) { assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, context.expectedJsonResponse, string(bodyBytes)) readerMock.AssertExpectations(t) writerMock.AssertExpectations(t) } func buildUpdateEnbRequest(context *controllerUpdateEnbTestContext) *http.Request { updateEnbUrl := fmt.Sprintf("/nodeb/enb/%s", RanName) requestBody := getJsonRequestAsBuffer(context.requestBody) req, _ := http.NewRequest(http.MethodPut, updateEnbUrl, requestBody) req.Header.Set("Content-Type", "application/json") req = mux.SetURLVars(req, map[string]string{"ranName": RanName}) return req } func buildUpdateGnbRequest(context *controllerUpdateGnbTestContext) *http.Request { updateGnbUrl := fmt.Sprintf("/nodeb/gnb/%s", RanName) requestBody := getJsonRequestAsBuffer(context.requestBody) req, _ := http.NewRequest(http.MethodPut, updateGnbUrl, requestBody) req.Header.Set("Content-Type", "application/json") req = mux.SetURLVars(req, map[string]string{"ranName": RanName}) return req } func buildAddEnbRequest(context *controllerAddEnbTestContext) *http.Request { requestBody := getJsonRequestAsBuffer(context.requestBody) req, _ := http.NewRequest(http.MethodPost, AddEnbUrl, requestBody) req.Header.Set("Content-Type", "application/json") return req } func controllerUpdateEnbTestExecuter(t *testing.T, context *controllerUpdateEnbTestContext) { controller, readerMock, writerMock, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() r := buildUpdateEnbRequest(context) body, _ := ioutil.ReadAll(io.LimitReader(r.Body, LimitRequest)) updateEnbRequest := models.UpdateEnbRequest{} _ = json.Unmarshal(body, &updateEnbRequest) activateControllerUpdateEnbMocks(context, readerMock, writerMock, &updateEnbRequest) r = buildUpdateEnbRequest(context) defer r.Body.Close() controller.UpdateEnb(writer, r) assertControllerUpdateEnb(t, context, writer, readerMock, writerMock) } func controllerUpdateGnbTestExecuter(t *testing.T, context *controllerUpdateGnbTestContext) { controller, readerMock, writerMock, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() activateControllerUpdateGnbMocks(context, readerMock, writerMock) req := buildUpdateGnbRequest(context) controller.UpdateGnb(writer, req) assertControllerUpdateGnb(t, context, writer, readerMock, writerMock) } func activateControllerAddEnbMocks(context *controllerAddEnbTestContext, readerMock *mocks.RnibReaderMock, writerMock *mocks.RnibWriterMock, addEnbRequest *models.AddEnbRequest) { if context.getNodebInfoResult != nil { readerMock.On("GetNodeb", RanName).Return(context.getNodebInfoResult.nodebInfo, context.getNodebInfoResult.rnibError) } if context.addEnbParams != nil { nodebInfo := entities.NodebInfo{ RanName: addEnbRequest.RanName, Ip: addEnbRequest.Ip, Port: addEnbRequest.Port, NodeType: entities.Node_ENB, GlobalNbId: addEnbRequest.GlobalNbId, Configuration: &entities.NodebInfo_Enb{Enb: addEnbRequest.Enb}, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, } writerMock.On("AddEnb", &nodebInfo).Return(context.addEnbParams.err) } if context.addNbIdentityParams != nil { nbIdentity := entities.NbIdentity{InventoryName: addEnbRequest.RanName, GlobalNbId: addEnbRequest.GlobalNbId, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED} writerMock.On("AddNbIdentity", entities.Node_ENB, &nbIdentity).Return(context.addNbIdentityParams.err) } } func controllerAddEnbTestExecuter(t *testing.T, context *controllerAddEnbTestContext) { controller, readerMock, writerMock, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() r := buildAddEnbRequest(context) body, _ := ioutil.ReadAll(io.LimitReader(r.Body, LimitRequest)) addEnbRequest := models.AddEnbRequest{} _ = json.Unmarshal(body, &addEnbRequest) activateControllerAddEnbMocks(context, readerMock, writerMock, &addEnbRequest) r = buildAddEnbRequest(context) defer r.Body.Close() controller.AddEnb(writer, r) assertControllerAddEnb(t, context, writer, readerMock, writerMock) } func controllerDeleteEnbTestExecuter(t *testing.T, context *controllerDeleteEnbTestContext, preAddNbIdentity bool) { controller, readerMock, writerMock, nbIdentity := setupDeleteEnbControllerTest(t, preAddNbIdentity) readerMock.On("GetNodeb", RanName).Return(context.getNodebInfoResult.nodebInfo, context.getNodebInfoResult.rnibError) if context.getNodebInfoResult.rnibError == nil && context.getNodebInfoResult.nodebInfo.GetNodeType() == entities.Node_ENB && !context.getNodebInfoResult.nodebInfo.SetupFromNetwork { writerMock.On("RemoveEnb", context.getNodebInfoResult.nodebInfo).Return(nil) if preAddNbIdentity { writerMock.On("RemoveNbIdentity", entities.Node_ENB, nbIdentity).Return(nil) } } writer := httptest.NewRecorder() r, _ := http.NewRequest(http.MethodDelete, AddEnbUrl+"/"+RanName, nil) r.Header.Set("Content-Type", "application/json") r = mux.SetURLVars(r, map[string]string{"ranName": RanName}) controller.DeleteEnb(writer, r) assertControllerDeleteEnb(t, context, writer, readerMock, writerMock) } /* UpdateGnb UTs */ // BEGIN - UpdateGnb Validation Failure UTs func TestControllerUpdateGnbEmptyServedNrCells(t *testing.T) { context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbMissingServedNrCellInformation(t *testing.T) { context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": nil, }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbMissingServedNrCellRequiredProp(t *testing.T) { for _, v := range ServedNrCellInformationRequiredFields { context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(v), }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } } func TestControllerUpdateGnbMissingServedNrCellFddOrTdd(t *testing.T) { servedNrCellInformation := buildServedNrCellInformation("") servedNrCellInformation["choiceNrMode"] = map[string]interface{}{} context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": servedNrCellInformation, }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbMissingNeighbourInfoFddOrTdd(t *testing.T) { nrNeighbourInfo := buildNrNeighbourInformation("") nrNeighbourInfo["choiceNrMode"] = map[string]interface{}{} context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ nrNeighbourInfo, }, }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbMissingNrNeighbourInformationRequiredProp(t *testing.T) { for _, v := range NrNeighbourInformationRequiredFields { context := controllerUpdateGnbTestContext{ getNodebInfoResult: nil, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(v), }, }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } } // END - UpdateGnb Validation Failure UTs func TestControllerUpdateGnbValidServedNrCellInformationGetNodebNotFound(t *testing.T) { context := controllerUpdateGnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), }, }, }, expectedStatusCode: http.StatusNotFound, expectedJsonResponse: ResourceNotFoundJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbValidServedNrCellInformationGetNodebInternalError(t *testing.T) { context := controllerUpdateGnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")), }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), }, }, }, expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbGetNodebSuccessInvalidGnbConfiguration(t *testing.T) { context := controllerUpdateGnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, }, rnibError: nil, }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(""), }, }, }, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbGetNodebSuccessRemoveServedNrCellsFailure(t *testing.T) { oldServedNrCells := generateServedNrCells("whatever1", "whatever2") context := controllerUpdateGnbTestContext{ removeServedNrCellsParams: &removeServedNrCellsParams{ err: common.NewInternalError(errors.New("#writer.UpdateGnbCells - Internal Error")), servedNrCells: oldServedNrCells, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, Configuration: &entities.NodebInfo_Gnb{Gnb: &entities.Gnb{ServedNrCells: oldServedNrCells}}, NodeType: entities.Node_GNB, }, rnibError: nil, }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(""), }, }, }, }, expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbGetNodebSuccessUpdateGnbCellsFailure(t *testing.T) { oldServedNrCells := generateServedNrCells("whatever1", "whatever2") context := controllerUpdateGnbTestContext{ removeServedNrCellsParams: &removeServedNrCellsParams{ err: nil, servedNrCells: oldServedNrCells, }, updateGnbCellsParams: &updateGnbCellsParams{ err: common.NewInternalError(errors.New("#writer.UpdateGnbCells - Internal Error")), }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, Configuration: &entities.NodebInfo_Gnb{Gnb: &entities.Gnb{ServedNrCells: oldServedNrCells}}, NodeType: entities.Node_GNB, }, rnibError: nil, }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(""), }, }, }, }, expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbExistingEmptyCellsSuccess(t *testing.T) { oldServedNrCells := []*entities.ServedNRCell{} context := controllerUpdateGnbTestContext{ updateGnbCellsParams: &updateGnbCellsParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, Configuration: &entities.NodebInfo_Gnb{Gnb: &entities.Gnb{ServedNrCells: oldServedNrCells}}, NodeType: entities.Node_GNB, }, rnibError: nil, }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(""), }, }, }, }, expectedStatusCode: http.StatusOK, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"nodeType\":\"GNB\",\"gnb\":{\"servedNrCells\":[{\"servedNrCellInformation\":{\"nrPci\":1,\"cellId\":\"whatever\",\"servedPlmns\":[\"whatever\"],\"nrMode\":\"FDD\",\"choiceNrMode\":{\"fdd\":{}}},\"nrNeighbourInfos\":[{\"nrPci\":1,\"nrCgi\":\"whatever\",\"nrMode\":\"FDD\",\"choiceNrMode\":{\"tdd\":{}}}]}]},\"associatedE2tInstanceAddress\":\"10.0.2.15:38000\"}", } controllerUpdateGnbTestExecuter(t, &context) } func TestControllerUpdateGnbSuccess(t *testing.T) { oldServedNrCells := generateServedNrCells("whatever1", "whatever2") context := controllerUpdateGnbTestContext{ removeServedNrCellsParams: &removeServedNrCellsParams{ err: nil, servedNrCells: oldServedNrCells, }, updateGnbCellsParams: &updateGnbCellsParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, Configuration: &entities.NodebInfo_Gnb{Gnb: &entities.Gnb{ServedNrCells: oldServedNrCells}}, NodeType: entities.Node_GNB, }, rnibError: nil, }, requestBody: map[string]interface{}{ "servedNrCells": []interface{}{ map[string]interface{}{ "servedNrCellInformation": buildServedNrCellInformation(""), "nrNeighbourInfos": []interface{}{ buildNrNeighbourInformation(""), }, }, }, }, expectedStatusCode: http.StatusOK, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"nodeType\":\"GNB\",\"gnb\":{\"servedNrCells\":[{\"servedNrCellInformation\":{\"nrPci\":1,\"cellId\":\"whatever\",\"servedPlmns\":[\"whatever\"],\"nrMode\":\"FDD\",\"choiceNrMode\":{\"fdd\":{}}},\"nrNeighbourInfos\":[{\"nrPci\":1,\"nrCgi\":\"whatever\",\"nrMode\":\"FDD\",\"choiceNrMode\":{\"tdd\":{}}}]}]},\"associatedE2tInstanceAddress\":\"10.0.2.15:38000\"}", } controllerUpdateGnbTestExecuter(t, &context) } /* UpdateEnb UTs */ func TestControllerUpdateEnbInvalidRequest(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() invalidJson := strings.NewReader("{enb:\"whatever\"") updateEnbUrl := fmt.Sprintf("/nodeb/enb/%s", RanName) req, _ := http.NewRequest(http.MethodPut, updateEnbUrl, invalidJson) req.Header.Set("Content-Type", "application/json") req = mux.SetURLVars(req, map[string]string{"ranName": RanName}) controller.UpdateEnb(writer, req) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, CorruptedJson, string(bodyBytes)) } func TestControllerUpdateEnbEmptyEnbType(t *testing.T) { context := controllerUpdateEnbTestContext{ getNodebInfoResult: nil, requestBody: getUpdateEnbRequest(EnbRequiredFields[0]), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbEmptyServedCells(t *testing.T) { context := controllerUpdateEnbTestContext{ getNodebInfoResult: nil, requestBody: getUpdateEnbRequest(EnbRequiredFields[1]), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbMissingEnb(t *testing.T) { context := controllerUpdateEnbTestContext{ getNodebInfoResult: nil, requestBody: getUpdateEnbRequest(UpdateEnbRequestRequiredFields[0]), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbMissingRequiredServedCellProps(t *testing.T) { r := getUpdateEnbRequest("") for _, v := range ServedCellRequiredFields { enb := r["enb"] enbMap, _ := enb.(map[string]interface{}) enbMap["servedCells"] = []interface{}{ buildServedCell(v), } context := controllerUpdateEnbTestContext{ requestBody: r, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } } func TestControllerUpdateEnbValidServedCellsGetNodebNotFound(t *testing.T) { context := controllerUpdateEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusNotFound, expectedJsonResponse: ResourceNotFoundJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbValidServedCellsGetNodebInternalError(t *testing.T) { context := controllerUpdateEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")), }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbGetNodebSuccessGnbTypeFailure(t *testing.T) { oldServedCells := generateServedCells("whatever1", "whatever2") context := controllerUpdateEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_GNB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells}}, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbGetNodebSuccessRemoveServedCellsFailure(t *testing.T) { oldServedCells := generateServedCells("whatever1", "whatever2") context := controllerUpdateEnbTestContext{ removeServedCellsParams: &removeServedCellsParams{ err: common.NewInternalError(errors.New("#writer.RemoveServedCells - Internal Error")), servedCellInfo: oldServedCells, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells}}, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbGetNodebSuccessUpdateEnbFailure(t *testing.T) { oldServedCells := generateServedCells("whatever1", "whatever2") context := controllerUpdateEnbTestContext{ removeServedCellsParams: &removeServedCellsParams{ err: nil, servedCellInfo: oldServedCells, }, updateEnbCellsParams: &updateEnbCellsParams{ err: common.NewInternalError(errors.New("#writer.UpdateEnb - Internal Error")), }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells, EnbType: entities.EnbType_MACRO_ENB}}, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbExistingEmptyCellsSuccess(t *testing.T) { oldServedCells := []*entities.ServedCellInfo{} context := controllerUpdateEnbTestContext{ updateEnbCellsParams: &updateEnbCellsParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells, EnbType: entities.EnbType_MACRO_ENB}}, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusOK, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"nodeType\":\"ENB\",\"enb\":{\"enbType\":\"MACRO_ENB\",\"servedCells\":[{\"pci\":1,\"cellId\":\"whatever\",\"tac\":\"whatever3\",\"broadcastPlmns\":[\"whatever\"],\"choiceEutraMode\":{\"fdd\":{}},\"eutraMode\":\"FDD\"}]},\"associatedE2tInstanceAddress\":\"10.0.2.15:38000\"}", } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbNgEnbFailure(t *testing.T) { requestBody := map[string]interface{}{ "enb": map[string]interface{}{ "enbType": 5, "servedCells": []interface{}{ buildServedCell(""), }}, } context := controllerUpdateEnbTestContext{ getNodebInfoResult: nil, requestBody: requestBody, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbSuccessSetupFromNwFalse(t *testing.T) { oldServedCells := generateServedCells("whatever1", "whatever2") context := controllerUpdateEnbTestContext{ removeServedCellsParams: &removeServedCellsParams{ err: nil, servedCellInfo: oldServedCells, }, updateEnbCellsParams: &updateEnbCellsParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells, EnbType: entities.EnbType_LONG_MACRO_ENB}}, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusOK, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"nodeType\":\"ENB\",\"enb\":{\"enbType\":\"MACRO_ENB\",\"servedCells\":[{\"pci\":1,\"cellId\":\"whatever\",\"tac\":\"whatever3\",\"broadcastPlmns\":[\"whatever\"],\"choiceEutraMode\":{\"fdd\":{}},\"eutraMode\":\"FDD\"}]},\"associatedE2tInstanceAddress\":\"10.0.2.15:38000\"}", } controllerUpdateEnbTestExecuter(t, &context) } func TestControllerUpdateEnbSuccessSetupFromNwTrue(t *testing.T) { oldServedCells := generateServedCells("whatever1", "whatever2") context := controllerUpdateEnbTestContext{ removeServedCellsParams: &removeServedCellsParams{ err: nil, servedCellInfo: oldServedCells, }, updateEnbCellsParams: &updateEnbCellsParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{ RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: AssociatedE2TInstanceAddress, NodeType: entities.Node_ENB, Configuration: &entities.NodebInfo_Enb{Enb: &entities.Enb{ServedCells: oldServedCells, EnbType: entities.EnbType_LONG_MACRO_ENB}}, SetupFromNetwork: true, }, rnibError: nil, }, requestBody: getUpdateEnbRequest(""), expectedStatusCode: http.StatusOK, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"nodeType\":\"ENB\",\"enb\":{\"enbType\":\"LONG_MACRO_ENB\",\"servedCells\":[{\"pci\":1,\"cellId\":\"whatever\",\"tac\":\"whatever3\",\"broadcastPlmns\":[\"whatever\"],\"choiceEutraMode\":{\"fdd\":{}},\"eutraMode\":\"FDD\"}]},\"associatedE2tInstanceAddress\":\"10.0.2.15:38000\",\"setupFromNetwork\":true}", } controllerUpdateEnbTestExecuter(t, &context) } /* AddEnb UTs */ func TestControllerAddEnbGetNodebInternalError(t *testing.T) { context := controllerAddEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")), }, requestBody: getAddEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerAddEnbTestExecuter(t, &context) } func TestControllerAddEnbNodebExistsFailure(t *testing.T) { context := controllerAddEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{}, rnibError: nil, }, requestBody: getAddEnbRequest(""), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: NodebExistsJson, } controllerAddEnbTestExecuter(t, &context) } func TestControllerAddEnbSaveNodebFailure(t *testing.T) { context := controllerAddEnbTestContext{ addEnbParams: &addEnbParams{ err: common.NewInternalError(errors.New("#reader.AddEnb - Internal Error")), }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), }, requestBody: getAddEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerAddEnbTestExecuter(t, &context) } func TestControllerAddEnbAddNbIdentityFailure(t *testing.T) { context := controllerAddEnbTestContext{ addEnbParams: &addEnbParams{ err: nil, }, addNbIdentityParams: &addNbIdentityParams{ err: common.NewInternalError(errors.New("#writer.addNbIdentity - Internal Error")), }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), }, requestBody: getAddEnbRequest(""), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerAddEnbTestExecuter(t, &context) } func TestControllerAddEnbMissingRequiredRequestProps(t *testing.T) { for _, v := range AddEnbRequestRequiredFields { context := controllerAddEnbTestContext{ requestBody: getAddEnbRequest(v), expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerAddEnbTestExecuter(t, &context) } } func TestControllerAddEnbInvalidRequest(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() // Invalid json: attribute name without quotes (should be "cause":). invalidJson := strings.NewReader("{ranName:\"whatever\"") req, _ := http.NewRequest(http.MethodPost, AddEnbUrl, invalidJson) controller.AddEnb(writer, req) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) bodyBytes, _ := ioutil.ReadAll(writer.Body) assert.Equal(t, CorruptedJson, string(bodyBytes)) } func TestControllerAddEnbMissingRequiredGlobalNbIdProps(t *testing.T) { r := getAddEnbRequest("") for _, v := range GlobalIdRequiredFields { r["globalNbId"] = buildGlobalNbId(v) context := controllerAddEnbTestContext{ requestBody: r, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerAddEnbTestExecuter(t, &context) } } func TestControllerAddEnbMissingRequiredEnbProps(t *testing.T) { r := getAddEnbRequest("") for _, v := range EnbRequiredFields { r["enb"] = buildEnb(v) context := controllerAddEnbTestContext{ requestBody: r, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerAddEnbTestExecuter(t, &context) } } func TestControllerAddEnbMissingRequiredServedCellProps(t *testing.T) { r := getAddEnbRequest("") for _, v := range ServedCellRequiredFields { enb := r["enb"] enbMap, _ := enb.(map[string]interface{}) enbMap["servedCells"] = []interface{}{ buildServedCell(v), } context := controllerAddEnbTestContext{ requestBody: r, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerAddEnbTestExecuter(t, &context) } } func TestControllerAddEnbNgEnbFailure(t *testing.T) { requestBody := map[string]interface{}{ "ranName": RanName, "globalNbId": buildGlobalNbId(""), "enb": map[string]interface{}{ "enbType": 5, "servedCells": []interface{}{ buildServedCell(""), }, }, } context := controllerAddEnbTestContext{ requestBody: requestBody, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerAddEnbTestExecuter(t, &context) } func TestControllerAddEnbSuccess(t *testing.T) { context := controllerAddEnbTestContext{ addEnbParams: &addEnbParams{ err: nil, }, addNbIdentityParams: &addNbIdentityParams{ err: nil, }, getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), }, requestBody: getAddEnbRequest(""), expectedStatusCode: http.StatusCreated, expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"DISCONNECTED\",\"globalNbId\":{\"plmnId\":\"whatever\",\"nbId\":\"whatever2\"},\"nodeType\":\"ENB\",\"enb\":{\"enbType\":\"MACRO_ENB\",\"servedCells\":[{\"pci\":1,\"cellId\":\"whatever\",\"tac\":\"whatever3\",\"broadcastPlmns\":[\"whatever\"],\"choiceEutraMode\":{\"fdd\":{}},\"eutraMode\":\"FDD\"}]}}", } controllerAddEnbTestExecuter(t, &context) } func TestControllerDeleteEnbGetNodebInternalError(t *testing.T) { context := controllerDeleteEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")), }, expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerDeleteEnbTestExecuter(t, &context, false) } /* DeleteEnb UTs */ func TestControllerDeleteEnbNodebNotExistsFailure(t *testing.T) { context := controllerDeleteEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: nil, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found"), }, expectedStatusCode: http.StatusNotFound, expectedJsonResponse: ResourceNotFoundJson, } controllerDeleteEnbTestExecuter(t, &context, false) } func TestControllerDeleteEnbNodebNotEnb(t *testing.T) { context := controllerDeleteEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{RanName: "ran1", NodeType: entities.Node_GNB, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}, rnibError: nil, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerDeleteEnbTestExecuter(t, &context, false) } func TestControllerDeleteEnbSetupFromNetworkTrueFailure(t *testing.T) { context := controllerDeleteEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{RanName: RanName, NodeType: entities.Node_ENB, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, SetupFromNetwork: true}, rnibError: nil, }, expectedStatusCode: http.StatusBadRequest, expectedJsonResponse: ValidationFailureJson, } controllerDeleteEnbTestExecuter(t, &context, true) } func TestControllerDeleteEnbSuccess(t *testing.T) { context := controllerDeleteEnbTestContext{ getNodebInfoResult: &getNodebInfoResult{ nodebInfo: &entities.NodebInfo{RanName: RanName, NodeType: entities.Node_ENB, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}, rnibError: nil, }, expectedStatusCode: http.StatusNoContent, expectedJsonResponse: "", } controllerDeleteEnbTestExecuter(t, &context, true) } func getJsonRequestAsBuffer(requestJson map[string]interface{}) *bytes.Buffer { b := new(bytes.Buffer) _ = json.NewEncoder(b).Encode(requestJson) return b } /* GetNodeb UTs */ func TestControllerGetNodebSuccess(t *testing.T) { ranName := "test" var rnibError error context := controllerGetNodebTestContext{ ranName: ranName, nodebInfo: &entities.NodebInfo{RanName: ranName, Ip: "10.0.2.15", Port: 1234}, rnibError: rnibError, expectedStatusCode: http.StatusOK, expectedJsonResponse: fmt.Sprintf("{\"ranName\":\"%s\",\"ip\":\"10.0.2.15\",\"port\":1234}", ranName), } controllerGetNodebTestExecuter(t, &context) } func TestControllerGetNodebNotFound(t *testing.T) { ranName := "test" var nodebInfo *entities.NodebInfo context := controllerGetNodebTestContext{ ranName: ranName, nodebInfo: nodebInfo, rnibError: common.NewResourceNotFoundError("#reader.GetNodeb - Not found Error"), expectedStatusCode: http.StatusNotFound, expectedJsonResponse: ResourceNotFoundJson, } controllerGetNodebTestExecuter(t, &context) } func TestControllerGetNodebInternal(t *testing.T) { ranName := "test" var nodebInfo *entities.NodebInfo context := controllerGetNodebTestContext{ ranName: ranName, nodebInfo: nodebInfo, rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")), expectedStatusCode: http.StatusInternalServerError, expectedJsonResponse: RnibErrorJson, } controllerGetNodebTestExecuter(t, &context) } func TestControllerGetNodebIdListSuccess(t *testing.T) { var rnibError error nodebIdList := []*entities.NbIdentity{ {InventoryName: "test1", GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}, {InventoryName: "test2", GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId2", NbId: "nbId2"}}, } context := controllerGetNodebIdListTestContext{ nodebIdList: nodebIdList, rnibError: rnibError, expectedStatusCode: http.StatusOK, expectedJsonResponse: "[{\"inventoryName\":\"test1\",\"globalNbId\":{\"plmnId\":\"plmnId1\",\"nbId\":\"nbId1\"}},{\"inventoryName\":\"test2\",\"globalNbId\":{\"plmnId\":\"plmnId2\",\"nbId\":\"nbId2\"}}][{\"inventoryName\":\"test2\",\"globalNbId\":{\"plmnId\":\"plmnId2\",\"nbId\":\"nbId2\"}},{\"inventoryName\":\"test1\",\"globalNbId\":{\"plmnId\":\"plmnId1\",\"nbId\":\"nbId1\"}}]", } controllerGetNodebIdListTestExecuter(t, &context) } func TestControllerGetNodebIdListEmptySuccess(t *testing.T) { var rnibError error var nodebIdList []*entities.NbIdentity context := controllerGetNodebIdListTestContext{ nodebIdList: nodebIdList, rnibError: rnibError, expectedStatusCode: http.StatusOK, expectedJsonResponse: "[]", } controllerGetNodebIdListTestExecuter(t, &context) } func TestHeaderValidationFailed(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() header := &http.Header{} controller.handleRequest(writer, header, httpmsghandlerprovider.ShutdownRequest, nil, true, 0) var errorResponse = parseJsonRequest(t, writer.Body) err := e2managererrors.NewHeaderValidationError() assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, err.Code) assert.Equal(t, errorResponse.Message, err.Message) } func TestShutdownStatusNoContent(t *testing.T) { controller, readerMock, _, _, e2tInstancesManagerMock, _ := setupControllerTest(t) e2tInstancesManagerMock.On("GetE2TAddresses").Return([]string{}, nil) readerMock.On("GetListNodebIds").Return([]*entities.NbIdentity{}, nil) writer := httptest.NewRecorder() controller.Shutdown(writer, tests.GetHttpRequest()) assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode) } func TestHandleInternalError(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() err := e2managererrors.NewInternalError() controller.handleErrorResponse(err, writer) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, err.Code) assert.Equal(t, errorResponse.Message, err.Message) } func TestHandleCommandAlreadyInProgressError(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() err := e2managererrors.NewCommandAlreadyInProgressError() controller.handleErrorResponse(err, writer) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, err.Code) assert.Equal(t, errorResponse.Message, err.Message) } func TestHandleRoutingManagerError(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() err := e2managererrors.NewRoutingManagerError() controller.handleErrorResponse(err, writer) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusServiceUnavailable, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, err.Code) assert.Equal(t, errorResponse.Message, err.Message) } func TestHandleE2TInstanceAbsenceError(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() err := e2managererrors.NewE2TInstanceAbsenceError() controller.handleErrorResponse(err, writer) var errorResponse = parseJsonRequest(t, writer.Body) assert.Equal(t, http.StatusServiceUnavailable, writer.Result().StatusCode) assert.Equal(t, errorResponse.Code, err.Code) assert.Equal(t, errorResponse.Message, err.Message) } func TestValidateHeaders(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) header := http.Header{} header.Set("Content-Type", "application/json") result := controller.validateRequestHeader(&header) assert.Nil(t, result) } func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse { var errorResponse models.ErrorResponse body, err := ioutil.ReadAll(r) if err != nil { t.Errorf("Error cannot deserialize json request") } _ = json.Unmarshal(body, &errorResponse) return errorResponse } func initLog(t *testing.T) *logger.Logger { log, err := logger.InitLogger(logger.InfoLevel) if err != nil { t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err) } return log } func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T) { controller, readerMock, _, rmrMessengerMock, _, _ := setupControllerTest(t) ranName := "test1" payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40} var xAction []byte var msgSrc unsafe.Pointer msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xAction, msgSrc) rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil) writer := httptest.NewRecorder() var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED} readerMock.On("GetNodeb", ranName).Return(nodeb, nil) data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"} b := new(bytes.Buffer) _ = json.NewEncoder(b).Encode(data4Req) req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b) req = mux.SetURLVars(req, map[string]string{"ranName": ranName}) controller.X2Reset(writer, req) assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode) } func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T) { controller, readerMock, _, rmrMessengerMock, _, _ := setupControllerTest(t) ranName := "test1" // o&m intervention payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64} var xAction []byte var msgSrc unsafe.Pointer msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xAction, msgSrc) rmrMessengerMock.On("SendMsg", msg, true).Return(msg, nil) writer := httptest.NewRecorder() var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED} readerMock.On("GetNodeb", ranName).Return(nodeb, nil) // no body b := new(bytes.Buffer) data4Req := map[string]interface{}{} _ = json.NewEncoder(b).Encode(data4Req) req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b) req = mux.SetURLVars(req, map[string]string{"ranName": ranName}) controller.X2Reset(writer, req) assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode) } func TestX2ResetHandleFailureInvalidBody(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) ranName := "test1" writer := httptest.NewRecorder() // Invalid json: attribute name without quotes (should be "cause":). b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"") req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b) req = mux.SetURLVars(req, map[string]string{"ranName": ranName}) controller.X2Reset(writer, req) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) } func TestHandleErrorResponse(t *testing.T) { controller, _, _, _, _, _ := setupControllerTest(t) writer := httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewRnibDbError(), writer) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(), writer) assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(), writer) assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewWrongStateError("", ""), writer) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewRequestValidationError(), writer) assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewRmrError(), writer) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(), writer) assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode) writer = httptest.NewRecorder() controller.handleErrorResponse(fmt.Errorf("ErrorError"), writer) assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode) } func getRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender { rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock) rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger) return rmrsender.NewRmrSender(log, rmrMessenger) }