138 lines
4.7 KiB
Go
138 lines
4.7 KiB
Go
//
|
|
// 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)
|
|
}
|