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 readerMock := &mocks.RnibReaderMock{}
46 readerProvider := func() reader.RNibReader {
49 writerMock := &mocks.RnibWriterMock{}
50 writerProvider := func() rNibWriter.RNibWriter {
53 var messageChannel chan<- *models.NotificationResponse
55 rnibErr := &common.RNibError{}
56 var nbIdentityList []*entities.NbIdentity
57 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
59 writer := httptest.NewRecorder()
60 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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 readerMock := &mocks.RnibReaderMock{}
74 readerProvider := func() reader.RNibReader {
77 writerMock := &mocks.RnibWriterMock{}
78 writerProvider := func() rNibWriter.RNibWriter {
81 var messageChannel chan<- *models.NotificationResponse
83 writer := httptest.NewRecorder()
85 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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 readerMock := &mocks.RnibReaderMock{}
103 readerProvider := func() reader.RNibReader {
106 writerMock := &mocks.RnibWriterMock{}
107 writerProvider := func() rNibWriter.RNibWriter {
110 config := configuration.ParseConfiguration()
111 var messageChannel chan<- *models.NotificationResponse
113 var rnibError common.IRNibError
114 nbIdentityList := []*entities.NbIdentity{}
115 readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
117 writer := httptest.NewRecorder()
118 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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 readerMock := &mocks.RnibReaderMock{}
129 readerProvider := func() reader.RNibReader {
132 writerMock := &mocks.RnibWriterMock{}
133 writerProvider := func() rNibWriter.RNibWriter {
136 var messageChannel chan<- *models.NotificationResponse
138 writer := httptest.NewRecorder()
139 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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 readerMock := &mocks.RnibReaderMock{}
155 readerProvider := func() reader.RNibReader {
158 writerMock := &mocks.RnibWriterMock{}
159 writerProvider := func() rNibWriter.RNibWriter {
162 var messageChannel chan<- *models.NotificationResponse
163 writer := httptest.NewRecorder()
164 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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 readerMock := &mocks.RnibReaderMock{}
179 readerProvider := func() reader.RNibReader {
182 writerMock := &mocks.RnibWriterMock{}
183 writerProvider := func() rNibWriter.RNibWriter {
186 config := configuration.ParseConfiguration()
187 var messageChannel chan<- *models.NotificationResponse
189 controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
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)