2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
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
9 // http://www.apache.org/licenses/LICENSE-2.0
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.
17 // This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 // platform project (RICP).
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/pkg/errors"
32 "github.com/stretchr/testify/assert"
33 "github.com/stretchr/testify/mock"
36 const EventChannelForTest = "RAN_CONNECTION_STATUS_CHANGE"
38 func initRanConnectStatusChangeManagerTest(t *testing.T) (*mocks.RnibWriterMock, *mocks.RanListManagerMock, *mocks.RanAlarmServiceMock, *RanConnectStatusChangeManager) {
39 log, err := logger.InitLogger(logger.DebugLevel)
41 t.Errorf("#... - failed to initialize log, error: %s", err)
43 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3,
44 RnibWriter: configuration.RnibWriterConfig{
45 StateChangeMessageChannel: EventChannelForTest,
49 readerMock := &mocks.RnibReaderMock{}
50 writerMock := &mocks.RnibWriterMock{}
51 rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock)
52 ranListManagerMock := &mocks.RanListManagerMock{}
53 ranAlarmServiceMock := &mocks.RanAlarmServiceMock{}
54 ranConnectStatusChangeManager := NewRanConnectStatusChangeManager(log, rnibDataService, ranListManagerMock, ranAlarmServiceMock)
55 return writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager
58 func TestChangeStatusSuccessNewRan(t *testing.T) {
59 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
61 origNodebInfo := &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_UNKNOWN_CONNECTION_STATUS}
62 updatedNodebInfo := *origNodebInfo
63 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_CONNECTED
64 writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, RanName+"_"+CONNECTED_RAW_EVENT).Return(nil)
65 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
66 ranAlarmServiceMock.On("SetConnectivityChangeAlarm", mock.Anything).Return(nil)
67 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_CONNECTED)
69 writerMock.AssertExpectations(t)
70 ranListManagerMock.AssertExpectations(t)
71 ranAlarmServiceMock.AssertExpectations(t)
74 func TestChangeStatusSuccessEventNone1(t *testing.T) {
75 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
77 origNodebInfo := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN}
78 updatedNodebInfo := *origNodebInfo
79 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
80 writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)
81 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
82 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_SHUT_DOWN)
84 writerMock.AssertExpectations(t)
85 ranListManagerMock.AssertExpectations(t)
86 ranAlarmServiceMock.AssertExpectations(t)
89 func TestChangeStatusSuccessEventNone2(t *testing.T) {
90 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
92 origNodebInfo := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}
93 updatedNodebInfo := *origNodebInfo
94 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
95 writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)
96 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
97 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_SHUT_DOWN)
99 writerMock.AssertExpectations(t)
100 ranListManagerMock.AssertExpectations(t)
101 ranAlarmServiceMock.AssertExpectations(t)
104 func TestChangeStatusSuccessEventConnected(t *testing.T) {
105 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
107 origNodebInfo := &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}
108 updatedNodebInfo := *origNodebInfo
109 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_CONNECTED
110 writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, RanName+"_"+CONNECTED_RAW_EVENT).Return(nil)
111 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
112 ranAlarmServiceMock.On("SetConnectivityChangeAlarm", mock.Anything).Return(nil)
113 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_CONNECTED)
115 writerMock.AssertExpectations(t)
116 ranListManagerMock.AssertExpectations(t)
117 ranAlarmServiceMock.AssertExpectations(t)
120 func TestChangeStatusSuccessEventDisconnected(t *testing.T) {
121 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
123 origNodebInfo := &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_CONNECTED}
124 updatedNodebInfo := *origNodebInfo
125 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
126 writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, RanName+"_"+DISCONNECTED_RAW_EVENT).Return(nil)
127 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
128 ranAlarmServiceMock.On("SetConnectivityChangeAlarm", mock.Anything).Return(nil)
129 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_DISCONNECTED)
131 writerMock.AssertExpectations(t)
132 ranListManagerMock.AssertExpectations(t)
133 ranAlarmServiceMock.AssertExpectations(t)
136 func TestChangeStatusRnibErrorEventNone(t *testing.T) {
137 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
139 origNodebInfo := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN}
140 updatedNodebInfo := *origNodebInfo
141 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
142 writerMock.On("UpdateNodebInfo", mock.Anything).Return(common.NewInternalError(errors.New("Error")))
143 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_SHUT_DOWN)
144 assert.NotNil(t, err)
145 writerMock.AssertExpectations(t)
146 ranListManagerMock.AssertExpectations(t)
147 ranAlarmServiceMock.AssertExpectations(t)
150 func TestChangeStatusRnibErrorEventConnected(t *testing.T) {
151 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
153 origNodebInfo := &entities.NodebInfo{RanName: RanName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}
154 updatedNodebInfo := *origNodebInfo
155 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_CONNECTED
156 writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, RanName+"_"+CONNECTED_RAW_EVENT).Return(common.NewInternalError(errors.New("Error")))
157 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_CONNECTED)
158 assert.NotNil(t, err)
159 writerMock.AssertExpectations(t)
160 ranListManagerMock.AssertExpectations(t)
161 ranAlarmServiceMock.AssertExpectations(t)
164 func TestChangeStatusRanListManagerError(t *testing.T) {
165 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
167 origNodebInfo := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN}
168 updatedNodebInfo := *origNodebInfo
169 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
170 writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)
171 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), ranName, updatedNodebInfo.GetConnectionStatus()).Return(common.NewInternalError(errors.New("Error")))
172 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_SHUT_DOWN)
174 writerMock.AssertExpectations(t)
175 ranListManagerMock.AssertExpectations(t)
176 ranAlarmServiceMock.AssertExpectations(t)
179 func TestChangeStatusRanAlarmServiceErrorEventConnected(t *testing.T) {
180 writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
182 origNodebInfo := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}
183 updatedNodebInfo := *origNodebInfo
184 updatedNodebInfo.ConnectionStatus = entities.ConnectionStatus_CONNECTED
185 writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, RanName+"_"+CONNECTED_RAW_EVENT).Return(nil)
186 ranListManagerMock.On("UpdateNbIdentityConnectionStatus", updatedNodebInfo.GetNodeType(), RanName, updatedNodebInfo.GetConnectionStatus()).Return(nil)
187 ranAlarmServiceMock.On("SetConnectivityChangeAlarm", mock.Anything).Return(common.NewInternalError(errors.New("Error")))
188 _, err := ranConnectStatusChangeManager.ChangeStatus(origNodebInfo, entities.ConnectionStatus_CONNECTED)
190 writerMock.AssertExpectations(t)
191 ranListManagerMock.AssertExpectations(t)
192 ranAlarmServiceMock.AssertExpectations(t)