[RICPLT-2165] Add rnibDataService to support retries
[ric-plt/e2mgr.git] / E2Manager / providers / httpmsghandlerprovider / incoming_request_handler_provider_test.go
1 //
2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
4 //
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
8 //
9 //      http://www.apache.org/licenses/LICENSE-2.0
10 //
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.
16 //
17
18 package httpmsghandlerprovider
19
20 import (
21         "e2mgr/configuration"
22         "e2mgr/e2managererrors"
23         "e2mgr/handlers/httpmsghandlers"
24         "e2mgr/logger"
25         "e2mgr/managers"
26         "e2mgr/mocks"
27         "e2mgr/models"
28         "e2mgr/rNibWriter"
29         "e2mgr/rmrCgo"
30         "e2mgr/services"
31         "e2mgr/tests"
32         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
33         "github.com/stretchr/testify/assert"
34         "reflect"
35         "testing"
36 )
37
38 func getRmrService(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *services.RmrService {
39         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
40         messageChannel := make(chan *models.NotificationResponse)
41         rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
42         return services.NewRmrService(services.NewRmrConfig(tests.Port, tests.MaxMsgSize, tests.Flags, log), rmrMessenger, messageChannel)
43 }
44
45 func setupTest(t *testing.T) *IncomingRequestHandlerProvider {
46         rmrMessengerMock := &mocks.RmrMessengerMock{}
47         log := initLog(t)
48         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
49         readerProvider := func() reader.RNibReader {
50                 return &mocks.RnibReaderMock{}
51         }
52         writerProvider := func() rNibWriter.RNibWriter {
53                 return &mocks.RnibWriterMock{}
54         }
55         rnibDataService := services.NewRnibDataService(log, config, readerProvider, writerProvider)
56         ranSetupManager := managers.NewRanSetupManager(log, getRmrService(rmrMessengerMock, log), rnibDataService)
57         return NewIncomingRequestHandlerProvider(log, getRmrService(rmrMessengerMock, log), configuration.ParseConfiguration(), rnibDataService, ranSetupManager)
58 }
59
60 func TestNewIncomingRequestHandlerProvider(t *testing.T) {
61         provider := setupTest(t)
62
63         assert.NotNil(t, provider)
64 }
65
66 func TestShutdownRequestHandler(t *testing.T) {
67         provider := setupTest(t)
68         handler, err := provider.GetHandler(ShutdownRequest)
69
70         assert.NotNil(t, provider)
71         assert.Nil(t, err)
72
73         _, ok := handler.(*httpmsghandlers.DeleteAllRequestHandler)
74
75         assert.True(t, ok)
76 }
77
78 func TestX2SetupRequestHandler(t *testing.T) {
79         provider := setupTest(t)
80         handler, err := provider.GetHandler(X2SetupRequest)
81
82         assert.NotNil(t, provider)
83         assert.Nil(t, err)
84
85         _, ok := handler.(*httpmsghandlers.SetupRequestHandler)
86
87         assert.True(t, ok)
88 }
89
90 func TestEndcSetupRequestHandler(t *testing.T) {
91         provider := setupTest(t)
92         handler, err := provider.GetHandler(EndcSetupRequest)
93
94         assert.NotNil(t, provider)
95         assert.Nil(t, err)
96
97         _, ok := handler.(*httpmsghandlers.SetupRequestHandler)
98
99         assert.True(t, ok)
100 }
101
102 func TestGetShutdownHandlerFailure(t *testing.T) {
103         provider := setupTest(t)
104         _, actual := provider.GetHandler("test")
105         expected := &e2managererrors.InternalError{}
106
107         assert.NotNil(t, actual)
108         if reflect.TypeOf(actual) != reflect.TypeOf(expected) {
109                 t.Errorf("Error actual = %v, and Expected = %v.", actual, expected)
110         }
111 }
112
113 func initLog(t *testing.T) *logger.Logger {
114         log, err := logger.InitLogger(logger.InfoLevel)
115         if err != nil {
116                 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
117         }
118         return log
119 }