E2Manager/handlers/rmrmsghandlers/ran_lost_connection_handler_test.go

   1 //// Copyright 2019 AT&T Intellectual Property
   2 //// Copyright 2019 Nokia
   3 ////
   4 //// Licensed under the Apache License, Version 2.0 (the "License");
   5 //// you may not use this file except in compliance with the License.
   6 //// You may obtain a copy of the License at
   7 ////
   8 ////      http://www.apache.org/licenses/LICENSE-2.0
   9 ////
  10 //// Unless required by applicable law or agreed to in writing, software
  11 //// distributed under the License is distributed on an "AS IS" BASIS,
  12 //// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 //// See the License for the specific language governing permissions and
  14 //// limitations under the License.
  15
  16 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  17 //  platform project (RICP).
  18
  19 ////
  20 package rmrmsghandlers
  21
  22 import (
  23         "e2mgr/logger"
  24         "e2mgr/mocks"
  25         "e2mgr/models"
  26         "github.com/pkg/errors"
  27         "testing"
  28 )
  29
  30 func TestLostConnectionHandlerSuccess(t *testing.T) {
  31         logger, _ := logger.InitLogger(logger.InfoLevel)
  32         ranName := "test"
  33         notificationRequest := models.NotificationRequest{RanName: ranName}
  34         ranReconnectionManagerMock := &mocks.RanReconnectionManagerMock{}
  35         ranReconnectionManagerMock.On("ReconnectRan", ranName).Return(nil)
  36         handler := NewRanLostConnectionHandler(logger, ranReconnectionManagerMock)
  37         handler.Handle(&notificationRequest)
  38         ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)
  39 }
  40
  41 func TestLostConnectionHandlerFailure(t *testing.T) {
  42         logger, _ := logger.InitLogger(logger.InfoLevel)
  43         ranName := "test"
  44         notificationRequest := models.NotificationRequest{RanName: ranName}
  45         ranReconnectionManagerMock := &mocks.RanReconnectionManagerMock{}
  46         ranReconnectionManagerMock.On("ReconnectRan", ranName).Return(errors.New("error"))
  47         handler := NewRanLostConnectionHandler(logger, ranReconnectionManagerMock)
  48         handler.Handle(&notificationRequest)
  49         ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)
  50 }