Fix license headers
[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         log, err := logger.InitLogger(logger.InfoLevel)
35         if err != nil {
36                 t.Errorf("#test.utils.InitLog - failed to initialize logger, error: %s", err)
37         }
38         return log
39 }
40
41 func InitRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
42         rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
43         rmrMessengerMock.On("Init", GetPort(), MaxMsgSize, Flags, log).Return(&rmrMessenger)
44         return rmrsender.NewRmrSender(log, rmrMessenger)
45 }
46
47 func InitRnibDataService(t *testing.T) services.RNibDataService{
48
49         logger := InitLog(t)
50         config := InitConfiguration()
51
52         readerMock := &mocks.RnibReaderMock{}
53         writerMock := &mocks.RnibWriterMock{}
54
55         return services.NewRnibDataService(logger, config, readerMock, writerMock)
56 }
57
58 func InitConfiguration() *configuration.Configuration{
59         return &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3}
60 }
61
62