First commit
This commit is contained in:
Leonardo Bonati
115
setup/e2mgr/E2Manager/controllers/e2t_controller.go
Normal file
115
setup/e2mgr/E2Manager/controllers/e2t_controller.go
Normal file
@@ -0,0 +1,115 @@
|
||||
//
|
||||
// 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 (
|
||||
"e2mgr/e2managererrors"
|
||||
"e2mgr/logger"
|
||||
"e2mgr/models"
|
||||
"e2mgr/providers/httpmsghandlerprovider"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httputil"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type IE2TController interface {
|
||||
GetE2TInstances(writer http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
type E2TController struct {
|
||||
logger *logger.Logger
|
||||
handlerProvider *httpmsghandlerprovider.IncomingRequestHandlerProvider
|
||||
}
|
||||
|
||||
func NewE2TController(logger *logger.Logger, handlerProvider *httpmsghandlerprovider.IncomingRequestHandlerProvider) *E2TController {
|
||||
return &E2TController{
|
||||
logger: logger,
|
||||
handlerProvider: handlerProvider,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *E2TController) GetE2TInstances(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #E2TController.GetE2TInstances - request: %v", c.prettifyRequest(r))
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.GetE2TInstancesRequest, nil, false)
|
||||
}
|
||||
|
||||
func (c *E2TController) handleRequest(writer http.ResponseWriter, header *http.Header, requestName httpmsghandlerprovider.IncomingRequest, request models.Request, validateHeader bool) {
|
||||
|
||||
handler, err := c.handlerProvider.GetHandler(requestName)
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
response, err := handler.Handle(request)
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
result, err := response.Marshal()
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
c.logger.Infof("[E2 Manager -> Client] #E2TController.handleRequest - response: %s", result)
|
||||
writer.Header().Set("Content-Type", "application/json")
|
||||
writer.Write(result)
|
||||
}
|
||||
|
||||
func (c *E2TController) handleErrorResponse(err error, writer http.ResponseWriter) {
|
||||
|
||||
var errorResponseDetails models.ErrorResponse
|
||||
var httpError int
|
||||
|
||||
if err != nil {
|
||||
switch err.(type) {
|
||||
case *e2managererrors.RnibDbError:
|
||||
e2Error, _ := err.(*e2managererrors.RnibDbError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusInternalServerError
|
||||
default:
|
||||
e2Error := e2managererrors.NewInternalError()
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusInternalServerError
|
||||
}
|
||||
}
|
||||
|
||||
errorResponse, _ := json.Marshal(errorResponseDetails)
|
||||
|
||||
c.logger.Errorf("[E2 Manager -> Client] #E2TController.handleErrorResponse - http status: %d, error response: %+v", httpError, errorResponseDetails)
|
||||
|
||||
writer.Header().Set("Content-Type", "application/json")
|
||||
writer.WriteHeader(httpError)
|
||||
_, err = writer.Write(errorResponse)
|
||||
}
|
||||
|
||||
func (c *E2TController) prettifyRequest(request *http.Request) string {
|
||||
dump, _ := httputil.DumpRequest(request, true)
|
||||
requestPrettyPrint := strings.Replace(string(dump), "\r\n", " ", -1)
|
||||
return strings.Replace(requestPrettyPrint, "\n", "", -1)
|
||||
}
|
||||
137
setup/e2mgr/E2Manager/controllers/e2t_controller_test.go
Normal file
137
setup/e2mgr/E2Manager/controllers/e2t_controller_test.go
Normal file
@@ -0,0 +1,137 @@
|
||||
//
|
||||
// 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.
|
||||
//
|
||||
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"e2mgr/configuration"
|
||||
"e2mgr/managers"
|
||||
"e2mgr/mocks"
|
||||
"e2mgr/models"
|
||||
"e2mgr/providers/httpmsghandlerprovider"
|
||||
"e2mgr/services"
|
||||
"encoding/json"
|
||||
"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/magiconair/properties/assert"
|
||||
"github.com/pkg/errors"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
const E2TAddress string = "10.0.2.15:38000"
|
||||
const E2TAddress2 string = "10.0.2.16:38001"
|
||||
|
||||
type controllerE2TInstancesTestContext struct {
|
||||
e2tAddresses []string
|
||||
e2tInstances []*entities.E2TInstance
|
||||
error error
|
||||
expectedStatusCode int
|
||||
expectedJsonResponse string
|
||||
}
|
||||
|
||||
func setupE2TControllerTest(t *testing.T) (*E2TController, *mocks.RnibReaderMock) {
|
||||
log := initLog(t)
|
||||
config := configuration.ParseConfiguration()
|
||||
|
||||
readerMock := &mocks.RnibReaderMock{}
|
||||
|
||||
rnibDataService := services.NewRnibDataService(log, config, readerMock, nil)
|
||||
e2tInstancesManager := managers.NewE2TInstancesManager(rnibDataService, log)
|
||||
handlerProvider := httpmsghandlerprovider.NewIncomingRequestHandlerProvider(log, nil, config, rnibDataService, nil, e2tInstancesManager, &managers.E2TAssociationManager{}, nil)
|
||||
controller := NewE2TController(log, handlerProvider)
|
||||
return controller, readerMock
|
||||
}
|
||||
|
||||
func controllerGetE2TInstancesTestExecuter(t *testing.T, context *controllerE2TInstancesTestContext) {
|
||||
controller, readerMock := setupE2TControllerTest(t)
|
||||
writer := httptest.NewRecorder()
|
||||
readerMock.On("GetE2TAddresses").Return(context.e2tAddresses, context.error)
|
||||
|
||||
if context.e2tInstances != nil {
|
||||
readerMock.On("GetE2TInstances", context.e2tAddresses).Return(context.e2tInstances, context.error)
|
||||
}
|
||||
|
||||
req, _ := http.NewRequest("GET", "/e2t/list", nil)
|
||||
controller.GetE2TInstances(writer, req)
|
||||
assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode)
|
||||
bodyBytes, _ := ioutil.ReadAll(writer.Body)
|
||||
assert.Equal(t, context.expectedJsonResponse, string(bodyBytes))
|
||||
readerMock.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestControllerGetE2TInstancesSuccess(t *testing.T) {
|
||||
ranNames1 := []string{"test1", "test2", "test3"}
|
||||
e2tInstanceResponseModel1 := models.NewE2TInstanceResponseModel(E2TAddress, ranNames1)
|
||||
e2tInstanceResponseModel2 := models.NewE2TInstanceResponseModel(E2TAddress2, []string{})
|
||||
e2tInstancesResponse := models.E2TInstancesResponse{e2tInstanceResponseModel1, e2tInstanceResponseModel2}
|
||||
bytes, _ := json.Marshal(e2tInstancesResponse)
|
||||
|
||||
context := controllerE2TInstancesTestContext{
|
||||
e2tAddresses: []string{E2TAddress, E2TAddress2},
|
||||
e2tInstances: []*entities.E2TInstance{{Address: E2TAddress, AssociatedRanList: ranNames1}, {Address: E2TAddress2, AssociatedRanList: []string{}}},
|
||||
error: nil,
|
||||
expectedStatusCode: http.StatusOK,
|
||||
expectedJsonResponse: string(bytes),
|
||||
}
|
||||
|
||||
controllerGetE2TInstancesTestExecuter(t, &context)
|
||||
}
|
||||
|
||||
func TestControllerGetE2TInstancesEmptySuccess(t *testing.T) {
|
||||
e2tInstancesResponse := models.E2TInstancesResponse{}
|
||||
bytes, _ := json.Marshal(e2tInstancesResponse)
|
||||
|
||||
context := controllerE2TInstancesTestContext{
|
||||
e2tAddresses: []string{},
|
||||
e2tInstances: nil,
|
||||
error: nil,
|
||||
expectedStatusCode: http.StatusOK,
|
||||
expectedJsonResponse: string(bytes),
|
||||
}
|
||||
|
||||
controllerGetE2TInstancesTestExecuter(t, &context)
|
||||
}
|
||||
|
||||
func TestControllerGetE2TInstancesInternal(t *testing.T) {
|
||||
context := controllerE2TInstancesTestContext{
|
||||
e2tAddresses: nil,
|
||||
e2tInstances: nil,
|
||||
error: common.NewInternalError(errors.New("error")),
|
||||
expectedStatusCode: http.StatusInternalServerError,
|
||||
expectedJsonResponse: "{\"errorCode\":500,\"errorMessage\":\"RNIB error\"}",
|
||||
}
|
||||
|
||||
controllerGetE2TInstancesTestExecuter(t, &context)
|
||||
}
|
||||
|
||||
func TestInvalidRequestName(t *testing.T) {
|
||||
controller, _ := setupE2TControllerTest(t)
|
||||
|
||||
writer := httptest.NewRecorder()
|
||||
|
||||
header := &http.Header{}
|
||||
|
||||
controller.handleRequest(writer, header, "", nil, true)
|
||||
|
||||
var errorResponse = parseJsonRequest(t, writer.Body)
|
||||
|
||||
assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
|
||||
assert.Equal(t, errorResponse.Code, 501)
|
||||
}
|
||||
293
setup/e2mgr/E2Manager/controllers/nodeb_controller.go
Normal file
293
setup/e2mgr/E2Manager/controllers/nodeb_controller.go
Normal file
@@ -0,0 +1,293 @@
|
||||
//
|
||||
// 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 (
|
||||
"e2mgr/e2managererrors"
|
||||
"e2mgr/logger"
|
||||
"e2mgr/models"
|
||||
"e2mgr/providers/httpmsghandlerprovider"
|
||||
"encoding/json"
|
||||
"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
|
||||
"github.com/golang/protobuf/jsonpb"
|
||||
"github.com/golang/protobuf/proto"
|
||||
"github.com/gorilla/mux"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httputil"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
ParamRanName = "ranName"
|
||||
LimitRequest = 2000
|
||||
)
|
||||
|
||||
type INodebController interface {
|
||||
Shutdown(writer http.ResponseWriter, r *http.Request)
|
||||
X2Reset(writer http.ResponseWriter, r *http.Request)
|
||||
X2Setup(writer http.ResponseWriter, r *http.Request)
|
||||
EndcSetup(writer http.ResponseWriter, r *http.Request)
|
||||
GetNodeb(writer http.ResponseWriter, r *http.Request)
|
||||
UpdateGnb(writer http.ResponseWriter, r *http.Request)
|
||||
GetNodebIdList(writer http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
type NodebController struct {
|
||||
logger *logger.Logger
|
||||
handlerProvider *httpmsghandlerprovider.IncomingRequestHandlerProvider
|
||||
}
|
||||
|
||||
func NewNodebController(logger *logger.Logger, handlerProvider *httpmsghandlerprovider.IncomingRequestHandlerProvider) *NodebController {
|
||||
return &NodebController{
|
||||
logger: logger,
|
||||
handlerProvider: handlerProvider,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *NodebController) GetNodebIdList(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.GetNodebIdList - request: %v", c.prettifyRequest(r))
|
||||
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.GetNodebIdListRequest, nil, false)
|
||||
}
|
||||
|
||||
func (c *NodebController) GetNodeb(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.GetNodeb - request: %v", c.prettifyRequest(r))
|
||||
vars := mux.Vars(r)
|
||||
ranName := vars["ranName"]
|
||||
request := models.GetNodebRequest{RanName: ranName}
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.GetNodebRequest, request, false)
|
||||
}
|
||||
|
||||
func (c *NodebController) UpdateGnb(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.UpdateGnb - request: %v", c.prettifyRequest(r))
|
||||
vars := mux.Vars(r)
|
||||
ranName := vars[ParamRanName]
|
||||
|
||||
request := models.UpdateGnbRequest{}
|
||||
|
||||
gnb := entities.Gnb{}
|
||||
|
||||
if !c.extractRequestBodyToProto(r, &gnb, writer) {
|
||||
return
|
||||
}
|
||||
|
||||
request.Gnb = &gnb;
|
||||
request.RanName = ranName
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.UpdateGnbRequest, request, true)
|
||||
}
|
||||
|
||||
func (c *NodebController) Shutdown(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.Shutdown - request: %v", c.prettifyRequest(r))
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.ShutdownRequest, nil, false)
|
||||
}
|
||||
|
||||
func (c *NodebController) X2Reset(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.X2Reset - request: %v", c.prettifyRequest(r))
|
||||
request := models.ResetRequest{}
|
||||
vars := mux.Vars(r)
|
||||
ranName := vars[ParamRanName]
|
||||
|
||||
if r.ContentLength > 0 && !c.extractJsonBody(r, &request, writer) {
|
||||
return
|
||||
}
|
||||
request.RanName = ranName
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.ResetRequest, request, false)
|
||||
}
|
||||
|
||||
func (c *NodebController) X2Setup(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.X2Setup - request: %v", c.prettifyRequest(r))
|
||||
|
||||
request := models.SetupRequest{}
|
||||
|
||||
if !c.extractJsonBody(r, &request, writer) {
|
||||
return
|
||||
}
|
||||
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.X2SetupRequest, request, true)
|
||||
}
|
||||
|
||||
func (c *NodebController) EndcSetup(writer http.ResponseWriter, r *http.Request) {
|
||||
c.logger.Infof("[Client -> E2 Manager] #NodebController.EndcSetup - request: %v", c.prettifyRequest(r))
|
||||
|
||||
request := models.SetupRequest{}
|
||||
|
||||
if !c.extractJsonBody(r, &request, writer) {
|
||||
return
|
||||
}
|
||||
|
||||
c.handleRequest(writer, &r.Header, httpmsghandlerprovider.EndcSetupRequest, request, true)
|
||||
}
|
||||
|
||||
func (c *NodebController) extractRequestBodyToProto(r *http.Request, pb proto.Message , writer http.ResponseWriter) bool {
|
||||
defer r.Body.Close()
|
||||
|
||||
err := jsonpb.Unmarshal(r.Body, pb)
|
||||
|
||||
if err != nil {
|
||||
c.logger.Errorf("[Client -> E2 Manager] #NodebController.extractJsonBody - unable to extract json body - error: %s", err)
|
||||
c.handleErrorResponse(e2managererrors.NewInvalidJsonError(), writer)
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (c *NodebController) extractJsonBody(r *http.Request, request models.Request, writer http.ResponseWriter) bool {
|
||||
defer r.Body.Close()
|
||||
body, err := ioutil.ReadAll(io.LimitReader(r.Body, LimitRequest))
|
||||
|
||||
if err != nil {
|
||||
c.logger.Errorf("[Client -> E2 Manager] #NodebController.extractJsonBody - unable to extract json body - error: %s", err)
|
||||
c.handleErrorResponse(e2managererrors.NewInvalidJsonError(), writer)
|
||||
return false
|
||||
}
|
||||
|
||||
err = json.Unmarshal(body, &request)
|
||||
if err != nil {
|
||||
c.logger.Errorf("[Client -> E2 Manager] #NodebController.extractJsonBody - unable to extract json body - error: %s", err)
|
||||
c.handleErrorResponse(e2managererrors.NewInvalidJsonError(), writer)
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (c *NodebController) handleRequest(writer http.ResponseWriter, header *http.Header, requestName httpmsghandlerprovider.IncomingRequest, request models.Request, validateRequestHeaders bool) {
|
||||
|
||||
if validateRequestHeaders {
|
||||
|
||||
err := c.validateRequestHeader(header)
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
handler, err := c.handlerProvider.GetHandler(requestName)
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
response, err := handler.Handle(request)
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
if response == nil {
|
||||
writer.WriteHeader(http.StatusNoContent)
|
||||
c.logger.Infof("[E2 Manager -> Client] #NodebController.handleRequest - status response: %v", http.StatusNoContent)
|
||||
return
|
||||
}
|
||||
|
||||
result, err := response.Marshal()
|
||||
|
||||
if err != nil {
|
||||
c.handleErrorResponse(err, writer)
|
||||
return
|
||||
}
|
||||
|
||||
c.logger.Infof("[E2 Manager -> Client] #NodebController.handleRequest - response: %s", result)
|
||||
writer.Header().Set("Content-Type", "application/json")
|
||||
writer.Write(result)
|
||||
}
|
||||
|
||||
func (c *NodebController) validateRequestHeader(header *http.Header) error {
|
||||
|
||||
if header.Get("Content-Type") != "application/json" {
|
||||
c.logger.Errorf("#NodebController.validateRequestHeader - validation failure, incorrect content type")
|
||||
|
||||
return e2managererrors.NewHeaderValidationError()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *NodebController) handleErrorResponse(err error, writer http.ResponseWriter) {
|
||||
|
||||
var errorResponseDetails models.ErrorResponse
|
||||
var httpError int
|
||||
|
||||
if err != nil {
|
||||
switch err.(type) {
|
||||
case *e2managererrors.RnibDbError:
|
||||
e2Error, _ := err.(*e2managererrors.RnibDbError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusInternalServerError
|
||||
case *e2managererrors.CommandAlreadyInProgressError:
|
||||
e2Error, _ := err.(*e2managererrors.CommandAlreadyInProgressError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusMethodNotAllowed
|
||||
case *e2managererrors.HeaderValidationError:
|
||||
e2Error, _ := err.(*e2managererrors.HeaderValidationError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusUnsupportedMediaType
|
||||
case *e2managererrors.WrongStateError:
|
||||
e2Error, _ := err.(*e2managererrors.WrongStateError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusBadRequest
|
||||
case *e2managererrors.RequestValidationError:
|
||||
e2Error, _ := err.(*e2managererrors.RequestValidationError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusBadRequest
|
||||
case *e2managererrors.InvalidJsonError:
|
||||
e2Error, _ := err.(*e2managererrors.InvalidJsonError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusBadRequest
|
||||
case *e2managererrors.RmrError:
|
||||
e2Error, _ := err.(*e2managererrors.RmrError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusInternalServerError
|
||||
case *e2managererrors.ResourceNotFoundError:
|
||||
e2Error, _ := err.(*e2managererrors.ResourceNotFoundError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusNotFound
|
||||
case *e2managererrors.E2TInstanceAbsenceError:
|
||||
e2Error, _ := err.(*e2managererrors.E2TInstanceAbsenceError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusServiceUnavailable
|
||||
case *e2managererrors.RoutingManagerError:
|
||||
e2Error, _ := err.(*e2managererrors.RoutingManagerError)
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusServiceUnavailable
|
||||
default:
|
||||
e2Error := e2managererrors.NewInternalError()
|
||||
errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
|
||||
httpError = http.StatusInternalServerError
|
||||
}
|
||||
}
|
||||
errorResponse, _ := json.Marshal(errorResponseDetails)
|
||||
|
||||
c.logger.Errorf("[E2 Manager -> Client] #NodebController.handleErrorResponse - http status: %d, error response: %+v", httpError, errorResponseDetails)
|
||||
|
||||
writer.Header().Set("Content-Type", "application/json")
|
||||
writer.WriteHeader(httpError)
|
||||
_, err = writer.Write(errorResponse)
|
||||
}
|
||||
|
||||
func (c *NodebController) prettifyRequest(request *http.Request) string {
|
||||
dump, _ := httputil.DumpRequest(request, true)
|
||||
requestPrettyPrint := strings.Replace(string(dump), "\r\n", " ", -1)
|
||||
return strings.Replace(requestPrettyPrint, "\n", "", -1)
|
||||
}
|
||||
960
setup/e2mgr/E2Manager/controllers/nodeb_controller_test.go
Normal file
960
setup/e2mgr/E2Manager/controllers/nodeb_controller_test.go
Normal file
@@ -0,0 +1,960 @@
|
||||
//
|
||||
// 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/e2pdus"
|
||||
"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"
|
||||
ValidationFailureJson = "{\"errorCode\":402,\"errorMessage\":\"Validation error\"}"
|
||||
ResourceNotFoundJson = "{\"errorCode\":404,\"errorMessage\":\"Resource not found\"}"
|
||||
RnibErrorJson = "{\"errorCode\":500,\"errorMessage\":\"RNIB error\"}"
|
||||
InternalErrorJson = "{\"errorCode\":501,\"errorMessage\":\"Internal Server Error. Please try again later\"}"
|
||||
)
|
||||
|
||||
var (
|
||||
ServedNrCellInformationRequiredFields = []string{"cellId", "choiceNrMode", "nrMode", "nrPci", "servedPlmns"}
|
||||
NrNeighbourInformationRequiredFields = []string{"nrCgi", "choiceNrMode", "nrMode", "nrPci"}
|
||||
)
|
||||
|
||||
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 removeServedNrCellsParams struct {
|
||||
servedNrCells []*entities.ServedNRCell
|
||||
err error
|
||||
}
|
||||
|
||||
type controllerUpdateGnbTestContext struct {
|
||||
getNodebInfoResult *getNodebInfoResult
|
||||
removeServedNrCellsParams *removeServedNrCellsParams
|
||||
updateGnbCellsParams *updateGnbCellsParams
|
||||
requestBody map[string]interface{}
|
||||
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 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 setupControllerTest(t *testing.T) (*NodebController, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock, *mocks.E2TInstancesManagerMock) {
|
||||
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)
|
||||
ranSetupManager := managers.NewRanSetupManager(log, rmrSender, rnibDataService)
|
||||
e2tInstancesManager := &mocks.E2TInstancesManagerMock{}
|
||||
httpClientMock := &mocks.HttpClientMock{}
|
||||
rmClient := clients.NewRoutingManagerClient(log, config, httpClientMock)
|
||||
e2tAssociationManager := managers.NewE2TAssociationManager(log, rnibDataService, e2tInstancesManager, rmClient)
|
||||
handlerProvider := httpmsghandlerprovider.NewIncomingRequestHandlerProvider(log, rmrSender, config, rnibDataService, ranSetupManager, e2tInstancesManager, e2tAssociationManager, rmClient)
|
||||
controller := NewNodebController(log, handlerProvider)
|
||||
return controller, readerMock, writerMock, rmrMessengerMock, e2tInstancesManager
|
||||
}
|
||||
|
||||
func TestX2SetupInvalidBody(t *testing.T) {
|
||||
|
||||
controller, _, _, _, _ := setupControllerTest(t)
|
||||
|
||||
header := http.Header{}
|
||||
header.Set("Content-Type", "application/json")
|
||||
httpRequest, _ := http.NewRequest("POST", "http://localhost:3800/v1/nodeb/x2-setup", strings.NewReader("{}{}"))
|
||||
httpRequest.Header = header
|
||||
|
||||
writer := httptest.NewRecorder()
|
||||
controller.X2Setup(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 TestX2SetupSuccess(t *testing.T) {
|
||||
|
||||
controller, readerMock, writerMock, rmrMessengerMock, _ := setupControllerTest(t)
|
||||
|
||||
ranName := "test"
|
||||
nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, AssociatedE2TInstanceAddress: "10.0.2.15:8989"}
|
||||
readerMock.On("GetNodeb", ranName).Return(nb, nil)
|
||||
var nbUpdated = *nb
|
||||
writerMock.On("UpdateNodebInfo", &nbUpdated).Return(nil)
|
||||
|
||||
var nbUpdated2 = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, AssociatedE2TInstanceAddress: "10.0.2.15:8989"}
|
||||
writerMock.On("UpdateNodebInfo", nbUpdated2).Return(nil)
|
||||
|
||||
payload := e2pdus.PackedX2setupRequest
|
||||
var xAction []byte
|
||||
var msgSrc unsafe.Pointer
|
||||
msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
|
||||
|
||||
rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
|
||||
|
||||
header := http.Header{}
|
||||
header.Set("Content-Type", "application/json")
|
||||
httpRequest := tests.GetHttpRequest()
|
||||
httpRequest.Header = header
|
||||
|
||||
writer := httptest.NewRecorder()
|
||||
controller.X2Setup(writer, httpRequest)
|
||||
|
||||
assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
|
||||
}
|
||||
|
||||
func TestEndcSetupSuccess(t *testing.T) {
|
||||
|
||||
controller, readerMock, writerMock, rmrMessengerMock, _ := setupControllerTest(t)
|
||||
|
||||
ranName := "test"
|
||||
nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, AssociatedE2TInstanceAddress: "10.0.2.15:8989"}
|
||||
readerMock.On("GetNodeb", ranName).Return(nb, nil)
|
||||
var nbUpdated = *nb
|
||||
writerMock.On("UpdateNodebInfo", &nbUpdated).Return(nil)
|
||||
|
||||
var nbUpdated2 = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, AssociatedE2TInstanceAddress: "10.0.2.15:8989"}
|
||||
writerMock.On("UpdateNodebInfo", nbUpdated2).Return(nil)
|
||||
|
||||
payload := e2pdus.PackedEndcX2setupRequest
|
||||
var xAction []byte
|
||||
var msgSrc unsafe.Pointer
|
||||
msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xAction, msgSrc)
|
||||
|
||||
rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
|
||||
|
||||
header := http.Header{}
|
||||
header.Set("Content-Type", "application/json")
|
||||
httpRequest := tests.GetHttpRequest()
|
||||
httpRequest.Header = header
|
||||
|
||||
writer := httptest.NewRecorder()
|
||||
controller.EndcSetup(writer, httpRequest)
|
||||
|
||||
assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
|
||||
}
|
||||
|
||||
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 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, _, _, _ := setupControllerTest(t)
|
||||
writer := httptest.NewRecorder()
|
||||
readerMock.On("GetListNodebIds").Return(context.nodebIdList, context.rnibError)
|
||||
req, _ := http.NewRequest(http.MethodGet, "/nodeb/ids", nil)
|
||||
controller.GetNodebIdList(writer, req)
|
||||
assert.Equal(t, context.expectedStatusCode, writer.Result().StatusCode)
|
||||
bodyBytes, _ := ioutil.ReadAll(writer.Body)
|
||||
assert.Equal(t, context.expectedJsonResponse, string(bodyBytes))
|
||||
}
|
||||
|
||||
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)
|
||||
|
||||
if context.getNodebInfoResult != nil {
|
||||
readerMock.AssertNotCalled(t, "GetNodeb")
|
||||
}
|
||||
|
||||
if context.updateGnbCellsParams != nil {
|
||||
writerMock.AssertNotCalled(t, "UpdateGnb")
|
||||
}
|
||||
|
||||
if context.removeServedNrCellsParams != nil {
|
||||
writerMock.AssertNotCalled(t, "RemoveServedNrCells")
|
||||
}
|
||||
}
|
||||
|
||||
func buildUpdateGnbRequest(context *controllerUpdateGnbTestContext) *http.Request {
|
||||
updateGnbUrl := fmt.Sprintf("/nodeb/%s/update", RanName)
|
||||
requestBody := getJsonRequestAsBuffer(context.requestBody)
|
||||
req, _ := http.NewRequest(http.MethodGet, updateGnbUrl, requestBody)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req = mux.SetURLVars(req, map[string]string{"ranName": RanName})
|
||||
return req
|
||||
}
|
||||
|
||||
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 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)
|
||||
}
|
||||
}
|
||||
|
||||
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,
|
||||
},
|
||||
rnibError: nil,
|
||||
},
|
||||
requestBody: map[string]interface{}{
|
||||
"servedNrCells": []interface{}{
|
||||
map[string]interface{}{
|
||||
"servedNrCellInformation": buildServedNrCellInformation(""),
|
||||
"nrNeighbourInfos": []interface{}{
|
||||
buildNrNeighbourInformation(""),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
expectedStatusCode: http.StatusInternalServerError,
|
||||
expectedJsonResponse: InternalErrorJson,
|
||||
}
|
||||
|
||||
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}},
|
||||
},
|
||||
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}},
|
||||
},
|
||||
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 TestControllerUpdateGnbSuccess(t *testing.T) {
|
||||
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{}},
|
||||
},
|
||||
rnibError: nil,
|
||||
},
|
||||
requestBody: map[string]interface{}{
|
||||
"servedNrCells": []interface{}{
|
||||
map[string]interface{}{
|
||||
"servedNrCellInformation": buildServedNrCellInformation(""),
|
||||
"nrNeighbourInfos": []interface{}{
|
||||
buildNrNeighbourInformation(""),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
expectedStatusCode: http.StatusOK,
|
||||
expectedJsonResponse: "{\"ranName\":\"test\",\"connectionStatus\":\"CONNECTED\",\"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 getJsonRequestAsBuffer(requestJson map[string]interface{}) *bytes.Buffer {
|
||||
b := new(bytes.Buffer)
|
||||
_ = json.NewEncoder(b).Encode(requestJson)
|
||||
return b
|
||||
}
|
||||
|
||||
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\"}}]",
|
||||
}
|
||||
|
||||
controllerGetNodebIdListTestExecuter(t, &context)
|
||||
}
|
||||
|
||||
func TestControllerGetNodebIdListEmptySuccess(t *testing.T) {
|
||||
var rnibError error
|
||||
nodebIdList := []*entities.NbIdentity{}
|
||||
|
||||
context := controllerGetNodebIdListTestContext{
|
||||
nodebIdList: nodebIdList,
|
||||
rnibError: rnibError,
|
||||
expectedStatusCode: http.StatusOK,
|
||||
expectedJsonResponse: "[]",
|
||||
}
|
||||
|
||||
controllerGetNodebIdListTestExecuter(t, &context)
|
||||
}
|
||||
|
||||
func TestControllerGetNodebIdListInternal(t *testing.T) {
|
||||
var nodebIdList []*entities.NbIdentity
|
||||
context := controllerGetNodebIdListTestContext{
|
||||
nodebIdList: nodebIdList,
|
||||
rnibError: common.NewInternalError(errors.New("#reader.GetNodeb - Internal Error")),
|
||||
expectedStatusCode: http.StatusInternalServerError,
|
||||
expectedJsonResponse: RnibErrorJson,
|
||||
}
|
||||
|
||||
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)
|
||||
|
||||
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)
|
||||
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)
|
||||
}
|
||||
50
setup/e2mgr/E2Manager/controllers/root_controller.go
Normal file
50
setup/e2mgr/E2Manager/controllers/root_controller.go
Normal file
@@ -0,0 +1,50 @@
|
||||
//
|
||||
// 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 (
|
||||
"e2mgr/services"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type IRootController interface {
|
||||
HandleHealthCheckRequest(writer http.ResponseWriter, request *http.Request)
|
||||
}
|
||||
|
||||
type RootController struct {
|
||||
rnibDataService services.RNibDataService
|
||||
}
|
||||
|
||||
func NewRootController(rnibDataService services.RNibDataService) *RootController {
|
||||
return &RootController{
|
||||
rnibDataService: rnibDataService,
|
||||
}
|
||||
}
|
||||
|
||||
func (rc *RootController) HandleHealthCheckRequest(writer http.ResponseWriter, request *http.Request) {
|
||||
httpStatus := http.StatusOK
|
||||
isOn := rc.rnibDataService.PingRnib()
|
||||
if !isOn {
|
||||
httpStatus = http.StatusInternalServerError
|
||||
}
|
||||
|
||||
writer.WriteHeader(httpStatus)
|
||||
}
|
||||
94
setup/e2mgr/E2Manager/controllers/root_controller_test.go
Normal file
94
setup/e2mgr/E2Manager/controllers/root_controller_test.go
Normal file
@@ -0,0 +1,94 @@
|
||||
//
|
||||
// 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 (
|
||||
"e2mgr/configuration"
|
||||
"e2mgr/logger"
|
||||
"e2mgr/mocks"
|
||||
"e2mgr/services"
|
||||
"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/stretchr/testify/assert"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func setupNodebControllerTest(t *testing.T) (services.RNibDataService, *mocks.RnibReaderMock){
|
||||
logger, err := logger.InitLogger(logger.DebugLevel)
|
||||
if err != nil {
|
||||
t.Errorf("#... - failed to initialize logger, error: %s", err)
|
||||
}
|
||||
config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
|
||||
readerMock := &mocks.RnibReaderMock{}
|
||||
|
||||
writerMock := &mocks.RnibWriterMock{}
|
||||
|
||||
rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
|
||||
return rnibDataService, readerMock
|
||||
}
|
||||
|
||||
func TestNewRequestController(t *testing.T) {
|
||||
rnibDataService, _ := setupNodebControllerTest(t)
|
||||
assert.NotNil(t, NewRootController(rnibDataService))
|
||||
}
|
||||
|
||||
func TestHandleHealthCheckRequestGood(t *testing.T) {
|
||||
rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
|
||||
|
||||
var nbList []*entities.NbIdentity
|
||||
rnibReaderMock.On("GetListNodebIds").Return(nbList, nil)
|
||||
|
||||
rc := NewRootController(rnibDataService)
|
||||
writer := httptest.NewRecorder()
|
||||
rc.HandleHealthCheckRequest(writer, nil)
|
||||
assert.Equal(t, http.StatusOK, writer.Result().StatusCode)
|
||||
}
|
||||
|
||||
func TestHandleHealthCheckRequestOtherError(t *testing.T) {
|
||||
rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
|
||||
|
||||
mockOtherErr := &common.InternalError{Err: fmt.Errorf("non connection error")}
|
||||
var nbList []*entities.NbIdentity
|
||||
rnibReaderMock.On("GetListNodebIds").Return(nbList, mockOtherErr)
|
||||
|
||||
rc := NewRootController(rnibDataService)
|
||||
writer := httptest.NewRecorder()
|
||||
rc.HandleHealthCheckRequest(writer, nil)
|
||||
assert.Equal(t, http.StatusOK, writer.Result().StatusCode)
|
||||
}
|
||||
|
||||
func TestHandleHealthCheckRequestConnError(t *testing.T) {
|
||||
rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
|
||||
|
||||
mockConnErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
|
||||
var nbList []*entities.NbIdentity
|
||||
rnibReaderMock.On("GetListNodebIds").Return(nbList, mockConnErr)
|
||||
|
||||
|
||||
rc := NewRootController(rnibDataService)
|
||||
writer := httptest.NewRecorder()
|
||||
rc.HandleHealthCheckRequest(writer, nil)
|
||||
assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
|
||||
}
|
||||
Reference in New Issue
Block a user