[RICPLT-1853] Add UT & more
[ric-plt/e2mgr.git] / E2Manager / providers / rmrmsghandlerprovider / notification_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 rmrmsghandlerprovider
19
20 import (
21         "e2mgr/configuration"
22         "e2mgr/logger"
23         "e2mgr/managers"
24         "e2mgr/mocks"
25         "e2mgr/models"
26         "e2mgr/rNibWriter"
27         "e2mgr/services"
28         "e2mgr/sessions"
29         "e2mgr/tests"
30         "fmt"
31         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
32         "strings"
33         "testing"
34
35         "e2mgr/handlers"
36         "e2mgr/rmrCgo"
37 )
38
39 /*
40  * Verify support for known providers.
41  */
42
43 func TestGetNotificationHandlerSuccess(t *testing.T) {
44
45         logger := initLog(t)
46         rmrService := getRmrService(&mocks.RmrMessengerMock{}, logger)
47
48         readerMock := &mocks.RnibReaderMock{}
49         rnibReaderProvider := func() reader.RNibReader {
50                 return readerMock
51         }
52         writerMock := &mocks.RnibWriterMock{}
53         rnibWriterProvider := func() rNibWriter.RNibWriter {
54                 return writerMock
55         }
56
57         ranReconnectionManager := managers.NewRanReconnectionManager(logger, configuration.ParseConfiguration(), rnibReaderProvider, rnibWriterProvider, rmrService)
58
59         var testCases = []struct {
60                 msgType int
61                 handler handlers.NotificationHandler
62         }{
63                 {rmrCgo.RIC_X2_SETUP_RESP /*successful x2 setup response*/, handlers.X2SetupResponseNotificationHandler{}},
64                 {rmrCgo.RIC_X2_SETUP_FAILURE /*unsuccessful x2 setup response*/, handlers.X2SetupFailureResponseNotificationHandler{}},
65                 {rmrCgo.RIC_ENDC_X2_SETUP_RESP /*successful en-dc x2 setup response*/, handlers.EndcX2SetupResponseNotificationHandler{}},
66                 {rmrCgo.RIC_ENDC_X2_SETUP_FAILURE /*unsuccessful en-dc x2 setup response*/, handlers.EndcX2SetupFailureResponseNotificationHandler{}},
67                 {rmrCgo.RIC_SCTP_CONNECTION_FAILURE /*sctp errors*/, handlers.NewRanLostConnectionHandler(ranReconnectionManager)},
68                 {rmrCgo.RIC_ENB_LOAD_INFORMATION, handlers.NewEnbLoadInformationNotificationHandler(rnibWriterProvider)},
69                 {rmrCgo.RIC_ENB_CONF_UPDATE, handlers.X2EnbConfigurationUpdateHandler{}},
70                 {rmrCgo.RIC_ENDC_CONF_UPDATE, handlers.EndcConfigurationUpdateHandler{}},
71         }
72
73         for _, tc := range testCases {
74
75                 provider := NewNotificationHandlerProvider(rnibReaderProvider, rnibWriterProvider, ranReconnectionManager)
76                 t.Run(fmt.Sprintf("%d", tc.msgType), func(t *testing.T) {
77                         handler, err := provider.GetNotificationHandler(tc.msgType)
78                         if err != nil {
79                                 t.Errorf("want: handler for message type %d, got: error %s", tc.msgType, err)
80                         }
81                         //Note struct is empty, so it will match any other empty struct.
82                         // https://golang.org/ref/spec#Comparison_operators: Struct values are comparable if all their fields are comparable. Two struct values are equal if their corresponding non-blank fields are equal.
83                         if /*handler != tc.handler &&*/ strings.Compare(fmt.Sprintf("%T", handler), fmt.Sprintf("%T", tc.handler)) != 0 {
84                                 t.Errorf("want: handler %T for message type %d, got: %T", tc.handler, tc.msgType, handler)
85                         }
86                 })
87         }
88 }
89
90 /*
91  * Verify handling of a request for an unsupported message.
92  */
93
94 func TestGetNotificationHandlerFailure(t *testing.T) {
95
96         logger := initLog(t)
97         rmrService := getRmrService(&mocks.RmrMessengerMock{}, logger)
98
99         var testCases = []struct {
100                 msgType   int
101                 errorText string
102         }{
103                 {9999 /*unknown*/, "notification handler not found"},
104         }
105         for _, tc := range testCases {
106                 readerMock := &mocks.RnibReaderMock{}
107                 rnibReaderProvider := func() reader.RNibReader {
108                         return readerMock
109                 }
110                 writerMock := &mocks.RnibWriterMock{}
111                 rnibWriterProvider := func() rNibWriter.RNibWriter {
112                         return writerMock
113                 }
114
115                 ranReconnectionManager := managers.NewRanReconnectionManager(logger, configuration.ParseConfiguration(), rnibReaderProvider, rnibWriterProvider, rmrService)
116
117                 provider := NewNotificationHandlerProvider(rnibReaderProvider, rnibWriterProvider, ranReconnectionManager)
118                 t.Run(fmt.Sprintf("%d", tc.msgType), func(t *testing.T) {
119                         _, err := provider.GetNotificationHandler(tc.msgType)
120                         if err == nil {
121                                 t.Errorf("want: no handler for message type %d, got: success", tc.msgType)
122                         }
123                         if !strings.Contains(fmt.Sprintf("%s", err), tc.errorText) {
124                                 t.Errorf("want: error [%s] for message type %d, got: %s", tc.errorText, tc.msgType, err)
125                         }
126                 })
127         }
128 }
129
130 // TODO: extract to test_utils
131 func getRmrService(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *services.RmrService {
132         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
133         messageChannel := make(chan *models.NotificationResponse)
134         rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
135         return services.NewRmrService(services.NewRmrConfig(tests.Port, tests.MaxMsgSize, tests.Flags, log), rmrMessenger, make(sessions.E2Sessions), messageChannel)
136 }
137
138 // TODO: extract to test_utils
139 func initLog(t *testing.T) *logger.Logger {
140         log, err := logger.InitLogger(logger.InfoLevel)
141         if err != nil {
142                 t.Errorf("#delete_all_request_handler_test.TestHandleSuccessFlow - failed to initialize logger, error: %s", err)
143         }
144         return log
145 }