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"
32 func setupRouterAndMocks() (*mux.Router, *mocks.ControllerMock, *mocks.NodebControllerMock) {
33 controllerMock := &mocks.ControllerMock{}
34 controllerMock.On("Shutdown").Return(nil)
35 controllerMock.On("X2Reset").Return(nil)
36 controllerMock.On("X2Setup").Return(nil)
37 controllerMock.On("EndcSetup").Return(nil)
38 controllerMock.On("GetNodeb").Return(nil)
39 controllerMock.On("GetNodebIdList").Return(nil)
43 nodebControllerMock := &mocks.NodebControllerMock{}
44 nodebControllerMock.On("GetNodebIdList").Return(nil)
45 nodebControllerMock.On("GetNodeb").Return(nil) // TODO: remove
46 nodebControllerMock.On("HandleHealthCheckRequest").Return(nil)
48 router := mux.NewRouter()
49 initializeRoutes(router, nodebControllerMock, controllerMock)
50 return router, controllerMock, nodebControllerMock
53 func TestRoutePostEndcSetup(t *testing.T) {
54 router, controllerMock, _ := setupRouterAndMocks()
56 req, err := http.NewRequest("POST", "/v1/nodeb/endc-setup", nil)
60 rr := httptest.NewRecorder()
61 router.ServeHTTP(rr, req)
63 controllerMock.AssertNumberOfCalls(t,"EndcSetup", 1)
66 func TestRoutePostX2Setup(t *testing.T) {
67 router, controllerMock, _ := setupRouterAndMocks()
69 req, err := http.NewRequest("POST", "/v1/nodeb/x2-setup", nil)
73 rr := httptest.NewRecorder()
74 router.ServeHTTP(rr, req)
76 controllerMock.AssertNumberOfCalls(t,"X2Setup", 1)
79 func TestRouteGetNodebIds(t *testing.T) {
80 router, controllerMock, _ := setupRouterAndMocks()
82 req, err := http.NewRequest("GET", "/v1/nodeb/ids", nil)
86 rr := httptest.NewRecorder()
87 router.ServeHTTP(rr, req)
89 controllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
92 func TestRouteGetNodebRanName(t *testing.T) {
93 router, controllerMock,_ := setupRouterAndMocks()
95 req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
99 rr := httptest.NewRecorder()
100 router.ServeHTTP(rr, req)
102 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
103 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
104 controllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
107 func TestRouteGetHealth(t *testing.T) {
108 router, _, nodebControllerMock := setupRouterAndMocks()
110 req, err := http.NewRequest("GET", "/v1/health", nil)
114 rr := httptest.NewRecorder()
115 router.ServeHTTP(rr, req)
117 nodebControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
120 func TestRoutePutNodebShutdown(t *testing.T) {
121 router, controllerMock, _ := setupRouterAndMocks()
123 req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
127 rr := httptest.NewRecorder()
128 router.ServeHTTP(rr, req)
130 controllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
133 func TestRoutePutNodebResetRanName(t *testing.T) {
134 router, controllerMock, _ := setupRouterAndMocks()
136 req, err := http.NewRequest("PUT", "/v1/nodeb/ran1/reset", nil)
140 rr := httptest.NewRecorder()
141 router.ServeHTTP(rr, req)
143 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
144 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
145 controllerMock.AssertNumberOfCalls(t, "X2Reset", 1)
148 func TestRouteNotFound(t *testing.T) {
149 router, _, _ := setupRouterAndMocks()
151 req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
155 rr := httptest.NewRecorder()
156 router.ServeHTTP(rr, req)
158 assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")