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"
28 "e2mgr/providers/httpmsghandlerprovider"
34 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
35 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
36 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
37 "github.com/gorilla/mux"
38 "github.com/stretchr/testify/assert"
39 "github.com/stretchr/testify/mock"
48 func TestX2SetupSuccess(t *testing.T) {
51 rmrMessengerMock := &mocks.RmrMessengerMock{}
52 readerMock := &mocks.RnibReaderMock{}
53 readerProvider := func() reader.RNibReader {
56 writerMock := &mocks.RnibWriterMock{}
57 writerProvider := func() rNibWriter.RNibWriter {
60 config := configuration.ParseConfiguration()
62 header := http.Header{}
63 header.Set("Content-Type", "application/json")
65 writer := httptest.NewRecorder()
66 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
67 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
69 httpRequest := tests.GetHttpRequest()
70 httpRequest.Header = header
71 controller.X2SetupHandler(writer, httpRequest)
73 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)*/
76 func TestShutdownHandlerRnibError(t *testing.T) {
78 config := configuration.ParseConfiguration()
80 rmrMessengerMock := &mocks.RmrMessengerMock{}
81 readerMock := &mocks.RnibReaderMock{}
82 readerProvider := func() reader.RNibReader {
85 writerMock := &mocks.RnibWriterMock{}
86 writerProvider := func() rNibWriter.RNibWriter {
90 rnibErr := &common.ResourceNotFoundError{}
91 var nbIdentityList []*entities.NbIdentity
92 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
94 writer := httptest.NewRecorder()
95 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
96 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
97 controller.ShutdownHandler(writer, tests.GetHttpRequest())
99 var errorResponse = parseJsonRequest(t, writer.Body)
101 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
102 assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Code)
105 func TestHeaderValidationFailed(t *testing.T) {
107 config := configuration.ParseConfiguration()
109 rmrMessengerMock := &mocks.RmrMessengerMock{}
110 readerMock := &mocks.RnibReaderMock{}
111 readerProvider := func() reader.RNibReader {
114 writerMock := &mocks.RnibWriterMock{}
115 writerProvider := func() rNibWriter.RNibWriter {
119 writer := httptest.NewRecorder()
120 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
121 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
123 header := &http.Header{}
125 controller.handleRequest(writer, header, httpmsghandlerprovider.ShutdownRequest, nil, true)
127 var errorResponse = parseJsonRequest(t, writer.Body)
128 err := e2managererrors.NewHeaderValidationError()
130 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
131 assert.Equal(t, errorResponse.Code, err.Code)
132 assert.Equal(t, errorResponse.Message, err.Message)
135 func TestShutdownStatusNoContent(t *testing.T) {
138 rmrMessengerMock := &mocks.RmrMessengerMock{}
139 readerMock := &mocks.RnibReaderMock{}
140 readerProvider := func() reader.RNibReader {
143 writerMock := &mocks.RnibWriterMock{}
144 writerProvider := func() rNibWriter.RNibWriter {
147 config := configuration.ParseConfiguration()
150 nbIdentityList := []*entities.NbIdentity{}
151 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
153 writer := httptest.NewRecorder()
154 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
155 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
156 controller.ShutdownHandler(writer, tests.GetHttpRequest())
158 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
161 func TestHandleInternalError(t *testing.T) {
163 config := configuration.ParseConfiguration()
165 rmrMessengerMock := &mocks.RmrMessengerMock{}
166 readerMock := &mocks.RnibReaderMock{}
167 readerProvider := func() reader.RNibReader {
170 writerMock := &mocks.RnibWriterMock{}
171 writerProvider := func() rNibWriter.RNibWriter {
175 writer := httptest.NewRecorder()
176 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
177 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
178 err := e2managererrors.NewInternalError()
180 controller.handleErrorResponse(err, writer)
181 var errorResponse = parseJsonRequest(t, writer.Body)
183 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
184 assert.Equal(t, errorResponse.Code, err.Code)
185 assert.Equal(t, errorResponse.Message, err.Message)
188 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
190 config := configuration.ParseConfiguration()
192 rmrMessengerMock := &mocks.RmrMessengerMock{}
193 readerMock := &mocks.RnibReaderMock{}
194 readerProvider := func() reader.RNibReader {
197 writerMock := &mocks.RnibWriterMock{}
198 writerProvider := func() rNibWriter.RNibWriter {
201 writer := httptest.NewRecorder()
202 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
203 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
204 err := e2managererrors.NewCommandAlreadyInProgressError()
206 controller.handleErrorResponse(err, writer)
207 var errorResponse = parseJsonRequest(t, writer.Body)
209 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
210 assert.Equal(t, errorResponse.Code, err.Code)
211 assert.Equal(t, errorResponse.Message, err.Message)
214 func TestValidateHeaders(t *testing.T) {
217 rmrMessengerMock := &mocks.RmrMessengerMock{}
218 readerMock := &mocks.RnibReaderMock{}
219 readerProvider := func() reader.RNibReader {
222 writerMock := &mocks.RnibWriterMock{}
223 writerProvider := func() rNibWriter.RNibWriter {
226 config := configuration.ParseConfiguration()
227 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
228 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config, ranSetupManager)
230 header := http.Header{}
231 header.Set("Content-Type", "application/json")
232 result := controller.validateRequestHeader(&header)
234 assert.Nil(t, result)
237 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
239 var errorResponse models.ErrorResponse
240 body, err := ioutil.ReadAll(r)
242 t.Errorf("Error cannot deserialize json request")
244 json.Unmarshal(body, &errorResponse)
249 func initLog(t *testing.T) *logger.Logger {
250 log, err := logger.InitLogger(logger.InfoLevel)
252 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
257 func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T) {
262 readerMock := &mocks.RnibReaderMock{}
263 readerProvider := func() reader.RNibReader {
266 writerMock := &mocks.RnibWriterMock{}
267 writerProvider := func() rNibWriter.RNibWriter {
270 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
271 xaction := []byte(ranName)
272 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
273 rmrMessengerMock := &mocks.RmrMessengerMock{}
274 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
276 config := configuration.ParseConfiguration()
277 rmrService := getRmrService(rmrMessengerMock, log)
279 writer := httptest.NewRecorder()
280 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
281 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
283 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
284 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
286 data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"}
287 b := new(bytes.Buffer)
288 _ = json.NewEncoder(b).Encode(data4Req)
289 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
290 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
292 controller.X2ResetHandler(writer, req)
293 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
297 func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T) {
302 readerMock := &mocks.RnibReaderMock{}
303 readerProvider := func() reader.RNibReader {
306 writerMock := &mocks.RnibWriterMock{}
307 writerProvider := func() rNibWriter.RNibWriter {
311 payload := []byte{0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
312 xaction := []byte(ranName)
313 msg := rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
314 rmrMessengerMock := &mocks.RmrMessengerMock{}
315 rmrMessengerMock.On("SendMsg", msg, mock.Anything).Return(msg, nil)
317 config := configuration.ParseConfiguration()
318 rmrService := getRmrService(rmrMessengerMock, log)
320 writer := httptest.NewRecorder()
321 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
322 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
324 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED}
325 readerMock.On("GetNodeb", ranName).Return(nodeb, nil)
328 b := new(bytes.Buffer)
329 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
330 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
332 controller.X2ResetHandler(writer, req)
333 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
337 func TestX2ResetHandleFailureInvalidBody(t *testing.T) {
342 readerMock := &mocks.RnibReaderMock{}
343 readerProvider := func() reader.RNibReader {
346 writerMock := &mocks.RnibWriterMock{}
347 writerProvider := func() rNibWriter.RNibWriter {
350 rmrMessengerMock := &mocks.RmrMessengerMock{}
352 config := configuration.ParseConfiguration()
353 rmrService := getRmrService(rmrMessengerMock, log)
355 writer := httptest.NewRecorder()
356 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
357 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
359 // Invalid json: attribute name without quotes (should be "cause":).
360 b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"")
361 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
362 req = mux.SetURLVars(req, map[string]string{"ranName": ranName})
364 controller.X2ResetHandler(writer, req)
365 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
367 _, ok := rmrService.E2sessions[ranName]
372 func TestHandleErrorResponse(t *testing.T) {
375 readerMock := &mocks.RnibReaderMock{}
376 readerProvider := func() reader.RNibReader {
379 writerMock := &mocks.RnibWriterMock{}
380 writerProvider := func() rNibWriter.RNibWriter {
383 rmrMessengerMock := &mocks.RmrMessengerMock{}
385 config := configuration.ParseConfiguration()
386 rmrService := getRmrService(rmrMessengerMock, log)
387 ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rNibWriter.GetRNibWriter)
388 controller := NewController(log, rmrService, readerProvider, writerProvider, config, ranSetupManager)
390 writer := httptest.NewRecorder()
391 controller.handleErrorResponse(e2managererrors.NewRnibDbError(), writer)
392 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
394 writer = httptest.NewRecorder()
395 controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(), writer)
396 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
398 writer = httptest.NewRecorder()
399 controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(), writer)
400 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
402 writer = httptest.NewRecorder()
403 controller.handleErrorResponse(e2managererrors.NewWrongStateError("", ""), writer)
404 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
406 writer = httptest.NewRecorder()
407 controller.handleErrorResponse(e2managererrors.NewRequestValidationError(), writer)
408 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
410 writer = httptest.NewRecorder()
411 controller.handleErrorResponse(e2managererrors.NewRmrError(), writer)
412 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
414 writer = httptest.NewRecorder()
415 controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(), writer)
416 assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode)
418 writer = httptest.NewRecorder()
419 controller.handleErrorResponse(fmt.Errorf("ErrorError"), writer)
420 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)