2 //// Copyright 2019 AT&T Intellectual Property
3 //// Copyright 2019 Nokia
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
9 //// http://www.apache.org/licenses/LICENSE-2.0
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.
24 "github.com/pkg/errors"
28 func TestLostConnectionHandlerSuccess(t *testing.T) {
29 logger, _ := logger.InitLogger(logger.InfoLevel)
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), ¬ificationRequest, nil)
36 ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)
39 func TestLostConnectionHandlerFailure(t *testing.T) {
40 logger, _ := logger.InitLogger(logger.InfoLevel)
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), ¬ificationRequest, nil)
47 ranReconnectionManagerMock.AssertCalled(t, "ReconnectRan", ranName)