E2Manager/httpserver/http_server_test.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   4 //
   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
   8 //
   9 //      http://www.apache.org/licenses/LICENSE-2.0
  10 //
  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.
  16 //
  17
  18 package httpserver
  19
  20 import (
  21         "e2mgr/mocks"
  22         "github.com/gorilla/mux"
  23         "github.com/stretchr/testify/assert"
  24         "net/http"
  25         "net/http/httptest"
  26         "testing"
  27 )
  28
  29 func setupRouterAndMocks() (*mux.Router, *mocks.ControllerMock, *mocks.NodebControllerMock) {
  30         controllerMock := &mocks.ControllerMock{}
  31         controllerMock.On("Shutdown").Return(nil)
  32         controllerMock.On("X2Reset").Return(nil)
  33         controllerMock.On("X2Setup").Return(nil)
  34         controllerMock.On("EndcSetup").Return(nil)
  35         controllerMock.On("GetNodeb").Return(nil)
  36         controllerMock.On("GetNodebIdList").Return(nil)
  37
  38
  39
  40         nodebControllerMock := &mocks.NodebControllerMock{}
  41         nodebControllerMock.On("GetNodebIdList").Return(nil)
  42         nodebControllerMock.On("GetNodeb").Return(nil) // TODO: remove
  43         nodebControllerMock.On("HandleHealthCheckRequest").Return(nil)
  44
  45         router := mux.NewRouter()
  46         initializeRoutes(router, nodebControllerMock, controllerMock)
  47         return router, controllerMock, nodebControllerMock
  48 }
  49
  50 func TestRoutePostEndcSetup(t *testing.T) {
  51         router, controllerMock, _ := setupRouterAndMocks()
  52
  53         req, err := http.NewRequest("POST", "/v1/nodeb/endc-setup", nil)
  54         if err != nil {
  55                 t.Fatal(err)
  56         }
  57         rr := httptest.NewRecorder()
  58         router.ServeHTTP(rr, req)
  59
  60         controllerMock.AssertNumberOfCalls(t,"EndcSetup", 1)
  61 }
  62
  63 func TestRoutePostX2Setup(t *testing.T) {
  64         router, controllerMock, _ := setupRouterAndMocks()
  65
  66         req, err := http.NewRequest("POST", "/v1/nodeb/x2-setup", nil)
  67         if err != nil {
  68                 t.Fatal(err)
  69         }
  70         rr := httptest.NewRecorder()
  71         router.ServeHTTP(rr, req)
  72
  73         controllerMock.AssertNumberOfCalls(t,"X2Setup", 1)
  74 }
  75
  76 func TestRouteGetNodebIds(t *testing.T) {
  77         router, controllerMock, _ := setupRouterAndMocks()
  78
  79         req, err := http.NewRequest("GET", "/v1/nodeb/ids", nil)
  80         if err != nil {
  81                 t.Fatal(err)
  82         }
  83         rr := httptest.NewRecorder()
  84         router.ServeHTTP(rr, req)
  85
  86         controllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
  87 }
  88
  89 func TestRouteGetNodebRanName(t *testing.T) {
  90         router, controllerMock,_ := setupRouterAndMocks()
  91
  92         req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
  93         if err != nil {
  94                 t.Fatal(err)
  95         }
  96         rr := httptest.NewRecorder()
  97         router.ServeHTTP(rr, req)
  98
  99         assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
 100         assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
 101         controllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
 102 }
 103
 104 func TestRouteGetHealth(t *testing.T) {
 105         router, _, nodebControllerMock := setupRouterAndMocks()
 106
 107         req, err := http.NewRequest("GET", "/v1/health", nil)
 108         if err != nil {
 109                 t.Fatal(err)
 110         }
 111         rr := httptest.NewRecorder()
 112         router.ServeHTTP(rr, req)
 113
 114         nodebControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
 115 }
 116
 117 func TestRoutePutNodebShutdown(t *testing.T) {
 118         router, controllerMock, _ := setupRouterAndMocks()
 119
 120         req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
 121         if err != nil {
 122                 t.Fatal(err)
 123         }
 124         rr := httptest.NewRecorder()
 125         router.ServeHTTP(rr, req)
 126
 127         controllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
 128 }
 129
 130 func TestRoutePutNodebResetRanName(t *testing.T) {
 131         router, controllerMock, _ := setupRouterAndMocks()
 132
 133         req, err := http.NewRequest("PUT", "/v1/nodeb/ran1/reset", nil)
 134         if err != nil {
 135                 t.Fatal(err)
 136         }
 137         rr := httptest.NewRecorder()
 138         router.ServeHTTP(rr, req)
 139
 140         assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
 141         assert.Equal(t, "ran1", rr.Body.String(), "handler returned wrong body")
 142         controllerMock.AssertNumberOfCalls(t, "X2Reset", 1)
 143 }
 144
 145 func TestRouteNotFound(t *testing.T) {
 146         router, _, _ := setupRouterAndMocks()
 147
 148         req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
 149         if err != nil {
 150                 t.Fatal(err)
 151         }
 152         rr := httptest.NewRecorder()
 153         router.ServeHTTP(rr, req)
 154
 155         assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
 156 }