E2Manager/httpserver/http_server_test.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   4 // Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved.
   5 //
   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
   9 //
  10 //      http://www.apache.org/licenses/LICENSE-2.0
  11 //
  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.
  17
  18 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  19 //  platform project (RICP).
  20
  21 package httpserver
  22
  23 import (
  24         "e2mgr/logger"
  25         "e2mgr/mocks"
  26         "github.com/gorilla/mux"
  27         "github.com/stretchr/testify/assert"
  28         "net/http"
  29         "net/http/httptest"
  30         "testing"
  31         "time"
  32 )
  33
  34 func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock, *mocks.NodebControllerMock, *mocks.E2TControllerMock, *mocks.SymptomdataControllerMock) {
  35         rootControllerMock := &mocks.RootControllerMock{}
  36         rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
  37
  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)
  48
  49         e2tControllerMock := &mocks.E2TControllerMock{}
  50         e2tControllerMock.On("GetE2TInstances").Return(nil)
  51
  52         symptomdataControllerMock := &mocks.SymptomdataControllerMock{}
  53         symptomdataControllerMock.On("GetSymptomData").Return(nil)
  54
  55         router := mux.NewRouter()
  56         initializeRoutes(router, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock)
  57         return router, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock
  58 }
  59
  60 func TestRouteGetNodebIdList(t *testing.T) {
  61         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
  62
  63         req, err := http.NewRequest("GET", "/v1/nodeb/states", nil)
  64         if err != nil {
  65                 t.Fatal(err)
  66         }
  67         rr := httptest.NewRecorder()
  68         router.ServeHTTP(rr, req)
  69
  70         nodebControllerMock.AssertNumberOfCalls(t, "GetNodebIdList", 1)
  71 }
  72
  73 func TestRouteGetNodebId(t *testing.T) {
  74         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
  75
  76         req, err := http.NewRequest("GET", "/v1/nodeb/states/ran1", nil)
  77         if err != nil {
  78                 t.Fatal(err)
  79         }
  80         rr := httptest.NewRecorder()
  81         router.ServeHTTP(rr, req)
  82
  83         assert.Equal(t, http.StatusOK, rr.Code, "handler returned wrong status code")
  84         nodebControllerMock.AssertNumberOfCalls(t, "GetNodebId", 1)
  85 }
  86
  87 func TestRouteGetNodebRanName(t *testing.T) {
  88         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
  89
  90         req, err := http.NewRequest("GET", "/v1/nodeb/ran1", nil)
  91         if err != nil {
  92                 t.Fatal(err)
  93         }
  94         rr := httptest.NewRecorder()
  95         router.ServeHTTP(rr, req)
  96
  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)
 100 }
 101
 102 func TestRouteGetHealth(t *testing.T) {
 103         router, rootControllerMock, _, _, _ := setupRouterAndMocks()
 104
 105         req, err := http.NewRequest("GET", "/v1/health", nil)
 106         if err != nil {
 107                 t.Fatal(err)
 108         }
 109         rr := httptest.NewRecorder()
 110         router.ServeHTTP(rr, req)
 111
 112         rootControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
 113 }
 114
 115 func TestRoutePutNodebShutdown(t *testing.T) {
 116         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 117
 118         req, err := http.NewRequest("PUT", "/v1/nodeb/shutdown", nil)
 119         if err != nil {
 120                 t.Fatal(err)
 121         }
 122         rr := httptest.NewRecorder()
 123         router.ServeHTTP(rr, req)
 124
 125         nodebControllerMock.AssertNumberOfCalls(t, "Shutdown", 1)
 126 }
 127
 128 func TestHealthCheckRequest(t *testing.T) {
 129         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 130
 131         req, err := http.NewRequest("PUT", "/v1/nodeb/health", nil)
 132         if err != nil {
 133                 t.Fatal(err)
 134         }
 135         rr := httptest.NewRecorder()
 136         router.ServeHTTP(rr, req)
 137
 138         assert.Equal(t, http.StatusAccepted, rr.Code, "handler returned wrong status code")
 139         nodebControllerMock.AssertNumberOfCalls(t, "HealthCheckRequest", 1)
 140 }
 141
 142 func TestRoutePutNodebSetGeneralConfiguration(t *testing.T) {
 143         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 144
 145         req, err := http.NewRequest("PUT", "/v1/nodeb/parameters", nil)
 146         if err != nil {
 147                 t.Fatal(err)
 148         }
 149         rr := httptest.NewRecorder()
 150         router.ServeHTTP(rr, req)
 151
 152         nodebControllerMock.AssertNumberOfCalls(t, "SetGeneralConfiguration", 1)
 153 }
 154
 155 func TestRoutePutUpdateEnb(t *testing.T) {
 156         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 157
 158         req, err := http.NewRequest("PUT", "/v1/nodeb/enb/ran1", nil)
 159         if err != nil {
 160                 t.Fatal(err)
 161         }
 162         rr := httptest.NewRecorder()
 163         router.ServeHTTP(rr, req)
 164
 165         nodebControllerMock.AssertNumberOfCalls(t, "UpdateEnb", 1)
 166 }
 167
 168 func TestRouteNotFound(t *testing.T) {
 169         router, _, _, _, _ := setupRouterAndMocks()
 170
 171         req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
 172         if err != nil {
 173                 t.Fatal(err)
 174         }
 175         rr := httptest.NewRecorder()
 176         router.ServeHTTP(rr, req)
 177
 178         assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
 179 }
 180
 181 func TestRunError(t *testing.T) {
 182         log := initLog(t)
 183         err := Run(log, 1234567, &mocks.RootControllerMock{}, &mocks.NodebControllerMock{}, &mocks.E2TControllerMock{}, &mocks.SymptomdataControllerMock{})
 184         assert.NotNil(t, err)
 185 }
 186
 187 func TestRun(t *testing.T) {
 188         log := initLog(t)
 189         _, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock := setupRouterAndMocks()
 190         go Run(log, 11223, rootControllerMock, nodebControllerMock, e2tControllerMock, symptomdataControllerMock)
 191
 192         time.Sleep(time.Millisecond * 100)
 193         resp, err := http.Get("http://localhost:11223/v1/health")
 194         if err != nil {
 195                 t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
 196         }
 197         assert.Equal(t, 200, resp.StatusCode)
 198 }
 199
 200 func TestRouteAddEnb(t *testing.T) {
 201         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 202
 203         req, err := http.NewRequest("POST", "/v1/nodeb/enb", nil)
 204         if err != nil {
 205                 t.Fatal(err)
 206         }
 207         rr := httptest.NewRecorder()
 208         router.ServeHTTP(rr, req)
 209
 210         assert.Equal(t, http.StatusCreated, rr.Code, "handler returned wrong status code")
 211         nodebControllerMock.AssertNumberOfCalls(t, "AddEnb", 1)
 212 }
 213
 214 func TestRouteDeleteEnb(t *testing.T) {
 215         router, _, nodebControllerMock, _, _ := setupRouterAndMocks()
 216
 217         req, err := http.NewRequest("DELETE", "/v1/nodeb/enb/ran1", nil)
 218         if err != nil {
 219                 t.Fatal(err)
 220         }
 221         rr := httptest.NewRecorder()
 222         router.ServeHTTP(rr, req)
 223
 224         assert.Equal(t, http.StatusNoContent, rr.Code, "handler returned wrong status code")
 225         nodebControllerMock.AssertNumberOfCalls(t, "DeleteEnb", 1)
 226 }
 227
 228 func initLog(t *testing.T) *logger.Logger {
 229         log, err := logger.InitLogger(logger.InfoLevel)
 230         if err != nil {
 231                 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)
 232         }
 233         return log
 234 }