E2Manager/handlers/ran_lost_connection_handler_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 package handlers
  18
  19 import (
  20         "e2mgr/logger"
  21         "e2mgr/mocks"
  22         "e2mgr/models"
  23         "e2mgr/sessions"
  24         "github.com/pkg/errors"
  25         "testing"
  26 )
  27
  28 func TestLostConnectionHandlerSuccess(t *testing.T) {
  29         logger, _ := logger.InitLogger(logger.InfoLevel)
  30         ranName := "test"
  31         notificationRequest := models.NotificationRequest{RanName: ranName}
  32         ranReconnectionManagerMock := &mocks.RanReconnectionManagerMock{}
  33         ranReconnectionManagerMock.On("ReconnectRan", ranName).Return(nil)
  34         handler := NewRanLostConnectionHandler(ranReconnectionManagerMock)
  35         handler.Handle(logger, make(sessions.E2Sessions), &notificationRequest, nil)
  36         ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)
  37 }
  38
  39 func TestLostConnectionHandlerFailure(t *testing.T) {
  40         logger, _ := logger.InitLogger(logger.InfoLevel)
  41         ranName := "test"
  42         notificationRequest := models.NotificationRequest{RanName: ranName}
  43         ranReconnectionManagerMock := &mocks.RanReconnectionManagerMock{}
  44         ranReconnectionManagerMock.On("ReconnectRan", ranName).Return(errors.New("error"))
  45         handler := NewRanLostConnectionHandler(ranReconnectionManagerMock)
  46         handler.Handle(logger, make(sessions.E2Sessions), &notificationRequest, nil)
  47         ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)
  48 }