RIC:1060: Change in PTL
[ric-plt/e2mgr.git] / E2Manager / tests / utils.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 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
19 //  platform project (RICP).
20
21 package tests
22
23 import (
24         "e2mgr/configuration"
25         "e2mgr/logger"
26         "e2mgr/mocks"
27         "e2mgr/rmrCgo"
28         "e2mgr/services"
29         "e2mgr/services/rmrsender"
30         "testing"
31 )
32
33 func InitLog(t *testing.T) *logger.Logger {
34         InfoLevel := int8(1)
35         log, err := logger.InitLogger(InfoLevel)
36         if err != nil {
37                 t.Errorf("#test.utils.InitLog - failed to initialize logger, error: %s", err)
38         }
39         return log
40 }
41
42 func InitRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
43         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
44         rmrMessengerMock.On("Init", GetPort(), MaxMsgSize, Flags, log).Return(&rmrMessenger)
45         return rmrsender.NewRmrSender(log, rmrMessenger)
46 }
47
48 func InitRnibDataService(t *testing.T) services.RNibDataService{
49
50         logger := InitLog(t)
51         config := InitConfiguration()
52
53         readerMock := &mocks.RnibReaderMock{}
54         writerMock := &mocks.RnibWriterMock{}
55
56         return services.NewRnibDataService(logger, config, readerMock, writerMock)
57 }
58
59 func InitConfiguration() *configuration.Configuration{
60         return &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
61 }
62
63