controller_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 controllers
  19
  20 import (
  21         "bytes"
  22         "e2mgr/configuration"
  23         "e2mgr/e2managererrors"
  24         "e2mgr/logger"
  25         "e2mgr/mocks"
  26         "e2mgr/models"
  27         "e2mgr/providers"
  28         "e2mgr/rNibWriter"
  29         "e2mgr/rmrCgo"
  30         "e2mgr/tests"
  31         "encoding/json"
  32         "fmt"
  33         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
  34         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  35         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
  36         "github.com/julienschmidt/httprouter"
  37         "github.com/stretchr/testify/assert"
  38         "github.com/stretchr/testify/mock"
  39         "io"
  40         "io/ioutil"
  41         "net/http"
  42         "net/http/httptest"
  43         "strings"
  44         "testing"
  45 )
  46
  47 func TestShutdownHandlerRnibError(t *testing.T) {
  48         log := initLog(t)
  49         config := configuration.ParseConfiguration()
  50
  51         rmrMessengerMock := &mocks.RmrMessengerMock{}
  52         readerMock := &mocks.RnibReaderMock{}
  53         readerProvider := func() reader.RNibReader {
  54                 return readerMock
  55         }
  56         writerMock := &mocks.RnibWriterMock{}
  57         writerProvider := func() rNibWriter.RNibWriter {
  58                 return writerMock
  59         }
  60
  61         rnibErr := &common.RNibError{}
  62         var nbIdentityList []*entities.NbIdentity
  63         readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
  64
  65         writer := httptest.NewRecorder()
  66         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
  67         controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
  68
  69         var errorResponse = parseJsonRequest(t, writer.Body)
  70
  71         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
  72         assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Err.Code)
  73 }
  74
  75 func TestHeaderValidationFailed(t *testing.T) {
  76         log := initLog(t)
  77         config := configuration.ParseConfiguration()
  78
  79         rmrMessengerMock := &mocks.RmrMessengerMock{}
  80         readerMock := &mocks.RnibReaderMock{}
  81         readerProvider := func() reader.RNibReader {
  82                 return readerMock
  83         }
  84         writerMock := &mocks.RnibWriterMock{}
  85         writerProvider := func() rNibWriter.RNibWriter {
  86                 return writerMock
  87         }
  88
  89         writer := httptest.NewRecorder()
  90
  91         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
  92
  93         header := &http.Header{}
  94
  95         controller.handleRequest(writer, header, providers.ShutdownRequest, nil, true, http.StatusNoContent)
  96
  97         var errorResponse = parseJsonRequest(t, writer.Body)
  98         err := e2managererrors.NewHeaderValidationError()
  99
 100         assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
 101         assert.Equal(t, errorResponse.Code, err.Err.Code)
 102         assert.Equal(t, errorResponse.Message, err.Err.Message)
 103 }
 104
 105 func TestShutdownStatusNoContent(t *testing.T){
 106         log := initLog(t)
 107
 108         rmrMessengerMock := &mocks.RmrMessengerMock{}
 109         readerMock := &mocks.RnibReaderMock{}
 110         readerProvider := func() reader.RNibReader {
 111                 return readerMock
 112         }
 113         writerMock := &mocks.RnibWriterMock{}
 114         writerProvider := func() rNibWriter.RNibWriter {
 115                 return writerMock
 116         }
 117         config := configuration.ParseConfiguration()
 118
 119         var rnibError common.IRNibError
 120         nbIdentityList := []*entities.NbIdentity{}
 121         readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
 122
 123         writer := httptest.NewRecorder()
 124         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 125         controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
 126
 127         assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
 128 }
 129
 130 func TestHandleInternalError(t *testing.T) {
 131         log := initLog(t)
 132         config := configuration.ParseConfiguration()
 133
 134         rmrMessengerMock := &mocks.RmrMessengerMock{}
 135         readerMock := &mocks.RnibReaderMock{}
 136         readerProvider := func() reader.RNibReader {
 137                 return readerMock
 138         }
 139         writerMock := &mocks.RnibWriterMock{}
 140         writerProvider := func() rNibWriter.RNibWriter {
 141                 return writerMock
 142         }
 143
 144         writer := httptest.NewRecorder()
 145         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 146         err := e2managererrors.NewInternalError()
 147
 148         controller.handleErrorResponse(err, writer)
 149         var errorResponse = parseJsonRequest(t, writer.Body)
 150
 151         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
 152         assert.Equal(t, errorResponse.Code, err.Err.Code)
 153         assert.Equal(t, errorResponse.Message, err.Err.Message)
 154 }
 155
 156 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
 157         log := initLog(t)
 158         config := configuration.ParseConfiguration()
 159
 160         rmrMessengerMock := &mocks.RmrMessengerMock{}
 161         readerMock := &mocks.RnibReaderMock{}
 162         readerProvider := func() reader.RNibReader {
 163                 return readerMock
 164         }
 165         writerMock := &mocks.RnibWriterMock{}
 166         writerProvider := func() rNibWriter.RNibWriter {
 167                 return writerMock
 168         }
 169         writer := httptest.NewRecorder()
 170         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 171         err := e2managererrors.NewCommandAlreadyInProgressError()
 172
 173         controller.handleErrorResponse(err, writer)
 174         var errorResponse = parseJsonRequest(t, writer.Body)
 175
 176         assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
 177         assert.Equal(t, errorResponse.Code, err.Err.Code)
 178         assert.Equal(t, errorResponse.Message, err.Err.Message)
 179 }
 180
 181 func TestValidateHeaders(t *testing.T){
 182         log := initLog(t)
 183
 184         rmrMessengerMock := &mocks.RmrMessengerMock{}
 185         readerMock := &mocks.RnibReaderMock{}
 186         readerProvider := func() reader.RNibReader {
 187                 return readerMock
 188         }
 189         writerMock := &mocks.RnibWriterMock{}
 190         writerProvider := func() rNibWriter.RNibWriter {
 191                 return writerMock
 192         }
 193         config := configuration.ParseConfiguration()
 194
 195         controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 196
 197         header := http.Header{}
 198         header.Set("Content-Type", "application/json")
 199         result := controller.validateRequestHeader(&header)
 200
 201         assert.Nil(t, result)
 202 }
 203
 204 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
 205
 206         var errorResponse models.ErrorResponse
 207         body, err := ioutil.ReadAll(r)
 208         if err != nil {
 209                 t.Errorf("Error cannot deserialize json request")
 210         }
 211         json.Unmarshal(body, &errorResponse)
 212
 213         return errorResponse
 214 }
 215
 216 func initLog(t *testing.T) *logger.Logger {
 217         log, err := logger.InitLogger(logger.InfoLevel)
 218         if err != nil {
 219                 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
 220         }
 221         return log
 222 }
 223
 224
 225 func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T){
 226         log := initLog(t)
 227
 228         ranName := "test1"
 229
 230         readerMock := &mocks.RnibReaderMock{}
 231         readerProvider := func() reader.RNibReader {
 232                 return readerMock
 233         }
 234         writerMock := &mocks.RnibWriterMock{}
 235         writerProvider := func() rNibWriter.RNibWriter {
 236                 return writerMock
 237         }
 238         payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
 239         xaction := []byte(ranName)
 240         msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
 241         rmrMessengerMock := &mocks.RmrMessengerMock{}
 242         rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)
 243
 244         config := configuration.ParseConfiguration()
 245         rmrService:=getRmrService(rmrMessengerMock, log)
 246
 247         writer := httptest.NewRecorder()
 248         controller := NewController(log, rmrService, readerProvider, writerProvider, config)
 249
 250         var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
 251         readerMock.On("GetNodeb",ranName).Return(nodeb, nil)
 252
 253         data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"}
 254         b := new(bytes.Buffer)
 255         _ = json.NewEncoder(b).Encode(data4Req)
 256         req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
 257
 258         param:= httprouter.Param{Key:"ranName", Value: ranName}
 259         controller.X2ResetHandler(writer, req, []httprouter.Param {param})
 260         assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
 261
 262         v, ok := rmrService.E2sessions[ranName]
 263         assert.True(t, ok)
 264
 265         assert.Equal(t, v.Request.RanName, ranName)
 266 }
 267
 268 func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T){
 269         log := initLog(t)
 270
 271         ranName := "test1"
 272
 273         readerMock := &mocks.RnibReaderMock{}
 274         readerProvider := func() reader.RNibReader {
 275                 return readerMock
 276         }
 277         writerMock := &mocks.RnibWriterMock{}
 278         writerProvider := func() rNibWriter.RNibWriter {
 279                 return writerMock
 280         }
 281         // o&m intervention
 282         payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
 283         xaction := []byte(ranName)
 284         msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
 285         rmrMessengerMock := &mocks.RmrMessengerMock{}
 286         rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)
 287
 288         config := configuration.ParseConfiguration()
 289         rmrService:=getRmrService(rmrMessengerMock, log)
 290
 291         writer := httptest.NewRecorder()
 292         controller := NewController(log, rmrService, readerProvider, writerProvider, config)
 293
 294         var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
 295         readerMock.On("GetNodeb",ranName).Return(nodeb, nil)
 296
 297         // no body
 298         b := new(bytes.Buffer)
 299         req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
 300
 301         param:= httprouter.Param{Key:"ranName", Value: ranName}
 302         controller.X2ResetHandler(writer, req, []httprouter.Param {param})
 303         assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
 304
 305         v, ok := rmrService.E2sessions[ranName]
 306         assert.True(t, ok)
 307
 308         assert.Equal(t, v.Request.RanName, ranName)
 309 }
 310
 311 func TestX2ResetHandleFailureInvalidBody(t *testing.T){
 312         log := initLog(t)
 313
 314         ranName := "test1"
 315
 316         readerMock := &mocks.RnibReaderMock{}
 317         readerProvider := func() reader.RNibReader {
 318                 return readerMock
 319         }
 320         writerMock := &mocks.RnibWriterMock{}
 321         writerProvider := func() rNibWriter.RNibWriter {
 322                 return writerMock
 323         }
 324         rmrMessengerMock := &mocks.RmrMessengerMock{}
 325
 326         config := configuration.ParseConfiguration()
 327         rmrService:=getRmrService(rmrMessengerMock, log)
 328
 329         writer := httptest.NewRecorder()
 330         controller := NewController(log, rmrService, readerProvider, writerProvider, config)
 331
 332         // Invalid json: attribute name without quotes (should be "cause":).
 333         b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"")
 334         req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
 335
 336         param:= httprouter.Param{Key:"ranName", Value: ranName}
 337         controller.X2ResetHandler(writer, req, []httprouter.Param {param})
 338         assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
 339
 340         _, ok := rmrService.E2sessions[ranName]
 341         assert.False(t, ok)
 342
 343 }
 344
 345 func TestHandleErrorResponse(t *testing.T){
 346         log := initLog(t)
 347
 348         readerMock := &mocks.RnibReaderMock{}
 349         readerProvider := func() reader.RNibReader {
 350                 return readerMock
 351         }
 352         writerMock := &mocks.RnibWriterMock{}
 353         writerProvider := func() rNibWriter.RNibWriter {
 354                 return writerMock
 355         }
 356         rmrMessengerMock := &mocks.RmrMessengerMock{}
 357
 358         config := configuration.ParseConfiguration()
 359         rmrService:=getRmrService(rmrMessengerMock, log)
 360
 361
 362         controller := NewController(log, rmrService, readerProvider, writerProvider, config)
 363
 364         writer := httptest.NewRecorder()
 365         controller.handleErrorResponse(e2managererrors.NewRnibDbError(),writer)
 366         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
 367
 368         writer = httptest.NewRecorder()
 369         controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(),writer)
 370         assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
 371
 372         writer = httptest.NewRecorder()
 373         controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(),writer)
 374         assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
 375
 376         writer = httptest.NewRecorder()
 377         controller.handleErrorResponse(e2managererrors.NewWrongStateError(""),writer)
 378         assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
 379
 380         writer = httptest.NewRecorder()
 381         controller.handleErrorResponse(e2managererrors.NewRequestValidationError(),writer)
 382         assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
 383
 384         writer = httptest.NewRecorder()
 385         controller.handleErrorResponse(e2managererrors.NewRmrError(),writer)
 386         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
 387
 388         writer = httptest.NewRecorder()
 389         controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(),writer)
 390         assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode)
 391
 392         writer = httptest.NewRecorder()
 393         controller.handleErrorResponse(fmt.Errorf("ErrorError"),writer)
 394         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
 395 }