RIC916 new reconnect timer in E2 to reject new connect for x seconds
[ric-plt/e2mgr.git] / E2Manager / managers / ran_connect_status_change_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 package managers
21
22 import (
23         "e2mgr/configuration"
24         "e2mgr/logger"
25         "e2mgr/mocks"
26         "e2mgr/services"
27         "testing"
28
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"
34 )
35
36 const EventChannelForTest = "RAN_CONNECTION_STATUS_CHANGE"
37
38 func initRanConnectStatusChangeManagerTest(t *testing.T) (*mocks.RnibWriterMock, *mocks.RanListManagerMock, *mocks.RanAlarmServiceMock, *RanConnectStatusChangeManager) {
39         log, err := logger.InitLogger(logger.DebugLevel)
40         if err != nil {
41                 t.Errorf("#... - failed to initialize log, error: %s", err)
42         }
43         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3,
44                 RnibWriter: configuration.RnibWriterConfig{
45                         StateChangeMessageChannel: EventChannelForTest,
46                 },
47         }
48
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
56 }
57
58 func TestChangeStatusSuccessNewRan(t *testing.T) {
59         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
60
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)
68         assert.Nil(t, err)
69         writerMock.AssertExpectations(t)
70         ranListManagerMock.AssertExpectations(t)
71         ranAlarmServiceMock.AssertExpectations(t)
72 }
73
74 func TestChangeStatusSuccessEventNone1(t *testing.T) {
75         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
76
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)
83         assert.Nil(t, err)
84         writerMock.AssertExpectations(t)
85         ranListManagerMock.AssertExpectations(t)
86         ranAlarmServiceMock.AssertExpectations(t)
87 }
88
89 func TestChangeStatusSuccessEventNone2(t *testing.T) {
90         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
91
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)
98         assert.Nil(t, err)
99         writerMock.AssertExpectations(t)
100         ranListManagerMock.AssertExpectations(t)
101         ranAlarmServiceMock.AssertExpectations(t)
102 }
103
104 func TestChangeStatusSuccessEventConnected(t *testing.T) {
105         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
106
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)
114         assert.Nil(t, err)
115         writerMock.AssertExpectations(t)
116         ranListManagerMock.AssertExpectations(t)
117         ranAlarmServiceMock.AssertExpectations(t)
118 }
119
120 func TestChangeStatusSuccessEventDisconnected(t *testing.T) {
121         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
122
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)
130         assert.Nil(t, err)
131         writerMock.AssertExpectations(t)
132         ranListManagerMock.AssertExpectations(t)
133         ranAlarmServiceMock.AssertExpectations(t)
134 }
135
136 func TestChangeStatusRnibErrorEventNone(t *testing.T) {
137         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
138
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)
148 }
149
150 func TestChangeStatusRnibErrorEventConnected(t *testing.T) {
151         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
152
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)
162 }
163
164 func TestChangeStatusRanListManagerError(t *testing.T) {
165         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
166
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)
173         assert.Nil(t, err)
174         writerMock.AssertExpectations(t)
175         ranListManagerMock.AssertExpectations(t)
176         ranAlarmServiceMock.AssertExpectations(t)
177 }
178
179 func TestChangeStatusRanAlarmServiceErrorEventConnected(t *testing.T) {
180         writerMock, ranListManagerMock, ranAlarmServiceMock, ranConnectStatusChangeManager := initRanConnectStatusChangeManagerTest(t)
181
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)
189         assert.Nil(t, err)
190         writerMock.AssertExpectations(t)
191         ranListManagerMock.AssertExpectations(t)
192         ranAlarmServiceMock.AssertExpectations(t)
193 }