RSM/rmrcgo/rmr_c_go_api_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 rmrcgo
  22
  23 import (
  24         "bytes"
  25         "encoding/json"
  26         "github.com/stretchr/testify/assert"
  27         "io/ioutil"
  28         "rsm/logger"
  29         "rsm/tests"
  30         "strconv"
  31         "testing"
  32 )
  33
  34 var (
  35         msgr RmrMessenger
  36 )
  37
  38 func TestLogger(t *testing.T) {
  39         log := initLog(t)
  40         data := map[string]interface{}{"messageType": 1001, "ranIp": "10.0.0.3", "ranPort": 879, "ranName": "test1"}
  41         b := new(bytes.Buffer)
  42         _ = json.NewEncoder(b).Encode(data)
  43         req := tests.GetHttpRequest()
  44         boo, _ := ioutil.ReadAll(req.Body)
  45         log.Debugf("#rmr_c_go_api_test.TestLogger - request header: %v\n; request body: %s\n", req.Header, string(boo))
  46 }
  47
  48 func TestNewMBufSuccess(t *testing.T) {
  49         msg := NewMBuf(tests.MessageType, len(tests.DummyPayload), "RanName", &tests.DummyPayload, &tests.DummyXAction)
  50         assert.NotNil(t, msg)
  51         assert.NotEmpty(t, msg.Payload)
  52         assert.NotEmpty(t, msg.XAction)
  53         assert.Equal(t, msg.MType, tests.MessageType)
  54         assert.Equal(t, msg.Meid, "RanName")
  55         assert.Equal(t, msg.Len, len(tests.DummyPayload))
  56 }
  57
  58 func TestSendRecvMsgSuccess(t *testing.T) {
  59         log := initLog(t)
  60
  61         initRmr(tests.ReadyIntervalSec, tests.GetPort(), tests.MaxMsgSize, tests.Flags, log)
  62         if msgr == nil || !msgr.IsReady() {
  63                 t.Errorf("#rmr_c_go_api_test.TestSendRecvMsgSuccess - The rmr router is not ready")
  64         }
  65
  66         msg := NewMBuf(1, tests.MaxMsgSize, "test 1", &tests.DummyPayload, &tests.DummyXAction)
  67         log.Debugf("#rmr_c_go_api_test.TestSendRecvMsgSuccess - Going to send the message: %#v\n", msg)
  68         result, err := msgr.SendMsg(msg)
  69
  70         assert.Nil(t, err)
  71         assert.NotNil(t, result)
  72
  73         msgR, err := msgr.RecvMsg()
  74
  75         assert.Nil(t, err)
  76         assert.NotNil(t, msgR)
  77         msgr.Close()
  78 }
  79
  80 func TestSendMsgRmrInvalidPortError(t *testing.T) {
  81         log := initLog(t)
  82
  83         initRmr(tests.ReadyIntervalSec, "tcp:"+strconv.Itoa(5555), tests.MaxMsgSize, tests.Flags, log)
  84         if msgr == nil || !msgr.IsReady() {
  85                 t.Errorf("#rmr_c_go_api_test.TestSendMsgRmrInvalidPortError - The rmr router is not ready")
  86         }
  87
  88         msg := NewMBuf(1, tests.MaxMsgSize, "test 1", &tests.DummyPayload, &tests.DummyXAction)
  89         log.Debugf("#rmr_c_go_api_test.TestSendMsgRmrInvalidPortError - Going to send the message: %#v\n", msg)
  90         result, err := msgr.SendMsg(msg)
  91
  92         assert.NotNil(t, err)
  93         assert.Nil(t, result)
  94
  95         msgr.Close()
  96 }
  97
  98 func TestSendMsgRmrInvalidMsgNumError(t *testing.T) {
  99         log := initLog(t)
 100
 101         initRmr(tests.ReadyIntervalSec, tests.GetPort(), tests.MaxMsgSize, tests.Flags, log)
 102         if msgr == nil || !msgr.IsReady() {
 103                 t.Errorf("#rmr_c_go_api_test.TestSendMsgRmrInvalidMsgNumError - The rmr router is not ready")
 104         }
 105
 106         msg := NewMBuf(10, tests.MaxMsgSize, "test 1", &tests.DummyPayload, &tests.DummyXAction)
 107         log.Debugf("#rmr_c_go_api_test.TestSendMsgRmrInvalidMsgNumError - Going to send the message: %#v\n", msg)
 108         result, err := msgr.SendMsg(msg)
 109
 110         assert.NotNil(t, err)
 111         assert.Nil(t, result)
 112
 113         msgr.Close()
 114 }
 115
 116 func TestIsReadySuccess(t *testing.T) {
 117         log := initLog(t)
 118
 119         initRmr(tests.ReadyIntervalSec, tests.GetPort(), tests.MaxMsgSize, tests.Flags, log)
 120         if msgr == nil || !msgr.IsReady() {
 121                 t.Errorf("#rmr_c_go_api_test.TestIsReadySuccess - The rmr router is not ready")
 122         }
 123
 124         msgr.Close()
 125 }
 126
 127 func initRmr(readyIntervalSec int, port string, maxMsgSize int, flags int, log *logger.Logger) {
 128         var ctx *Context
 129         msgr = ctx.Init(readyIntervalSec, port, maxMsgSize, flags, log)
 130 }
 131
 132 func initLog(t *testing.T) *logger.Logger {
 133         log, err := logger.InitLogger(logger.DebugLevel)
 134         if err != nil {
 135                 t.Errorf("#rmr_c_go_api_test.initLog - failed to initialize logger, error: %s", err)
 136         }
 137         return log
 138 }