E2Manager/controllers/root_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         "e2mgr/configuration"
  22         "e2mgr/logger"
  23         "e2mgr/mocks"
  24         "e2mgr/services"
  25         "fmt"
  26         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
  27         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  28         "github.com/stretchr/testify/assert"
  29         "net"
  30         "net/http"
  31         "net/http/httptest"
  32         "testing"
  33 )
  34
  35 func setupNodebControllerTest(t *testing.T) (services.RNibDataService, *mocks.RnibReaderMock){
  36         logger, err := logger.InitLogger(logger.DebugLevel)
  37         if err != nil {
  38                 t.Errorf("#... - failed to initialize logger, error: %s", err)
  39         }
  40         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
  41         readerMock := &mocks.RnibReaderMock{}
  42
  43         writerMock := &mocks.RnibWriterMock{}
  44
  45         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
  46         return rnibDataService, readerMock
  47 }
  48
  49 func TestNewRequestController(t *testing.T) {
  50         rnibDataService, _ := setupNodebControllerTest(t)
  51         assert.NotNil(t, NewRootController(rnibDataService))
  52 }
  53
  54 func TestHandleHealthCheckRequestGood(t *testing.T) {
  55         rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
  56
  57         var nbList []*entities.NbIdentity
  58         rnibReaderMock.On("GetListNodebIds").Return(nbList, nil)
  59
  60         rc := NewRootController(rnibDataService)
  61         writer := httptest.NewRecorder()
  62         rc.HandleHealthCheckRequest(writer, nil)
  63         assert.Equal(t, http.StatusOK, writer.Result().StatusCode)
  64 }
  65
  66 func TestHandleHealthCheckRequestOtherError(t *testing.T) {
  67         rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
  68
  69         mockOtherErr := &common.InternalError{Err: fmt.Errorf("non connection error")}
  70         var nbList []*entities.NbIdentity
  71         rnibReaderMock.On("GetListNodebIds").Return(nbList, mockOtherErr)
  72
  73         rc := NewRootController(rnibDataService)
  74         writer := httptest.NewRecorder()
  75         rc.HandleHealthCheckRequest(writer, nil)
  76         assert.Equal(t, http.StatusOK, writer.Result().StatusCode)
  77 }
  78
  79 func TestHandleHealthCheckRequestConnError(t *testing.T) {
  80         rnibDataService, rnibReaderMock := setupNodebControllerTest(t)
  81
  82         mockConnErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
  83         var nbList []*entities.NbIdentity
  84         rnibReaderMock.On("GetListNodebIds").Return(nbList, mockConnErr)
  85
  86
  87         rc := NewRootController(rnibDataService)
  88         writer := httptest.NewRecorder()
  89         rc.HandleHealthCheckRequest(writer, nil)
  90         assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
  91 }