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/logger"
  22         "e2mgr/mocks"
  23         "github.com/gorilla/mux"
  24         "github.com/stretchr/testify/assert"
  25         "net/http"
  26         "net/http/httptest"
  27         "testing"
  28         "time"
  29 )
  30
  31 func setupRouterAndMocks() (*mux.Router, *mocks.ControllerMock, *mocks.NodebControllerMock) {
  32         controllerMock := &mocks.ControllerMock{}
  33         controllerMock.On("Shutdown").Return(nil)
  34         controllerMock.On("X2Reset").Return(nil)
  35         controllerMock.On("X2Setup").Return(nil)
  36         controllerMock.On("EndcSetup").Return(nil)
  37         controllerMock.On("GetNodeb").Return(nil)
  38         controllerMock.On("GetNodebIdList").Return(nil)
  39
  40
  41
  42         nodebControllerMock := &mocks.NodebControllerMock{}
  43         nodebControllerMock.On("GetNodebIdList").Return(nil)
  44         nodebControllerMock.On("GetNodeb").Return(nil) // TODO: remove
  45         nodebControllerMock.On("HandleHealthCheckRequest").Return(nil)
  46
  47         router := mux.NewRouter()
  48         initializeRoutes(router, nodebControllerMock, controllerMock)
  49         return router, controllerMock, nodebControllerMock
  50 }
  51
  52 func TestRoutePostEndcSetup(t *testing.T) {
  53         router, controllerMock, _ := setupRouterAndMocks()
  54
  55         req, err := http.NewRequest("POST", "/v1/nodeb/endc-setup", nil)
  56         if err != nil {
  57                 t.Fatal(err)
  58         }
  59         rr := httptest.NewRecorder()
  60         router.ServeHTTP(rr, req)
  61
  62         controllerMock.AssertNumberOfCalls(t,"EndcSetup", 1)
  63 }
  64
  65 func TestRoutePostX2Setup(t *testing.T) {
  66         router, controllerMock, _ := setupRouterAndMocks()
  67
  68         req, err := http.NewRequest("POST", "/v1/nodeb/x2-setup", nil)
  69         if err != nil {
  70                 t.Fatal(err)
  71         }
  72         rr := httptest.NewRecorder()
  73         router.ServeHTTP(rr, req)
  74
  75         controllerMock.AssertNumberOfCalls(t,"X2Setup", 1)
  76 }
  77
  78 func TestRouteGetNodebIds(t *testing.T) {
  79         router, controllerMock, _ := setupRouterAndMocks()
  80
  81         req, err := http.NewRequest("GET", "/v1/nodeb/ids", nil)
  82         if err != nil {
  83                 t.Fatal(err)
  84         }
  85         rr := httptest.NewRecorder()
  86         router.ServeHTTP(rr, req)
  87
  88         controllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
  89 }
  90
  91 func TestRouteGetNodebRanName(t *testing.T) {
  92         router, controllerMock,_ := setupRouterAndMocks()
  93
  94         req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
  95         if err != nil {
  96                 t.Fatal(err)
  97         }
  98         rr := httptest.NewRecorder()
  99         router.ServeHTTP(rr, req)
 100
 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         controllerMock.AssertNumberOfCalls(t, "GetNodeb", 1)
 104 }
 105
 106 func TestRouteGetHealth(t *testing.T) {
 107         router, _, nodebControllerMock := setupRouterAndMocks()
 108
 109         req, err := http.NewRequest("GET", "/v1/health", nil)
 110         if err != nil {
 111                 t.Fatal(err)
 112         }
 113         rr := httptest.NewRecorder()
 114         router.ServeHTTP(rr, req)
 115
 116         nodebControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
 117 }
 118
 119 func TestRoutePutNodebShutdown(t *testing.T) {
 120         router, controllerMock, _ := setupRouterAndMocks()
 121
 122         req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
 123         if err != nil {
 124                 t.Fatal(err)
 125         }
 126         rr := httptest.NewRecorder()
 127         router.ServeHTTP(rr, req)
 128
 129         controllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
 130 }
 131
 132 func TestRoutePutNodebResetRanName(t *testing.T) {
 133         router, controllerMock, _ := setupRouterAndMocks()
 134
 135         req, err := http.NewRequest("PUT", "/v1/nodeb/ran1/reset", nil)
 136         if err != nil {
 137                 t.Fatal(err)
 138         }
 139         rr := httptest.NewRecorder()
 140         router.ServeHTTP(rr, req)
 141
 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         controllerMock.AssertNumberOfCalls(t, "X2Reset", 1)
 145 }
 146
 147 func TestRouteNotFound(t *testing.T) {
 148         router, _, _ := setupRouterAndMocks()
 149
 150         req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
 151         if err != nil {
 152                 t.Fatal(err)
 153         }
 154         rr := httptest.NewRecorder()
 155         router.ServeHTTP(rr, req)
 156
 157         assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
 158 }
 159
 160 func TestRunError(t *testing.T) {
 161         log := initLog(t)
 162         err := Run(log, 1234567, &mocks.NodebControllerMock{}, &mocks.ControllerMock{})
 163         assert.NotNil(t, err)
 164 }
 165
 166 func TestRun(t *testing.T) {
 167         log := initLog(t)
 168         _, controllerMock, nodebControllerMock := setupRouterAndMocks()
 169         go Run(log,11223, nodebControllerMock, controllerMock)
 170
 171         time.Sleep(time.Millisecond * 100)
 172         resp, err := http.Get("http://localhost:11223/v1/health")
 173         if err != nil {
 174                 t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
 175         }
 176         assert.Equal(t, 200, resp.StatusCode)
 177 }
 178
 179 func initLog(t *testing.T) *logger.Logger {
 180         log, err := logger.InitLogger(logger.InfoLevel)
 181         if err != nil {
 182                 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)
 183         }
 184         return log
 185 }