2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
4 // Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved.
6 // Licensed under the Apache License, Version 2.0 (the "License");
7 // you may not use this file except in compliance with the License.
8 // You may obtain a copy of the License at
10 // http://www.apache.org/licenses/LICENSE-2.0
12 // Unless required by applicable law or agreed to in writing, software
13 // distributed under the License is distributed on an "AS IS" BASIS,
14 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 // See the License for the specific language governing permissions and
16 // limitations under the License.
18 // This source code is part of the near-RT RIC (RAN Intelligent Controller)
19 // platform project (RICP).
26 "github.com/gorilla/mux"
27 "github.com/stretchr/testify/assert"
34 func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock, *mocks.NodebControllerMock, *mocks.E2TControllerMock, *mocks.SymptomdataControllerMock) {
35 rootControllerMock := &mocks.RootControllerMock{}
36 rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
38 nodebControllerMock := &mocks.NodebControllerMock{}
39 nodebControllerMock.On("Shutdown").Return(nil)
40 nodebControllerMock.On("GetNodeb").Return(nil)
41 nodebControllerMock.On("GetNodebIdList").Return(nil)
42 nodebControllerMock.On("GetNodebId").Return(nil)
43 nodebControllerMock.On("SetGeneralConfiguration").Return(nil)
44 nodebControllerMock.On("DeleteEnb").Return(nil)
45 nodebControllerMock.On("AddEnb").Return(nil)
46 nodebControllerMock.On("UpdateEnb").Return(nil)
47 nodebControllerMock.On("HealthCheckRequest").Return(nil)
49 e2tControllerMock := &mocks.E2TControllerMock{}
50 e2tControllerMock.On("GetE2TInstances").Return(nil)
52 symptomdataControllerMock := &mocks.SymptomdataControllerMock{}
53 symptomdataControllerMock.On("GetSymptomData").Return(nil)
55 router := mux.NewRouter()
56 initializeRoutes(router, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock)
57 return router, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock
60 func TestRouteGetNodebIdList(t *testing.T) {
61 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
63 req, err := http.NewRequest("GET", "/v1/nodeb/states", nil)
67 rr := httptest.NewRecorder()
68 router.ServeHTTP(rr, req)
70 nodebControllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
73 func TestRouteGetNodebId(t *testing.T) {
74 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
76 req, err := http.NewRequest("GET", "/v1/nodeb/states/ran1", nil)
80 rr := httptest.NewRecorder()
81 router.ServeHTTP(rr, req)
83 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
84 nodebControllerMock.AssertNumberOfCalls(t, "GetNodebId", 1)
87 func TestRouteGetNodebRanName(t *testing.T) {
88 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
90 req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
94 rr := httptest.NewRecorder()
95 router.ServeHTTP(rr, req)
97 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
98 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
99 nodebControllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
102 func TestRouteGetHealth(t *testing.T) {
103 router, rootControllerMock, _, _, _ := setupRouterAndMocks()
105 req, err := http.NewRequest("GET", "/v1/health", nil)
109 rr := httptest.NewRecorder()
110 router.ServeHTTP(rr, req)
112 rootControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
115 func TestRoutePutNodebShutdown(t *testing.T) {
116 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
118 req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
122 rr := httptest.NewRecorder()
123 router.ServeHTTP(rr, req)
125 nodebControllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
128 func TestHealthCheckRequest(t *testing.T) {
129 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
131 req, err := http.NewRequest("PUT", "/v1/nodeb/health", nil)
135 rr := httptest.NewRecorder()
136 router.ServeHTTP(rr, req)
138 assert.Equal(t, http.StatusAccepted, rr.Code, "handler returned wrong status code")
139 nodebControllerMock.AssertNumberOfCalls(t, "HealthCheckRequest", 1)
142 func TestRoutePutNodebSetGeneralConfiguration(t *testing.T) {
143 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
145 req, err := http.NewRequest("PUT", "/v1/nodeb/parameters", nil)
149 rr := httptest.NewRecorder()
150 router.ServeHTTP(rr, req)
152 nodebControllerMock.AssertNumberOfCalls(t, "SetGeneralConfiguration", 1)
155 func TestRoutePutUpdateEnb(t *testing.T) {
156 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
158 req, err := http.NewRequest("PUT", "/v1/nodeb/enb/ran1", nil)
162 rr := httptest.NewRecorder()
163 router.ServeHTTP(rr, req)
165 nodebControllerMock.AssertNumberOfCalls(t, "UpdateEnb", 1)
168 func TestRouteNotFound(t *testing.T) {
169 router, _, _, _, _ := setupRouterAndMocks()
171 req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
175 rr := httptest.NewRecorder()
176 router.ServeHTTP(rr, req)
178 assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
181 func TestRunError(t *testing.T) {
183 err := Run(log, 1234567, &mocks.RootControllerMock{}, &mocks.NodebControllerMock{}, &mocks.E2TControllerMock{}, &mocks.SymptomdataControllerMock{})
184 assert.NotNil(t, err)
187 func TestRun(t *testing.T) {
189 _, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock := setupRouterAndMocks()
190 go Run(log, 11223, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock)
192 time.Sleep(time.Millisecond * 100)
193 resp, err := http.Get("http://localhost:11223/v1/health")
195 t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
197 assert.Equal(t, 200, resp.StatusCode)
200 func TestRouteAddEnb(t *testing.T) {
201 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
203 req, err := http.NewRequest("POST", "/v1/nodeb/enb", nil)
207 rr := httptest.NewRecorder()
208 router.ServeHTTP(rr, req)
210 assert.Equal(t, http.StatusCreated, rr.Code, "handler returned wrong status code")
211 nodebControllerMock.AssertNumberOfCalls(t, "AddEnb", 1)
214 func TestRouteDeleteEnb(t *testing.T) {
215 router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
217 req, err := http.NewRequest("DELETE", "/v1/nodeb/enb/ran1", nil)
221 rr := httptest.NewRecorder()
222 router.ServeHTTP(rr, req)
224 assert.Equal(t, http.StatusNoContent, rr.Code, "handler returned wrong status code")
225 nodebControllerMock.AssertNumberOfCalls(t, "DeleteEnb", 1)
228 func initLog(t *testing.T) *logger.Logger {
230 log, err := logger.InitLogger(InfoLevel)
232 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)