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.
23 "e2mgr/e2managererrors"
29 "e2mgr/providers/httpmsghandlerprovider"
36 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
37 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
38 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
39 "github.com/gorilla/mux"
40 "github.com/stretchr/testify/assert"
41 "github.com/stretchr/testify/mock"
50 func setupControllerTest(t *testing.T) (*Controller, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock){
52 config := configuration.ParseConfiguration()
54 rmrMessengerMock := &mocks.RmrMessengerMock{}
55 readerMock := &mocks.RnibReaderMock{}
56 readerProvider := func() reader.RNibReader {
59 writerMock := &mocks.RnibWriterMock{}
60 writerProvider := func() rNibWriter.RNibWriter {
63 rnibDataService := services.NewRnibDataService(log, config, readerProvider, writerProvider)
65 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rnibDataService)
66 controller := NewController(log, getRmrService(rmrMessengerMock, log), rnibDataService, config, ranSetupManager)
67 return controller, readerMock, writerMock, rmrMessengerMock
70 func TestX2SetupInvalidBody(t *testing.T) {
72 controller, _, _, _ := setupControllerTest(t)
74 header := http.Header{}
75 header.Set("Content-Type", "application/json")
76 httpRequest, _ := http.NewRequest("POST", "http://localhost:3800/v1/nodeb/x2-setup", strings.NewReader("{}{}"))
77 httpRequest.Header = header
79 writer := httptest.NewRecorder()
80 controller.X2SetupHandler(writer, httpRequest)
82 var errorResponse = parseJsonRequest(t, writer.Body)
84 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
85 assert.Equal(t, e2managererrors.NewInvalidJsonError().Code, errorResponse.Code)
88 func TestX2SetupSuccess(t *testing.T) {
90 controller, readerMock, writerMock, rmrMessengerMock := setupControllerTest(t)
93 nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
94 readerMock.On("GetNodeb", ranName).Return(nb, nil)
96 var nbUpdated = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST, ConnectionAttempts: 1}
97 writerMock.On("UpdateNodebInfo", nbUpdated).Return(nil)
99 payload := e2pdus.PackedX2setupRequest
100 xaction := []byte(ranName)
101 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
103 rmrMessengerMock.On("SendMsg", mock.Anything, mock.Anything).Return(msg, nil)
105 header := http.Header{}
106 header.Set("Content-Type", "application/json")
107 httpRequest := tests.GetHttpRequest()
108 httpRequest.Header = header
110 writer := httptest.NewRecorder()
111 controller.X2SetupHandler(writer, httpRequest)
113 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
116 func TestEndcSetupSuccess(t *testing.T) {
118 controller, readerMock, writerMock, rmrMessengerMock := setupControllerTest(t)
121 nb := &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST}
122 readerMock.On("GetNodeb", ranName).Return(nb, nil)
124 var nbUpdated = &entities.NodebInfo{RanName: ranName, ConnectionStatus: entities.ConnectionStatus_CONNECTING, E2ApplicationProtocol: entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST, ConnectionAttempts: 1}
125 writerMock.On("UpdateNodebInfo", nbUpdated).Return(nil)
127 payload := e2pdus.PackedEndcX2setupRequest
128 xaction := []byte(ranName)
129 msg := rmrCgo.NewMBuf(rmrCgo.RIC_ENDC_X2_SETUP_REQ, len(payload), ranName, &payload, &xaction)
131 rmrMessengerMock.On("SendMsg", mock.Anything, mock.Anything).Return(msg, nil)
133 header := http.Header{}
134 header.Set("Content-Type", "application/json")
135 httpRequest := tests.GetHttpRequest()
136 httpRequest.Header = header
138 writer := httptest.NewRecorder()
139 controller.EndcSetupHandler(writer, httpRequest)
141 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
144 func TestShutdownHandlerRnibError(t *testing.T) {
145 controller, readerMock, _, _:= setupControllerTest(t)
147 rnibErr := &common.ResourceNotFoundError{}
148 var nbIdentityList []*entities.NbIdentity
149 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
151 writer := httptest.NewRecorder()
153 controller.ShutdownHandler(writer, tests.GetHttpRequest())
155 var errorResponse = parseJsonRequest(t, writer.Body)
157 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
158 assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Code)
161 func TestHeaderValidationFailed(t *testing.T) {
162 controller, _, _, _ := setupControllerTest(t)
164 writer := httptest.NewRecorder()
166 header := &http.Header{}
168 controller.handleRequest(writer, header, httpmsghandlerprovider.ShutdownRequest, nil, true)
170 var errorResponse = parseJsonRequest(t, writer.Body)
171 err := e2managererrors.NewHeaderValidationError()
173 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
174 assert.Equal(t, errorResponse.Code, err.Code)
175 assert.Equal(t, errorResponse.Message, err.Message)
178 func TestShutdownStatusNoContent(t *testing.T) {
179 controller, readerMock, _, _ := setupControllerTest(t)
182 nbIdentityList := []*entities.NbIdentity{}
183 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
185 writer := httptest.NewRecorder()
186 controller.ShutdownHandler(writer, tests.GetHttpRequest())
188 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
191 func TestHandleInternalError(t *testing.T) {
192 controller, _, _, _ := setupControllerTest(t)
194 writer := httptest.NewRecorder()
195 err := e2managererrors.NewInternalError()
197 controller.handleErrorResponse(err, writer)
198 var errorResponse = parseJsonRequest(t, writer.Body)
200 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
201 assert.Equal(t, errorResponse.Code, err.Code)
202 assert.Equal(t, errorResponse.Message, err.Message)
205 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
206 controller, _, _, _ := setupControllerTest(t)
207 writer := httptest.NewRecorder()
208 err := e2managererrors.NewCommandAlreadyInProgressError()
210 controller.handleErrorResponse(err, writer)
211 var errorResponse = parseJsonRequest(t, writer.Body)
213 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
214 assert.Equal(t, errorResponse.Code, err.Code)
215 assert.Equal(t, errorResponse.Message, err.Message)
218 func TestValidateHeaders(t *testing.T) {
219 controller, _, _, _ := setupControllerTest(t)
221 header := http.Header{}
222 header.Set("Content-Type", "application/json")
223 result := controller.validateRequestHeader(&header)
225 assert.Nil(t, result)
228 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
230 var errorResponse models.ErrorResponse
231 body, err := ioutil.ReadAll(r)
233 t.Errorf("Error cannot deserialize json request")
235 json.Unmarshal(body, &errorResponse)
240 func initLog(t *testing.T) *logger.Logger {
241 log, err := logger.InitLogger(logger.InfoLevel)
243 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
248 func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T) {
249 controller, readerMock, _, rmrMessengerMock := setupControllerTest(t)
252 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
253 xaction := []byte(ranName)
254 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
255 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
257 writer := httptest.NewRecorder()
259 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
260 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
262 data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"}
263 b := new(bytes.Buffer)
264 _ = json.NewEncoder(b).Encode(data4Req)
265 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
266 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
268 controller.X2ResetHandler(writer, req)
269 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
273 func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T) {
274 controller, readerMock, _, rmrMessengerMock := setupControllerTest(t)
278 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
279 xaction := []byte(ranName)
280 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
281 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
283 writer := httptest.NewRecorder()
285 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
286 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
289 b := new(bytes.Buffer)
290 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
291 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
293 controller.X2ResetHandler(writer, req)
294 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
298 func TestX2ResetHandleFailureInvalidBody(t *testing.T) {
299 controller, _, _, _ := setupControllerTest(t)
303 writer := httptest.NewRecorder()
305 // Invalid json: attribute name without quotes (should be "cause":).
306 b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"")
307 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
308 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
310 controller.X2ResetHandler(writer, req)
311 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
315 func TestHandleErrorResponse(t *testing.T) {
316 controller, _, _, _ := setupControllerTest(t)
318 writer := httptest.NewRecorder()
319 controller.handleErrorResponse(e2managererrors.NewRnibDbError(), writer)
320 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
322 writer = httptest.NewRecorder()
323 controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(), writer)
324 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
326 writer = httptest.NewRecorder()
327 controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(), writer)
328 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
330 writer = httptest.NewRecorder()
331 controller.handleErrorResponse(e2managererrors.NewWrongStateError("", ""), writer)
332 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
334 writer = httptest.NewRecorder()
335 controller.handleErrorResponse(e2managererrors.NewRequestValidationError(), writer)
336 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
338 writer = httptest.NewRecorder()
339 controller.handleErrorResponse(e2managererrors.NewRmrError(), writer)
340 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
342 writer = httptest.NewRecorder()
343 controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(), writer)
344 assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode)
346 writer = httptest.NewRecorder()
347 controller.handleErrorResponse(fmt.Errorf("ErrorError"), writer)
348 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)