E2Manager/managers/ran_setup_manager_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 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  18 //  platform project (RICP).
  19
  20
  21 package managers
  22
  23 import (
  24         "e2mgr/configuration"
  25         "e2mgr/e2managererrors"
  26         "e2mgr/e2pdus"
  27         "e2mgr/logger"
  28         "e2mgr/mocks"
  29         "e2mgr/rmrCgo"
  30         "e2mgr/services"
  31         "fmt"
  32         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
  33         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  34         "github.com/stretchr/testify/assert"
  35         "github.com/stretchr/testify/mock"
  36         "testing"
  37 )
  38
  39 func initRanSetupManagerTest(t *testing.T) (*mocks.RmrMessengerMock, *mocks.RnibWriterMock, *RanSetupManager) {
  40         logger, err := logger.InitLogger(logger.DebugLevel)
  41         if err != nil {
  42                 t.Errorf("#... - failed to initialize logger, error: %s", err)
  43         }
  44         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
  45
  46         rmrMessengerMock := &mocks.RmrMessengerMock{}
  47         rmrSender := initRmrSender(rmrMessengerMock, logger)
  48
  49         readerMock := &mocks.RnibReaderMock{}
  50
  51         writerMock := &mocks.RnibWriterMock{}
  52
  53         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)
  54         ranSetupManager := NewRanSetupManager(logger, rmrSender, rnibDataService)
  55         return rmrMessengerMock, writerMock, ranSetupManager
  56 }
  57
  58 func TestExecuteSetupConnectingX2Setup(t *testing.T) {
  59         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
  60
  61         ranName := "test1"
  62
  63         var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
  64         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
  65         var rnibErr error
  66         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
  67
  68         payload := e2pdus.PackedX2setupRequest
  69         xaction := []byte(ranName)
  70         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
  71         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
  72
  73         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
  74                 t.Errorf("want: success, got: error: %s", err)
  75         }
  76
  77         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
  78         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
  79 }
  80
  81 func TestExecuteSetupConnectingEndcX2Setup(t *testing.T) {
  82         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
  83
  84         ranName := "test1"
  85
  86                 var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
  87         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
  88         var rnibErr error
  89         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
  90
  91         payload := e2pdus.PackedEndcX2setupRequest
  92         xaction := []byte(ranName)
  93         msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
  94         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
  95
  96         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err != nil {
  97                 t.Errorf("want: success, got: error: %s", err)
  98         }
  99
 100         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 101         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 102 }
 103
 104 func TestExecuteSetupDisconnected(t *testing.T) {
 105         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
 106
 107         ranName := "test1"
 108
 109         var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 110         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 111         var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
 112         var rnibErr error
 113         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 114         writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
 115
 116         payload := []byte{0}
 117         xaction := []byte(ranName)
 118         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
 119         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
 120
 121         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
 122                 t.Errorf("want: failure, got: success")
 123         }
 124
 125         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 126         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 127 }
 128
 129 func TestExecuteSetupConnectingRnibError(t *testing.T) {
 130         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
 131
 132         ranName := "test1"
 133
 134         var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 135         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 136         var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
 137         var rnibErr = common.NewInternalError(fmt.Errorf("DB error"))
 138         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 139         writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(rnibErr)
 140
 141         payload := []byte{0}
 142         xaction := []byte(ranName)
 143         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
 144         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
 145
 146         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
 147                 t.Errorf("want: failure, got: success")
 148         } else {
 149                 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
 150         }
 151
 152         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 153         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
 154 }
 155
 156 func TestExecuteSetupDisconnectedRnibError(t *testing.T) {
 157         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
 158
 159         ranName := "test1"
 160
 161         var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
 162         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
 163         var argNodebDisconnected = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 0}
 164         var rnibErr error
 165         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 166         writerMock.On("UpdateNodebInfo", argNodebDisconnected).Return(common.NewInternalError(fmt.Errorf("DB error")))
 167
 168         payload := []byte{0}
 169         xaction := []byte(ranName)
 170         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
 171         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, fmt.Errorf("send failure"))
 172
 173         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
 174                 t.Errorf("want: failure, got: success")
 175         } else {
 176                 assert.IsType(t, e2managererrors.NewRnibDbError(), err)
 177         }
 178
 179         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
 180         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 1)
 181 }
 182
 183 func TestExecuteSetupUnsupportedProtocol(t *testing.T) {
 184         rmrMessengerMock, writerMock, mgr := initRanSetupManagerTest(t)
 185
 186         ranName := "test1"
 187
 188         var initialNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL}
 189         var argNodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_UNKNOWN_E2_APPLICATION_PROTOCOL, ConnectionAttempts: 1}
 190         var rnibErr error
 191         writerMock.On("UpdateNodebInfo", argNodeb).Return(rnibErr)
 192
 193         payload := e2pdus.PackedX2setupRequest
 194         xaction := []byte(ranName)
 195         msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
 196         rmrMessengerMock.On("SendMsg", mock.Anything, true).Return(msg, nil)
 197
 198         if err := mgr.ExecuteSetup(initialNodeb, entities.ConnectionStatus_CONNECTING); err == nil {
 199                 t.Errorf("want: error, got: success")
 200         }
 201
 202         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
 203         rmrMessengerMock.AssertNumberOfCalls(t, "SendMsg", 0)
 204 }
 205
 206 func initLog(t *testing.T) *logger.Logger {
 207         log, err := logger.InitLogger(logger.InfoLevel)
 208         if err != nil {
 209                 t.Errorf("#initLog test - failed to initialize logger, error: %s", err)
 210         }
 211         return log
 212 }