E2Manager/managers/e2_reset_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 \r
  19 package managers\r
  20 \r
  21 import (\r
  22         "e2mgr/configuration"\r
  23         "e2mgr/logger"\r
  24         "e2mgr/mocks"\r
  25         "e2mgr/services"\r
  26         "testing"\r
  27 \r
  28         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"\r
  29         "github.com/stretchr/testify/assert"\r
  30         "github.com/stretchr/testify/mock"\r
  31 )\r
  32 \r
  33 func initE2ResetStatusChangeTest(t *testing.T) (*logger.Logger, *mocks.RmrMessengerMock, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *RanResetManager) {\r
  34         DebugLevel := int8(4)\r
  35         logger, err := logger.InitLogger(DebugLevel)\r
  36         if err != nil {\r
  37                 t.Errorf("#... - failed to initialize logger, error: %s", err)\r
  38         }\r
  39         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}\r
  40 \r
  41         rmrMessengerMock := &mocks.RmrMessengerMock{}\r
  42         readerMock := &mocks.RnibReaderMock{}\r
  43         writerMock := &mocks.RnibWriterMock{}\r
  44         rnibDataService := services.NewRnibDataService(logger, config, readerMock, writerMock)\r
  45         ranListManager := NewRanListManager(logger, rnibDataService)\r
  46         ranAlarmService := services.NewRanAlarmService(logger, config)\r
  47         ranConnectStatusChangeManager := NewRanConnectStatusChangeManager(logger, rnibDataService, ranListManager, ranAlarmService)\r
  48         e2ResetStatusChangeManager := NewRanResetManager(logger, rnibDataService, ranConnectStatusChangeManager)\r
  49         return logger, rmrMessengerMock, readerMock, writerMock, e2ResetStatusChangeManager\r
  50 }\r
  51 \r
  52 func TestE2ResetStatusChangeSucceeds(t *testing.T) {\r
  53         logger, _, readerMock, writerMock, e2ResetStatusChangeManager := initE2ResetStatusChangeTest(t)\r
  54         logger.Infof("#TestRanResetManager.ResetRan - RAN name")\r
  55         origNodebInfo := &entities.NodebInfo{RanName: ranName, GlobalNbId: &entities.GlobalNbId{PlmnId: "xxx", NbId: "yyy"}, ConnectionStatus: entities.ConnectionStatus_CONNECTING}\r
  56         var rnibErr error\r
  57         readerMock.On("GetNodeb", ranName).Return(origNodebInfo, rnibErr)\r
  58         writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, mock.Anything).Return(nil)\r
  59         updatedNodebInfo1 := *origNodebInfo\r
  60         updatedNodebInfo1.ConnectionStatus = entities.ConnectionStatus_UNDER_RESET\r
  61         writerMock.On("UpdateNodebInfo", mock.Anything).Return(rnibErr)\r
  62         _, err := e2ResetStatusChangeManager.ResetRan(ranName)\r
  63         assert.Nil(t, err)\r
  64         writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)\r
  65 }\r