E2Manager/handlers/rmrmsghandlers/e2_term_init_notification_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 rmrmsghandlers
  18
  19 import (
  20         "e2mgr/configuration"
  21         "e2mgr/e2pdus"
  22         "e2mgr/logger"
  23         "e2mgr/managers"
  24         "e2mgr/mocks"
  25         "e2mgr/models"
  26         "e2mgr/rmrCgo"
  27         "e2mgr/services"
  28         "e2mgr/services/rmrsender"
  29         "e2mgr/tests"
  30         "fmt"
  31         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
  32         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  33         "github.com/stretchr/testify/mock"
  34         "testing"
  35 )
  36
  37 const e2tInstanceAddress = "10.0.2.15"
  38 const e2tInitPayload = "{\"address\":\"10.0.2.15\", \"fqdn\":\"\"}"
  39
  40 func initRanLostConnectionTest(t *testing.T) (*logger.Logger, E2TermInitNotificationHandler, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock, *mocks.E2TInstancesManagerMock) {
  41
  42         logger := initLog(t)
  43         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
  44
  45         rmrMessengerMock := &mocks.RmrMessengerMock{}
  46         rmrSender := initRmrSender(rmrMessengerMock, logger)
  47
  48         readerMock := &mocks.RnibReaderMock{}
  49
  50         writerMock := &mocks.RnibWriterMock{}
  51
  52         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
  53         ranSetupManager := managers.NewRanSetupManager(logger, rmrSender, rnibDataService)
  54
  55         e2tInstancesManagerMock := &mocks.E2TInstancesManagerMock{}
  56         ranReconnectionManager := managers.NewRanReconnectionManager(logger, configuration.ParseConfiguration(), rnibDataService, ranSetupManager, e2tInstancesManagerMock)
  57         handler := NewE2TermInitNotificationHandler(logger, ranReconnectionManager, rnibDataService, e2tInstancesManagerMock)
  58         return logger, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock
  59 }
  60
  61 func initRanLostConnectionTestWithRealE2tInstanceManager(t *testing.T) (*logger.Logger, E2TermInitNotificationHandler, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock) {
  62
  63         logger := initLog(t)
  64         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
  65
  66         rmrMessengerMock := &mocks.RmrMessengerMock{}
  67         rmrSender := initRmrSender(rmrMessengerMock, logger)
  68
  69         readerMock := &mocks.RnibReaderMock{}
  70
  71         writerMock := &mocks.RnibWriterMock{}
  72
  73         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
  74         ranSetupManager := managers.NewRanSetupManager(logger, rmrSender, rnibDataService)
  75
  76         e2tInstancesManager := managers.NewE2TInstancesManager(rnibDataService, logger)
  77         ranReconnectionManager := managers.NewRanReconnectionManager(logger, configuration.ParseConfiguration(), rnibDataService, ranSetupManager, e2tInstancesManager)
  78         handler := NewE2TermInitNotificationHandler(logger, ranReconnectionManager, rnibDataService, e2tInstancesManager)
  79         return logger, handler, readerMock, writerMock, rmrMessengerMock
  80 }
  81
  82 func TestE2TermInitUnmarshalPayloadFailure(t *testing.T) {
  83         _, handler, _, _, _, e2tInstancesManagerMock := initRanLostConnectionTest(t)
  84         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte("asd")}
  85         handler.Handle(notificationRequest)
  86         e2tInstancesManagerMock.AssertNotCalled(t, "GetE2TInstance")
  87         e2tInstancesManagerMock.AssertNotCalled(t, "AddE2TInstance")
  88 }
  89
  90 func TestE2TermInitEmptyE2TAddress(t *testing.T) {
  91         _, handler, _, _, _, e2tInstancesManagerMock := initRanLostConnectionTest(t)
  92         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte("{\"address\":\"\"}")}
  93         handler.Handle(notificationRequest)
  94         e2tInstancesManagerMock.AssertNotCalled(t, "GetE2TInstance")
  95         e2tInstancesManagerMock.AssertNotCalled(t, "AddE2TInstance")
  96 }
  97
  98 func TestE2TermInitGetE2TInstanceFailure(t *testing.T) {
  99         _, handler, _, _, _, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 100         var e2tInstance *entities.E2TInstance
 101         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, common.NewInternalError(fmt.Errorf("internal error")))
 102         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 103         handler.Handle(notificationRequest)
 104         e2tInstancesManagerMock.AssertNotCalled(t, "AddE2TInstance")
 105 }
 106
 107 func TestE2TermInitGetE2TInstanceDbFailure(t *testing.T) {
 108         _, handler, readerMock, writerMock, rmrMessengerMock := initRanLostConnectionTestWithRealE2tInstanceManager(t)
 109         var e2tInstance *entities.E2TInstance
 110         readerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, common.NewInternalError(fmt.Errorf("internal error")))
 111         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 112         handler.Handle(notificationRequest)
 113         writerMock.AssertNotCalled(t, "UpdateNodebInfo")
 114         rmrMessengerMock.AssertNotCalled(t, "SendMsg")
 115 }
 116
 117 func TestE2TermInitNewE2TInstance(t *testing.T) {
 118         _, handler, _, _, _, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 119         var e2tInstance *entities.E2TInstance
 120         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, common.NewResourceNotFoundError("not found"))
 121         e2tInstance = entities.NewE2TInstance(e2tInstanceAddress)
 122         e2tInstancesManagerMock.On("AddE2TInstance", e2tInstanceAddress).Return(nil)
 123         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 124         handler.Handle(notificationRequest)
 125         e2tInstancesManagerMock.AssertCalled(t, "AddE2TInstance", e2tInstanceAddress)
 126 }
 127
 128 func TestE2TermInitExistingE2TInstanceNoAssociatedRans(t *testing.T) {
 129         _, handler, _, _, _, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 130         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 131         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 132         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 133         handler.Handle(notificationRequest)
 134         e2tInstancesManagerMock.AssertCalled(t, "GetE2TInstance", e2tInstanceAddress)
 135 }
 136
 137 func TestE2TermInitHandlerSuccessOneRan(t *testing.T) {
 138         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 139         var rnibErr error
 140
 141         var initialNodeb = &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 142         readerMock.On("GetNodeb", RanName).Return(initialNodeb, rnibErr)
 143
 144         var argNodeb = &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 145         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 146
 147         payload := e2pdus.PackedX2setupRequest
 148         xaction := []byte(RanName)
 149         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), RanName, &payload, &xaction)
 150
 151         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
 152
 153         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 154         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, RanName)
 155         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 156         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 157
 158         handler.Handle(notificationRequest)
 159
 160         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 161         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 162 }
 163
 164 func TestE2TermInitHandlerSuccessOneRanShuttingdown(t *testing.T) {
 165         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 166         var rnibErr error
 167
 168         var initialNodeb = &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 169         readerMock.On("GetNodeb", RanName).Return(initialNodeb, rnibErr)
 170
 171         var argNodeb = &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
 172         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 173
 174         payload := e2pdus.PackedX2setupRequest
 175         xaction := []byte(RanName)
 176         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), RanName, &payload, &xaction)
 177
 178         rmrMessengerMock.On("SendMsg", mock.Anything).Return(msg, nil)
 179
 180         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 181         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, RanName)
 182         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 183         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 184
 185         handler.Handle(notificationRequest)
 186
 187         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 188         rmrMessengerMock.AssertNotCalled(t, "SendMsg")
 189 }
 190
 191 func TestE2TermInitHandlerSuccessTwoRans(t *testing.T) {
 192         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 193         var rnibErr error
 194         var initialNodeb0 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 195         var initialNodeb1 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 196         readerMock.On("GetNodeb", RanName).Return(initialNodeb0, rnibErr)
 197         readerMock.On("GetNodeb", "test2").Return(initialNodeb1, rnibErr)
 198
 199         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 200         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 201
 202         payload := e2pdus.PackedX2setupRequest
 203         xaction := []byte(RanName)
 204         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), RanName, &payload, &xaction)
 205
 206         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
 207
 208         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 209         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, RanName, "test2")
 210         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 211         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 212
 213         handler.Handle(notificationRequest)
 214
 215         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 216         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 2)
 217 }
 218
 219 func TestE2TermInitHandlerSuccessTwoRansSecondRanShutdown(t *testing.T) {
 220         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 221         var rnibErr error
 222         var initialNodeb0 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 223         var initialNodeb1 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 224         readerMock.On("GetNodeb", RanName).Return(initialNodeb0, rnibErr)
 225         readerMock.On("GetNodeb", "test2").Return(initialNodeb1, rnibErr)
 226
 227         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 228         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 229
 230         payload := e2pdus.PackedX2setupRequest
 231         xaction := []byte(RanName)
 232         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), RanName, &payload, &xaction)
 233
 234         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
 235
 236         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 237         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, RanName, "test2")
 238         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 239         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 240
 241         handler.Handle(notificationRequest)
 242
 243         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 244         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 245         writerMock.AssertExpectations(t)
 246 }
 247
 248 func TestE2TermInitHandlerSuccessThreeRansFirstRmrFailure(t *testing.T) {
 249         log, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 250         var rnibErr error
 251
 252         ids := []*entities.NbIdentity{{InventoryName: "test1"}, {InventoryName: "test2"}, {InventoryName: "test3"}}
 253
 254         var initialNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 255         var initialNodeb1 = &entities.NodebInfo{RanName: ids[1].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 256         var initialNodeb2 = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 257         readerMock.On("GetNodeb", ids[0].InventoryName).Return(initialNodeb0, rnibErr)
 258         readerMock.On("GetNodeb", ids[1].InventoryName).Return(initialNodeb1, rnibErr)
 259         readerMock.On("GetNodeb", ids[2].InventoryName).Return(initialNodeb2, rnibErr)
 260
 261         var argNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 262         var argNodeb0Fail = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
 263         writerMock.On("UpdateNodebInfo", argNodeb0).Return(rnibErr)
 264         writerMock.On("UpdateNodebInfo", argNodeb0Fail).Return(rnibErr)
 265
 266         payload := models.NewE2RequestMessage(ids[0].InventoryName /*tid*/, "", 0, ids[0].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 267         xaction := []byte(ids[0].InventoryName)
 268         msg0 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[0].InventoryName, &payload, &xaction)
 269
 270         // Cannot use Mock because request MBuf contains pointers
 271         //payload =models.NewE2RequestMessage(ids[1].InventoryName /*tid*/, "", 0,ids[1].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 272         //xaction = []byte(ids[1].InventoryName)
 273         //msg1 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[1].InventoryName, &payload, &xaction)
 274
 275         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg0, fmt.Errorf("RMR Error"))
 276
 277         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 278         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, "test1", "test2", "test3")
 279         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 280         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 281
 282         handler.Handle(notificationRequest)
 283
 284         //test1 (before send +1, after failure +1), test2 (0) test3 (0)
 285         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 286         //test1 failure (+1), test2  (0). test3 (0)
 287         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 288 }
 289
 290 func TestE2TermInitHandlerSuccessThreeRansSecondNotFoundFailure(t *testing.T) {
 291         log, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 292         var rnibErr error
 293
 294         ids := []*entities.NbIdentity{{InventoryName: "test1"}, {InventoryName: "test2"}, {InventoryName: "test3"}}
 295
 296         var initialNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 297         var initialNodeb1 = &entities.NodebInfo{RanName: ids[1].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 298         var initialNodeb2 = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 299         readerMock.On("GetNodeb", ids[0].InventoryName).Return(initialNodeb0, rnibErr)
 300         readerMock.On("GetNodeb", ids[1].InventoryName).Return(initialNodeb1, common.NewResourceNotFoundError("not found"))
 301         readerMock.On("GetNodeb", ids[2].InventoryName).Return(initialNodeb2, rnibErr)
 302
 303         var argNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 304         var argNodeb0Success = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 305         writerMock.On("UpdateNodebInfo", argNodeb0).Return(rnibErr)
 306         writerMock.On("UpdateNodebInfo", argNodeb0Success).Return(rnibErr)
 307
 308         var argNodeb2 = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 309         var argNodeb2Success = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 310         writerMock.On("UpdateNodebInfo", argNodeb2).Return(rnibErr)
 311         writerMock.On("UpdateNodebInfo", argNodeb2Success).Return(rnibErr)
 312
 313         payload := models.NewE2RequestMessage(ids[0].InventoryName /*tid*/, "", 0, ids[0].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 314         xaction := []byte(ids[0].InventoryName)
 315         msg0 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[0].InventoryName, &payload, &xaction)
 316
 317         // Cannot use Mock because request MBuf contains pointers
 318         //payload =models.NewE2RequestMessage(ids[1].InventoryName /*tid*/, "", 0,ids[1].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 319         //xaction = []byte(ids[1].InventoryName)
 320         //msg1 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[1].InventoryName, &payload, &xaction)
 321
 322         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg0, nil)
 323
 324         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 325         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, "test1", "test2", "test3")
 326         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 327         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 328
 329         handler.Handle(notificationRequest)
 330
 331         readerMock.AssertNumberOfCalls(t, "GetNodeb", 3)
 332         //test1 (+1), test2 failure (0) test3 (+1)
 333         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 334         //test1 success (+1), test2  (0). test3 (+1)
 335         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 2)
 336 }
 337
 338 func TestE2TermInitHandlerSuccessThreeRansSecondRnibInternalErrorFailure(t *testing.T) {
 339         log, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 340         var rnibErr error
 341
 342         ids := []*entities.NbIdentity{{InventoryName: "test1"}, {InventoryName: "test2"}, {InventoryName: "test3"}}
 343
 344         var initialNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 345         var initialNodeb1 = &entities.NodebInfo{RanName: ids[1].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 346         var initialNodeb2 = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 347         readerMock.On("GetNodeb", ids[0].InventoryName).Return(initialNodeb0, rnibErr)
 348         readerMock.On("GetNodeb", ids[1].InventoryName).Return(initialNodeb1, common.NewInternalError(fmt.Errorf("internal error")))
 349         readerMock.On("GetNodeb", ids[2].InventoryName).Return(initialNodeb2, rnibErr)
 350
 351         var argNodeb0 = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 352         var argNodeb0Success = &entities.NodebInfo{RanName: ids[0].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 353         writerMock.On("UpdateNodebInfo", argNodeb0).Return(rnibErr)
 354         writerMock.On("UpdateNodebInfo", argNodeb0Success).Return(rnibErr)
 355
 356         var argNodeb2 = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 357         var argNodeb2Success = &entities.NodebInfo{RanName: ids[2].InventoryName, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 358         writerMock.On("UpdateNodebInfo", argNodeb2).Return(rnibErr)
 359         writerMock.On("UpdateNodebInfo", argNodeb2Success).Return(rnibErr)
 360
 361         payload := models.NewE2RequestMessage(ids[0].InventoryName /*tid*/, "", 0, ids[0].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 362         xaction := []byte(ids[0].InventoryName)
 363         msg0 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[0].InventoryName, &payload, &xaction)
 364
 365         // Cannot use Mock because request MBuf contains pointers
 366         //payload =models.NewE2RequestMessage(ids[1].InventoryName /*tid*/, "", 0,ids[1].InventoryName, e2pdus.PackedX2setupRequest).GetMessageAsBytes(log)
 367         //xaction = []byte(ids[1].InventoryName)
 368         //msg1 := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ids[1].InventoryName, &payload, &xaction)
 369
 370         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg0, nil)
 371
 372         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 373         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, "test1", "test2", "test3")
 374         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 375         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 376
 377         handler.Handle(notificationRequest)
 378
 379         readerMock.AssertNumberOfCalls(t, "GetNodeb", 2)
 380         //test1 (+1), test2 failure (0) test3 (0)
 381         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 382         //test1 success (+1), test2  (0). test3 (+1)
 383         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 384 }
 385
 386 func TestE2TermInitHandlerSuccessZeroRans(t *testing.T) {
 387         _, handler, _, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 388
 389         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 390         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 391         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 392
 393         handler.Handle(notificationRequest)
 394
 395         writerMock.AssertNotCalled(t, "UpdateNodebInfo")
 396         rmrMessengerMock.AssertNotCalled(t, "SendMsg")
 397 }
 398
 399 func TestE2TermInitHandlerFailureGetNodebInternalError(t *testing.T) {
 400         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 401
 402         var nodebInfo *entities.NodebInfo
 403         readerMock.On("GetNodeb", "test1").Return(nodebInfo, common.NewInternalError(fmt.Errorf("internal error")))
 404
 405         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 406         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, "test1")
 407         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 408         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 409         handler.Handle(notificationRequest)
 410
 411         writerMock.AssertNotCalled(t, "UpdateNodebInfo")
 412         rmrMessengerMock.AssertNotCalled(t, "SendMsg")
 413 }
 414
 415 func TestE2TermInitHandlerSuccessTwoRansSecondIsDisconnected(t *testing.T) {
 416         _, handler, readerMock, writerMock, rmrMessengerMock, e2tInstancesManagerMock := initRanLostConnectionTest(t)
 417         var rnibErr error
 418         var initialNodeb0 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 419         var initialNodeb1 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 420         readerMock.On("GetNodeb", RanName).Return(initialNodeb0, rnibErr)
 421         readerMock.On("GetNodeb", "test2").Return(initialNodeb1, rnibErr)
 422
 423         var argNodeb1 = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 424         writerMock.On("UpdateNodebInfo", argNodeb1).Return(rnibErr)
 425
 426         payload := e2pdus.PackedX2setupRequest
 427         xaction := []byte(RanName)
 428         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), RanName, &payload, &xaction)
 429
 430         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
 431
 432         e2tInstance := entities.NewE2TInstance(e2tInstanceAddress)
 433         e2tInstance.AssociatedRanList = append(e2tInstance.AssociatedRanList, RanName, "test2")
 434         e2tInstancesManagerMock.On("GetE2TInstance", e2tInstanceAddress).Return(e2tInstance, nil)
 435         notificationRequest := &models.NotificationRequest{RanName: RanName, Payload: []byte(e2tInitPayload)}
 436
 437         handler.Handle(notificationRequest)
 438
 439         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 440         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 2)
 441 }
 442
 443
 444 // TODO: extract to test_utils
 445 func initRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
 446         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
 447         rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
 448         return rmrsender.NewRmrSender(log, rmrMessenger)
 449 }
 450
 451 // TODO: extract to test_utils
 452 func initLog(t *testing.T) *logger.Logger {
 453         log, err := logger.InitLogger(logger.InfoLevel)
 454         if err != nil {
 455                 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
 456         }
 457         return log
 458 }