Adding new comments for Oran in all files with licenses
[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 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 //  platform project (RICP).
19
20
21 package httpmsghandlerprovider
22
23 import (
24         "e2mgr/configuration"
25         "e2mgr/e2managererrors"
26         "e2mgr/handlers/httpmsghandlers"
27         "e2mgr/logger"
28         "e2mgr/managers"
29         "e2mgr/mocks"
30         "e2mgr/rNibWriter"
31         "e2mgr/rmrCgo"
32         "e2mgr/services"
33         "e2mgr/services/rmrsender"
34         "e2mgr/tests"
35         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
36         "github.com/stretchr/testify/assert"
37         "reflect"
38         "testing"
39 )
40
41 func getRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
42         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
43         rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
44         return rmrsender.NewRmrSender(log, &rmrMessenger)
45 }
46
47 func setupTest(t *testing.T) *IncomingRequestHandlerProvider {
48         rmrMessengerMock := &mocks.RmrMessengerMock{}
49         log := initLog(t)
50         config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
51         readerProvider := func() reader.RNibReader {
52                 return &mocks.RnibReaderMock{}
53         }
54         writerProvider := func() rNibWriter.RNibWriter {
55                 return &mocks.RnibWriterMock{}
56         }
57         rnibDataService := services.NewRnibDataService(log, config, readerProvider, writerProvider)
58         rmrSender := getRmrSender(rmrMessengerMock, log)
59         ranSetupManager := managers.NewRanSetupManager(log, rmrSender, rnibDataService)
60         return NewIncomingRequestHandlerProvider(log, rmrSender, configuration.ParseConfiguration(), rnibDataService, ranSetupManager)
61 }
62
63 func TestNewIncomingRequestHandlerProvider(t *testing.T) {
64         provider := setupTest(t)
65
66         assert.NotNil(t, provider)
67 }
68
69 func TestShutdownRequestHandler(t *testing.T) {
70         provider := setupTest(t)
71         handler, err := provider.GetHandler(ShutdownRequest)
72
73         assert.NotNil(t, provider)
74         assert.Nil(t, err)
75
76         _, ok := handler.(*httpmsghandlers.DeleteAllRequestHandler)
77
78         assert.True(t, ok)
79 }
80
81 func TestX2SetupRequestHandler(t *testing.T) {
82         provider := setupTest(t)
83         handler, err := provider.GetHandler(X2SetupRequest)
84
85         assert.NotNil(t, provider)
86         assert.Nil(t, err)
87
88         _, ok := handler.(*httpmsghandlers.SetupRequestHandler)
89
90         assert.True(t, ok)
91 }
92
93 func TestEndcSetupRequestHandler(t *testing.T) {
94         provider := setupTest(t)
95         handler, err := provider.GetHandler(EndcSetupRequest)
96
97         assert.NotNil(t, provider)
98         assert.Nil(t, err)
99
100         _, ok := handler.(*httpmsghandlers.SetupRequestHandler)
101
102         assert.True(t, ok)
103 }
104
105 func TestGetShutdownHandlerFailure(t *testing.T) {
106         provider := setupTest(t)
107         _, actual := provider.GetHandler("test")
108         expected := &e2managererrors.InternalError{}
109
110         assert.NotNil(t, actual)
111         if reflect.TypeOf(actual) != reflect.TypeOf(expected) {
112                 t.Errorf("Error actual = %v, and Expected = %v.", actual, expected)
113         }
114 }
115
116 func initLog(t *testing.T) *logger.Logger {
117         log, err := logger.InitLogger(logger.InfoLevel)
118         if err != nil {
119                 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
120         }
121         return log
122 }