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