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"
33 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
34 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
35 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
36 "github.com/julienschmidt/httprouter"
37 "github.com/stretchr/testify/assert"
38 "github.com/stretchr/testify/mock"
47 func TestShutdownHandlerRnibError(t *testing.T) {
49 config := configuration.ParseConfiguration()
51 rmrMessengerMock := &mocks.RmrMessengerMock{}
52 readerMock := &mocks.RnibReaderMock{}
53 readerProvider := func() reader.RNibReader {
56 writerMock := &mocks.RnibWriterMock{}
57 writerProvider := func() rNibWriter.RNibWriter {
61 rnibErr := &common.RNibError{}
62 var nbIdentityList []*entities.NbIdentity
63 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
65 writer := httptest.NewRecorder()
66 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
67 controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
69 var errorResponse = parseJsonRequest(t, writer.Body)
71 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
72 assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Err.Code)
75 func TestHeaderValidationFailed(t *testing.T) {
77 config := configuration.ParseConfiguration()
79 rmrMessengerMock := &mocks.RmrMessengerMock{}
80 readerMock := &mocks.RnibReaderMock{}
81 readerProvider := func() reader.RNibReader {
84 writerMock := &mocks.RnibWriterMock{}
85 writerProvider := func() rNibWriter.RNibWriter {
89 writer := httptest.NewRecorder()
91 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
93 header := &http.Header{}
95 controller.handleRequest(writer, header, providers.ShutdownRequest, nil, true, http.StatusNoContent)
97 var errorResponse = parseJsonRequest(t, writer.Body)
98 err := e2managererrors.NewHeaderValidationError()
100 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
101 assert.Equal(t, errorResponse.Code, err.Err.Code)
102 assert.Equal(t, errorResponse.Message, err.Err.Message)
105 func TestShutdownStatusNoContent(t *testing.T){
108 rmrMessengerMock := &mocks.RmrMessengerMock{}
109 readerMock := &mocks.RnibReaderMock{}
110 readerProvider := func() reader.RNibReader {
113 writerMock := &mocks.RnibWriterMock{}
114 writerProvider := func() rNibWriter.RNibWriter {
117 config := configuration.ParseConfiguration()
119 var rnibError common.IRNibError
120 nbIdentityList := []*entities.NbIdentity{}
121 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
123 writer := httptest.NewRecorder()
124 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
125 controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
127 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
130 func TestHandleInternalError(t *testing.T) {
132 config := configuration.ParseConfiguration()
134 rmrMessengerMock := &mocks.RmrMessengerMock{}
135 readerMock := &mocks.RnibReaderMock{}
136 readerProvider := func() reader.RNibReader {
139 writerMock := &mocks.RnibWriterMock{}
140 writerProvider := func() rNibWriter.RNibWriter {
144 writer := httptest.NewRecorder()
145 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
146 err := e2managererrors.NewInternalError()
148 controller.handleErrorResponse(err, writer)
149 var errorResponse = parseJsonRequest(t, writer.Body)
151 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
152 assert.Equal(t, errorResponse.Code, err.Err.Code)
153 assert.Equal(t, errorResponse.Message, err.Err.Message)
156 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
158 config := configuration.ParseConfiguration()
160 rmrMessengerMock := &mocks.RmrMessengerMock{}
161 readerMock := &mocks.RnibReaderMock{}
162 readerProvider := func() reader.RNibReader {
165 writerMock := &mocks.RnibWriterMock{}
166 writerProvider := func() rNibWriter.RNibWriter {
169 writer := httptest.NewRecorder()
170 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
171 err := e2managererrors.NewCommandAlreadyInProgressError()
173 controller.handleErrorResponse(err, writer)
174 var errorResponse = parseJsonRequest(t, writer.Body)
176 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
177 assert.Equal(t, errorResponse.Code, err.Err.Code)
178 assert.Equal(t, errorResponse.Message, err.Err.Message)
181 func TestValidateHeaders(t *testing.T){
184 rmrMessengerMock := &mocks.RmrMessengerMock{}
185 readerMock := &mocks.RnibReaderMock{}
186 readerProvider := func() reader.RNibReader {
189 writerMock := &mocks.RnibWriterMock{}
190 writerProvider := func() rNibWriter.RNibWriter {
193 config := configuration.ParseConfiguration()
195 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
197 header := http.Header{}
198 header.Set("Content-Type", "application/json")
199 result := controller.validateRequestHeader(&header)
201 assert.Nil(t, result)
204 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
206 var errorResponse models.ErrorResponse
207 body, err := ioutil.ReadAll(r)
209 t.Errorf("Error cannot deserialize json request")
211 json.Unmarshal(body, &errorResponse)
216 func initLog(t *testing.T) *logger.Logger {
217 log, err := logger.InitLogger(logger.InfoLevel)
219 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
225 func TestX2ResetHandleSuccessfulRequestedCause(t *testing.T){
230 readerMock := &mocks.RnibReaderMock{}
231 readerProvider := func() reader.RNibReader {
234 writerMock := &mocks.RnibWriterMock{}
235 writerProvider := func() rNibWriter.RNibWriter {
238 payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
239 xaction := []byte(ranName)
240 msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
241 rmrMessengerMock := &mocks.RmrMessengerMock{}
242 rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)
244 config := configuration.ParseConfiguration()
245 rmrService:=getRmrService(rmrMessengerMock, log)
247 writer := httptest.NewRecorder()
248 controller := NewController(log, rmrService, readerProvider, writerProvider, config)
250 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED }
251 readerMock.On("GetNodeb",ranName).Return(nodeb, nil)
253 data4Req := map[string]interface{}{"cause": "protocol:transfer-syntax-error"}
254 b := new(bytes.Buffer)
255 _ = json.NewEncoder(b).Encode(data4Req)
256 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
258 param:= httprouter.Param{Key:"ranName", Value: ranName}
259 controller.X2ResetHandler(writer, req, []httprouter.Param {param})
260 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
262 v, ok := rmrService.E2sessions[ranName]
265 assert.Equal(t, v.Request.RanName, ranName)
268 func TestX2ResetHandleSuccessfulRequestedDefault(t *testing.T){
273 readerMock := &mocks.RnibReaderMock{}
274 readerProvider := func() reader.RNibReader {
277 writerMock := &mocks.RnibWriterMock{}
278 writerProvider := func() rNibWriter.RNibWriter {
282 payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
283 xaction := []byte(ranName)
284 msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
285 rmrMessengerMock := &mocks.RmrMessengerMock{}
286 rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)
288 config := configuration.ParseConfiguration()
289 rmrService:=getRmrService(rmrMessengerMock, log)
291 writer := httptest.NewRecorder()
292 controller := NewController(log, rmrService, readerProvider, writerProvider, config)
294 var nodeb = &entities.NodebInfo{ConnectionStatus: entities.ConnectionStatus_CONNECTED }
295 readerMock.On("GetNodeb",ranName).Return(nodeb, nil)
298 b := new(bytes.Buffer)
299 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
301 param:= httprouter.Param{Key:"ranName", Value: ranName}
302 controller.X2ResetHandler(writer, req, []httprouter.Param {param})
303 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
305 v, ok := rmrService.E2sessions[ranName]
308 assert.Equal(t, v.Request.RanName, ranName)
311 func TestX2ResetHandleFailureInvalidBody(t *testing.T){
316 readerMock := &mocks.RnibReaderMock{}
317 readerProvider := func() reader.RNibReader {
320 writerMock := &mocks.RnibWriterMock{}
321 writerProvider := func() rNibWriter.RNibWriter {
324 rmrMessengerMock := &mocks.RmrMessengerMock{}
326 config := configuration.ParseConfiguration()
327 rmrService:=getRmrService(rmrMessengerMock, log)
329 writer := httptest.NewRecorder()
330 controller := NewController(log, rmrService, readerProvider, writerProvider, config)
332 // Invalid json: attribute name without quotes (should be "cause":).
333 b := strings.NewReader("{cause:\"protocol:transfer-syntax-error\"")
334 req, _ := http.NewRequest("PUT", "https://localhost:3800/nodeb-reset", b)
336 param:= httprouter.Param{Key:"ranName", Value: ranName}
337 controller.X2ResetHandler(writer, req, []httprouter.Param {param})
338 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
340 _, ok := rmrService.E2sessions[ranName]
345 func TestHandleErrorResponse(t *testing.T){
348 readerMock := &mocks.RnibReaderMock{}
349 readerProvider := func() reader.RNibReader {
352 writerMock := &mocks.RnibWriterMock{}
353 writerProvider := func() rNibWriter.RNibWriter {
356 rmrMessengerMock := &mocks.RmrMessengerMock{}
358 config := configuration.ParseConfiguration()
359 rmrService:=getRmrService(rmrMessengerMock, log)
362 controller := NewController(log, rmrService, readerProvider, writerProvider, config)
364 writer := httptest.NewRecorder()
365 controller.handleErrorResponse(e2managererrors.NewRnibDbError(),writer)
366 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
368 writer = httptest.NewRecorder()
369 controller.handleErrorResponse(e2managererrors.NewCommandAlreadyInProgressError(),writer)
370 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
372 writer = httptest.NewRecorder()
373 controller.handleErrorResponse(e2managererrors.NewHeaderValidationError(),writer)
374 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
376 writer = httptest.NewRecorder()
377 controller.handleErrorResponse(e2managererrors.NewWrongStateError(""),writer)
378 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
380 writer = httptest.NewRecorder()
381 controller.handleErrorResponse(e2managererrors.NewRequestValidationError(),writer)
382 assert.Equal(t, http.StatusBadRequest, writer.Result().StatusCode)
384 writer = httptest.NewRecorder()
385 controller.handleErrorResponse(e2managererrors.NewRmrError(),writer)
386 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
388 writer = httptest.NewRecorder()
389 controller.handleErrorResponse(e2managererrors.NewResourceNotFoundError(),writer)
390 assert.Equal(t, http.StatusNotFound, writer.Result().StatusCode)
392 writer = httptest.NewRecorder()
393 controller.handleErrorResponse(fmt.Errorf("ErrorError"),writer)
394 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)