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.
22 "e2mgr/e2managererrors"
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 "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
33 "github.com/stretchr/testify/assert"
41 func TestShutdownHandlerRnibError(t *testing.T) {
43 config := configuration.ParseConfiguration()
45 rmrMessengerMock := &mocks.RmrMessengerMock{}
46 readerMock := &mocks.RnibReaderMock{}
47 readerProvider := func() reader.RNibReader {
50 writerMock := &mocks.RnibWriterMock{}
51 writerProvider := func() rNibWriter.RNibWriter {
55 rnibErr := &common.RNibError{}
56 var nbIdentityList []*entities.NbIdentity
57 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
59 writer := httptest.NewRecorder()
60 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
61 controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
63 var errorResponse = parseJsonRequest(t, writer.Body)
65 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
66 assert.Equal(t, errorResponse.Code, e2managererrors.NewRnibDbError().Err.Code)
69 func TestHeaderValidationFailed(t *testing.T) {
71 config := configuration.ParseConfiguration()
73 rmrMessengerMock := &mocks.RmrMessengerMock{}
74 readerMock := &mocks.RnibReaderMock{}
75 readerProvider := func() reader.RNibReader {
78 writerMock := &mocks.RnibWriterMock{}
79 writerProvider := func() rNibWriter.RNibWriter {
83 writer := httptest.NewRecorder()
85 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
87 header := &http.Header{}
89 controller.handleRequest(writer, header, providers.ShutdownRequest, nil, true, http.StatusNoContent)
91 var errorResponse = parseJsonRequest(t, writer.Body)
92 err := e2managererrors.NewHeaderValidationError()
94 assert.Equal(t, http.StatusUnsupportedMediaType, writer.Result().StatusCode)
95 assert.Equal(t, errorResponse.Code, err.Err.Code)
96 assert.Equal(t, errorResponse.Message, err.Err.Message)
99 func TestShutdownStatusNoContent(t *testing.T){
102 rmrMessengerMock := &mocks.RmrMessengerMock{}
103 readerMock := &mocks.RnibReaderMock{}
104 readerProvider := func() reader.RNibReader {
107 writerMock := &mocks.RnibWriterMock{}
108 writerProvider := func() rNibWriter.RNibWriter {
111 config := configuration.ParseConfiguration()
113 var rnibError common.IRNibError
114 nbIdentityList := []*entities.NbIdentity{}
115 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
117 writer := httptest.NewRecorder()
118 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
119 controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
121 assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
124 func TestHandleInternalError(t *testing.T) {
126 config := configuration.ParseConfiguration()
128 rmrMessengerMock := &mocks.RmrMessengerMock{}
129 readerMock := &mocks.RnibReaderMock{}
130 readerProvider := func() reader.RNibReader {
133 writerMock := &mocks.RnibWriterMock{}
134 writerProvider := func() rNibWriter.RNibWriter {
138 writer := httptest.NewRecorder()
139 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
140 err := e2managererrors.NewInternalError()
142 controller.handleErrorResponse(err, writer)
143 var errorResponse = parseJsonRequest(t, writer.Body)
145 assert.Equal(t, http.StatusInternalServerError, writer.Result().StatusCode)
146 assert.Equal(t, errorResponse.Code, err.Err.Code)
147 assert.Equal(t, errorResponse.Message, err.Err.Message)
150 func TestHandleCommandAlreadyInProgressError(t *testing.T) {
152 config := configuration.ParseConfiguration()
154 rmrMessengerMock := &mocks.RmrMessengerMock{}
155 readerMock := &mocks.RnibReaderMock{}
156 readerProvider := func() reader.RNibReader {
159 writerMock := &mocks.RnibWriterMock{}
160 writerProvider := func() rNibWriter.RNibWriter {
163 writer := httptest.NewRecorder()
164 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
165 err := e2managererrors.NewCommandAlreadyInProgressError()
167 controller.handleErrorResponse(err, writer)
168 var errorResponse = parseJsonRequest(t, writer.Body)
170 assert.Equal(t, http.StatusMethodNotAllowed, writer.Result().StatusCode)
171 assert.Equal(t, errorResponse.Code, err.Err.Code)
172 assert.Equal(t, errorResponse.Message, err.Err.Message)
175 func TestValidateHeaders(t *testing.T){
178 rmrMessengerMock := &mocks.RmrMessengerMock{}
179 readerMock := &mocks.RnibReaderMock{}
180 readerProvider := func() reader.RNibReader {
183 writerMock := &mocks.RnibWriterMock{}
184 writerProvider := func() rNibWriter.RNibWriter {
187 config := configuration.ParseConfiguration()
189 controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
191 header := http.Header{}
192 header.Set("Content-Type", "application/json")
193 result := controller.validateRequestHeader(&header)
195 assert.Nil(t, result)
198 func parseJsonRequest(t *testing.T, r io.Reader) models.ErrorResponse {
200 var errorResponse models.ErrorResponse
201 body, err := ioutil.ReadAll(r)
203 t.Errorf("Error cannot deserialize json request")
205 json.Unmarshal(body, &errorResponse)
210 func initLog(t *testing.T) *logger.Logger {
211 log, err := logger.InitLogger(logger.InfoLevel)
213 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)