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.
23 "github.com/gorilla/mux"
24 "github.com/stretchr/testify/assert"
31 func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock, *mocks.NodebControllerMock, *mocks.E2TControllerMock) {
32 rootControllerMock := &mocks.RootControllerMock{}
33 rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
35 nodebControllerMock := &mocks.NodebControllerMock{}
36 nodebControllerMock.On("Shutdown").Return(nil)
37 nodebControllerMock.On("X2Reset").Return(nil)
38 nodebControllerMock.On("X2Setup").Return(nil)
39 nodebControllerMock.On("EndcSetup").Return(nil)
40 nodebControllerMock.On("GetNodeb").Return(nil)
41 nodebControllerMock.On("GetNodebIdList").Return(nil)
43 e2tControllerMock := &mocks.E2TControllerMock{}
45 e2tControllerMock.On("GetE2TInstances").Return(nil)
47 router := mux.NewRouter()
48 initializeRoutes(router, rootControllerMock, nodebControllerMock, e2tControllerMock)
49 return router, rootControllerMock, nodebControllerMock, e2tControllerMock
52 func TestRoutePostEndcSetup(t *testing.T) {
53 router, _, nodebControllerMock, _ := setupRouterAndMocks()
55 req, err := http.NewRequest("POST", "/v1/nodeb/endc-setup", nil)
59 rr := httptest.NewRecorder()
60 router.ServeHTTP(rr, req)
62 nodebControllerMock.AssertNumberOfCalls(t, "EndcSetup", 1)
65 func TestRoutePostX2Setup(t *testing.T) {
66 router, _, nodebControllerMock, _ := setupRouterAndMocks()
68 req, err := http.NewRequest("POST", "/v1/nodeb/x2-setup", nil)
72 rr := httptest.NewRecorder()
73 router.ServeHTTP(rr, req)
75 nodebControllerMock.AssertNumberOfCalls(t, "X2Setup", 1)
78 func TestRouteGetNodebIds(t *testing.T) {
79 router, _, nodebControllerMock, _ := setupRouterAndMocks()
81 req, err := http.NewRequest("GET", "/v1/nodeb/ids", nil)
85 rr := httptest.NewRecorder()
86 router.ServeHTTP(rr, req)
88 nodebControllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
91 func TestRouteGetNodebRanName(t *testing.T) {
92 router, _, nodebControllerMock, _ := setupRouterAndMocks()
94 req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
98 rr := httptest.NewRecorder()
99 router.ServeHTTP(rr, req)
101 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
102 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
103 nodebControllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
106 func TestRouteGetHealth(t *testing.T) {
107 router, rootControllerMock, _, _ := setupRouterAndMocks()
109 req, err := http.NewRequest("GET", "/v1/health", nil)
113 rr := httptest.NewRecorder()
114 router.ServeHTTP(rr, req)
116 rootControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
119 func TestRoutePutNodebShutdown(t *testing.T) {
120 router, _, nodebControllerMock, _ := setupRouterAndMocks()
122 req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
126 rr := httptest.NewRecorder()
127 router.ServeHTTP(rr, req)
129 nodebControllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
132 func TestRoutePutNodebResetRanName(t *testing.T) {
133 router, _, nodebControllerMock, _ := setupRouterAndMocks()
135 req, err := http.NewRequest("PUT", "/v1/nodeb/ran1/reset", nil)
139 rr := httptest.NewRecorder()
140 router.ServeHTTP(rr, req)
142 assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
143 assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
144 nodebControllerMock.AssertNumberOfCalls(t, "X2Reset", 1)
147 func TestRouteNotFound(t *testing.T) {
148 router, _, _,_ := setupRouterAndMocks()
150 req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
154 rr := httptest.NewRecorder()
155 router.ServeHTTP(rr, req)
157 assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
160 func TestRunError(t *testing.T) {
162 err := Run(log, 1234567, &mocks.RootControllerMock{}, &mocks.NodebControllerMock{}, &mocks.E2TControllerMock{})
163 assert.NotNil(t, err)
166 func TestRun(t *testing.T) {
168 _, rootControllerMock, nodebControllerMock, e2tControllerMock := setupRouterAndMocks()
169 go Run(log, 11223, rootControllerMock, nodebControllerMock, e2tControllerMock)
171 time.Sleep(time.Millisecond * 100)
172 resp, err := http.Get("http://localhost:11223/v1/health")
174 t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
176 assert.Equal(t, 200, resp.StatusCode)
179 func initLog(t *testing.T) *logger.Logger {
180 log, err := logger.InitLogger(logger.InfoLevel)
182 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)