sync from Azure to LF
[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/rNibWriter"
28         "e2mgr/rmrCgo"
29         "e2mgr/services"
30         "e2mgr/services/rmrsender"
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 getRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
39         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
40         rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
41         return rmrsender.NewRmrSender(log, &rmrMessenger)
42 }
43
44 func setupTest(t *testing.T) *IncomingRequestHandlerProvider {
45         rmrMessengerMock := &mocks.RmrMessengerMock{}
46         log := initLog(t)
47         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
48         readerProvider := func() reader.RNibReader {
49                 return &mocks.RnibReaderMock{}
50         }
51         writerProvider := func() rNibWriter.RNibWriter {
52                 return &mocks.RnibWriterMock{}
53         }
54         rnibDataService := services.NewRnibDataService(log, config, readerProvider, writerProvider)
55         rmrSender := getRmrSender(rmrMessengerMock, log)
56         ranSetupManager := managers.NewRanSetupManager(log, rmrSender, rnibDataService)
57         return NewIncomingRequestHandlerProvider(log, rmrSender, 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 }