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).
21 package httpmsghandlers
30 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
31 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
32 "github.com/stretchr/testify/assert"
36 func setupDeleteEnbRequestHandlerTest(t *testing.T, emptyList bool) (*DeleteEnbRequestHandler, *mocks.RnibReaderMock, *mocks.RnibWriterMock) {
38 config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
39 readerMock := &mocks.RnibReaderMock{}
40 writerMock := &mocks.RnibWriterMock{}
41 rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock)
42 ranListManager := managers.NewRanListManager(log, rnibDataService)
44 nbIdentity := &entities.NbIdentity{InventoryName: "ran1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}
45 writerMock.On("AddNbIdentity", entities.Node_ENB, nbIdentity).Return(nil)
46 if err := ranListManager.AddNbIdentity(entities.Node_ENB, nbIdentity); err != nil {
47 t.Errorf("#setupDeleteEnbRequestHandlerTest - Failed to add nbIdentity prior to DeleteEnb test")
50 handler := NewDeleteEnbRequestHandler(log, rnibDataService, ranListManager)
51 return handler, readerMock, writerMock
54 func TestHandleDeleteEnbSuccess(t *testing.T) {
55 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
59 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_ENB}
60 readerMock.On("GetNodeb", ranName).Return(nodebInfo, rnibError)
61 writerMock.On("RemoveEnb", nodebInfo).Return(nil)
62 writerMock.On("RemoveNbIdentity", entities.Node_ENB, &entities.NbIdentity{InventoryName: "ran1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}).Return(nil)
63 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
65 assert.NotNil(t, result)
66 assert.IsType(t, &models.NodebResponse{}, result)
67 readerMock.AssertExpectations(t)
68 writerMock.AssertExpectations(t)
71 func TestHandleDeleteEnbSuccessNoEnb(t *testing.T) {
72 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, true)
76 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_ENB}
77 readerMock.On("GetNodeb", ranName).Return(nodebInfo, rnibError)
78 writerMock.On("RemoveEnb", nodebInfo).Return(nil)
79 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
81 assert.NotNil(t, result)
82 assert.IsType(t, &models.NodebResponse{}, result)
83 readerMock.AssertExpectations(t)
84 writerMock.AssertExpectations(t)
87 func TestHandleDeleteEnbInternalGetNodebError(t *testing.T) {
88 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
91 rnibError := errors.New("for test")
92 var nodebInfo *entities.NodebInfo
93 readerMock.On("GetNodeb", ranName).Return(nodebInfo, rnibError)
94 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
97 readerMock.AssertExpectations(t)
98 writerMock.AssertExpectations(t)
101 func TestHandleDeleteEnbInternalRemoveEnbError(t *testing.T) {
102 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
105 rnibError := errors.New("for test")
106 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_ENB}
107 readerMock.On("GetNodeb", ranName).Return(nodebInfo, nil)
108 writerMock.On("RemoveEnb", nodebInfo).Return(rnibError)
109 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
110 assert.NotNil(t, err)
111 assert.Nil(t, result)
112 readerMock.AssertExpectations(t)
113 writerMock.AssertExpectations(t)
116 func TestHandleDeleteEnbFromNetworkError(t *testing.T) {
117 handler, readerMock, _ := setupDeleteEnbRequestHandlerTest(t, false)
119 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_ENB, SetupFromNetwork: true}
120 readerMock.On("GetNodeb", ranName).Return(nodebInfo, nil)
121 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
122 assert.NotNil(t, err)
123 assert.Nil(t, result)
124 readerMock.AssertExpectations(t)
127 func TestHandleDeleteEnbInternalRemoveNbIdentityError(t *testing.T) {
128 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
131 rnibError := errors.New("for test")
132 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_ENB}
133 readerMock.On("GetNodeb", ranName).Return(nodebInfo, nil)
134 writerMock.On("RemoveEnb", nodebInfo).Return(nil)
135 writerMock.On("RemoveNbIdentity", entities.Node_ENB, &entities.NbIdentity{InventoryName: "ran1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}).Return(rnibError)
136 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
137 assert.NotNil(t, err)
138 assert.Nil(t, result)
139 readerMock.AssertExpectations(t)
140 writerMock.AssertExpectations(t)
143 func TestHandleDeleteEnbResourceNotFoundError(t *testing.T) {
144 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
147 rnibError := common.NewResourceNotFoundError("for test")
148 var nodebInfo *entities.NodebInfo
149 readerMock.On("GetNodeb", ranName).Return(nodebInfo, rnibError)
150 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
151 assert.NotNil(t, err)
152 assert.Nil(t, result)
153 readerMock.AssertExpectations(t)
154 writerMock.AssertExpectations(t)
157 func TestHandleDeleteEnbNodeTypeNotEnbError(t *testing.T) {
158 handler, readerMock, writerMock := setupDeleteEnbRequestHandlerTest(t, false)
161 nodebInfo := &entities.NodebInfo{RanName: ranName, NodeType: entities.Node_GNB}
162 readerMock.On("GetNodeb", ranName).Return(nodebInfo, nil)
163 result, err := handler.Handle(&models.DeleteEnbRequest{RanName: ranName})
164 assert.NotNil(t, err)
165 assert.Nil(t, result)
166 readerMock.AssertExpectations(t)
167 writerMock.AssertExpectations(t)