E2Manager/managers/e2_connected_status_change_manager_test.go

   1 //\r
   2 // Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved.\r
   3 //\r
   4 // Licensed under the Apache License, Version 2.0 (the "License");\r
   5 // you may not use this file except in compliance with the License.\r
   6 // You may obtain a copy of the License at\r
   7 //\r
   8 //      http://www.apache.org/licenses/LICENSE-2.0\r
   9 //\r
  10 // Unless required by applicable law or agreed to in writing, software\r
  11 // distributed under the License is distributed on an "AS IS" BASIS,\r
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13 // See the License for the specific language governing permissions and\r
  14 // limitations under the License.\r
  15 \r
  16 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)\r
  17 //  platform project (RICP).\r
  18 package managers\r
  19 \r
  20 import (\r
  21         "e2mgr/configuration"\r
  22         "e2mgr/logger"\r
  23         "e2mgr/mocks"\r
  24         "e2mgr/services"\r
  25         "testing"\r
  26 \r
  27         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"\r
  28         "github.com/stretchr/testify/assert"\r
  29         "github.com/stretchr/testify/mock"\r
  30 )\r
  31 \r
  32 func initChangeStatusToConnectedRanTest(t *testing.T) (*logger.Logger, *mocks.RmrMessengerMock, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *ChangeStatusToConnectedRanManager) {\r
  33         logger, err := logger.InitLogger(logger.DebugLevel)\r
  34         if err != nil {\r
  35                 t.Errorf("#... - failed to initialize logger, error: %s", err)\r
  36         }\r
  37         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}\r
  38 \r
  39         rmrMessengerMock := &mocks.RmrMessengerMock{}\r
  40         readerMock := &mocks.RnibReaderMock{}\r
  41         writerMock := &mocks.RnibWriterMock{}\r
  42         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)\r
  43         ranListManager := NewRanListManager(logger, rnibDataService)\r
  44         ranAlarmService := services.NewRanAlarmService(logger, config)\r
  45         ranConnectStatusChangeManager := NewRanConnectStatusChangeManager(logger, rnibDataService, ranListManager, ranAlarmService)\r
  46         changeStatusToConnectedRanManager := NewChangeStatusToConnectedRanManager(logger, rnibDataService, ranConnectStatusChangeManager)\r
  47         return logger, rmrMessengerMock, readerMock, writerMock, changeStatusToConnectedRanManager\r
  48 }\r
  49 \r
  50 func TestChangeStatusToConnectedRanSucceeds(t *testing.T) {\r
  51         logger, _, readerMock, writerMock, changeStatusToConnectedRanManager := initChangeStatusToConnectedRanTest(t)\r
  52         logger.Infof("#TestChangeStatusToConnectedRanManager.ConnectedRan - RAN name")\r
  53         origNodebInfo := &entities.NodebInfo{RanName: ranName, GlobalNbId: &entities.GlobalNbId{PlmnId: "xxx", NbId: "yyy"}, ConnectionStatus: entities.ConnectionStatus_CONNECTING}\r
  54         var rnibErr error\r
  55         readerMock.On("GetNodeb", ranName).Return(origNodebInfo, rnibErr)\r
  56         writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test_CONNECTED").Return(rnibErr)\r
  57         updatedNodebInfo1 := *origNodebInfo\r
  58         updatedNodebInfo1.ConnectionStatus = entities.ConnectionStatus_CONNECTED\r
  59         writerMock.On("UpdateNodebInfo", mock.Anything).Return(rnibErr)\r
  60         _, err := changeStatusToConnectedRanManager.ChangeStatusToConnectedRan(ranName)\r
  61         assert.Nil(t, err)\r
  62         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)\r
  63 }\r