2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
9 // http://www.apache.org/licenses/LICENSE-2.0
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
17 // This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 // platform project (RICP).
25 "github.com/gorilla/mux"
26 "github.com/stretchr/testify/assert"
33 func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock, *mocks.NodebControllerMock, *mocks.E2TControllerMock) {
34 rootControllerMock := &mocks.RootControllerMock{}
35 rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
37 nodebControllerMock := &mocks.NodebControllerMock{}
38 nodebControllerMock.On("Shutdown").Return(nil)
39 nodebControllerMock.On("GetNodeb").Return(nil)
40 nodebControllerMock.On("GetNodebIdList").Return(nil)
41 nodebControllerMock.On("SetGeneralConfiguration").Return(nil)
42 nodebControllerMock.On("DeleteEnb").Return(nil)
43 nodebControllerMock.On("AddEnb").Return(nil)
44 nodebControllerMock.On("UpdateEnb").Return(nil)
45 nodebControllerMock.On("HealthCheckRequest").Return(nil)
47 e2tControllerMock := &mocks.E2TControllerMock{}
49 e2tControllerMock.On("GetE2TInstances").Return(nil)
51 router := mux.NewRouter()
52 initializeRoutes(router, rootControllerMock, nodebControllerMock, e2tControllerMock)
53 return router, rootControllerMock, nodebControllerMock, e2tControllerMock
56 func TestRouteGetNodebIds(t *testing.T) {
57 router, _, nodebControllerMock, _ := setupRouterAndMocks()
59 req, err := http.NewRequest("GET", "/v1/nodeb/states", nil)
63 rr := httptest.NewRecorder()
64 router.ServeHTTP(rr, req)
66 nodebControllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
69 func TestRouteGetNodebRanName(t *testing.T) {
70 router, _, nodebControllerMock, _ := setupRouterAndMocks()
72 req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
76 rr := httptest.NewRecorder()
77 router.ServeHTTP(rr, req)
79 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
80 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
81 nodebControllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
84 func TestRouteGetHealth(t *testing.T) {
85 router, rootControllerMock, _, _ := setupRouterAndMocks()
87 req, err := http.NewRequest("GET", "/v1/health", nil)
91 rr := httptest.NewRecorder()
92 router.ServeHTTP(rr, req)
94 rootControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
97 func TestRoutePutNodebShutdown(t *testing.T) {
98 router, _, nodebControllerMock, _ := setupRouterAndMocks()
100 req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
104 rr := httptest.NewRecorder()
105 router.ServeHTTP(rr, req)
107 nodebControllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
110 func TestHealthCheckRequest(t *testing.T) {
111 router, _, nodebControllerMock, _ := setupRouterAndMocks()
113 req, err := http.NewRequest("PUT", "/v1/nodeb/health", nil)
117 rr := httptest.NewRecorder()
118 router.ServeHTTP(rr, req)
120 nodebControllerMock.AssertNumberOfCalls(t, "HealthCheckRequest", 1)
123 func TestRoutePutNodebSetGeneralConfiguration(t *testing.T) {
124 router, _, nodebControllerMock, _ := setupRouterAndMocks()
126 req, err := http.NewRequest("PUT", "/v1/nodeb/parameters", nil)
130 rr := httptest.NewRecorder()
131 router.ServeHTTP(rr, req)
133 nodebControllerMock.AssertNumberOfCalls(t, "SetGeneralConfiguration", 1)
136 func TestRoutePutUpdateEnb(t *testing.T) {
137 router, _, nodebControllerMock, _ := setupRouterAndMocks()
139 req, err := http.NewRequest("PUT", "/v1/nodeb/enb/ran1", nil)
143 rr := httptest.NewRecorder()
144 router.ServeHTTP(rr, req)
146 nodebControllerMock.AssertNumberOfCalls(t, "UpdateEnb", 1)
149 func TestRouteNotFound(t *testing.T) {
150 router, _, _, _ := setupRouterAndMocks()
152 req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
156 rr := httptest.NewRecorder()
157 router.ServeHTTP(rr, req)
159 assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
162 func TestRunError(t *testing.T) {
164 err := Run(log, 1234567, &mocks.RootControllerMock{}, &mocks.NodebControllerMock{}, &mocks.E2TControllerMock{})
165 assert.NotNil(t, err)
168 func TestRun(t *testing.T) {
170 _, rootControllerMock, nodebControllerMock, e2tControllerMock := setupRouterAndMocks()
171 go Run(log, 11223, rootControllerMock, nodebControllerMock, e2tControllerMock)
173 time.Sleep(time.Millisecond * 100)
174 resp, err := http.Get("http://localhost:11223/v1/health")
176 t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
178 assert.Equal(t, 200, resp.StatusCode)
181 func TestRouteAddEnb(t *testing.T) {
182 router, _, nodebControllerMock, _ := setupRouterAndMocks()
184 req, err := http.NewRequest("POST", "/v1/nodeb/enb", nil)
188 rr := httptest.NewRecorder()
189 router.ServeHTTP(rr, req)
191 assert.Equal(t, http.StatusCreated, rr.Code, "handler returned wrong status code")
192 nodebControllerMock.AssertNumberOfCalls(t, "AddEnb", 1)
195 func TestRouteDeleteEnb(t *testing.T) {
196 router, _, nodebControllerMock, _ := setupRouterAndMocks()
198 req, err := http.NewRequest("DELETE", "/v1/nodeb/enb/ran1", nil)
202 rr := httptest.NewRecorder()
203 router.ServeHTTP(rr, req)
205 assert.Equal(t, http.StatusNoContent, rr.Code, "handler returned wrong status code")
206 nodebControllerMock.AssertNumberOfCalls(t, "DeleteEnb", 1)
209 func initLog(t *testing.T) *logger.Logger {
210 log, err := logger.InitLogger(logger.InfoLevel)
212 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)