Improve unit test coverage

[ric-plt/e2mgr.git] / E2Manager / models / e2_setup_request_message_test.go

diff --git a/E2Manager/models/e2_setup_request_message_test.go b/E2Manager/models/e2_setup_request_message_test.go
new file mode 100644 (file)
index 0000000..8252c71
--- /dev/null
+++ b/E2Manager/models/e2_setup_request_message_test.go
@@ -0,0 +1,111 @@
+//
+// Copyright 2020 AT&T Intellectual Property
+// Copyright 2020 Nokia
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//  platform project (RICP).
+
+package models_test
+
+import (
+       "e2mgr/models"
+       "e2mgr/utils"
+       "encoding/xml"
+       "github.com/stretchr/testify/assert"
+       "testing"
+)
+
+const (
+       e2SetupReqGnbSetupRequestXmlPath   = "../tests/resources/setupRequest/setupRequest_gnb.xml"
+       e2SetupReqEnGnbSetupRequestXmlPath = "../tests/resources/setupRequest/setupRequest_en-gNB.xml"
+       e2SetupReqEnbSetupRequestXmlPath   = "../tests/resources/setupRequest/setupRequest_enb.xml"
+       e2SetupReqNgEnbSetupRequestXmlPath = "../tests/resources/setupRequest/setupRequest_ng-eNB.xml"
+)
+
+func getTestE2SetupRequest(t *testing.T, reqXmlPath string) *models.E2SetupRequestMessage {
+       xmlGnb := utils.ReadXmlFile(t, reqXmlPath)
+       setupRequest := &models.E2SetupRequestMessage{}
+       err := xml.Unmarshal(utils.NormalizeXml(xmlGnb), &setupRequest.E2APPDU)
+       assert.Nil(t, err)
+       return setupRequest
+}
+
+func TestExtractRanFunctionsListFromGnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqGnbSetupRequestXmlPath)
+
+       ranFuncList := setupRequest.ExtractRanFunctionsList()
+       assert.Equal(t, uint32(1), ranFuncList[0].RanFunctionId)
+       assert.Equal(t, uint32(2), ranFuncList[1].RanFunctionId)
+       assert.Equal(t, uint32(3), ranFuncList[2].RanFunctionId)
+       assert.Equal(t, uint32(1), ranFuncList[0].RanFunctionRevision)
+       assert.Equal(t, uint32(1), ranFuncList[1].RanFunctionRevision)
+       assert.Equal(t, uint32(1), ranFuncList[2].RanFunctionRevision)
+}
+
+func TestGetPlmnIdFromGnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqGnbSetupRequestXmlPath)
+
+       plmnID := setupRequest.GetPlmnId()
+       assert.Equal(t, "02F829", plmnID)
+}
+
+func TestGetPlmnIdFromEnGnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqEnGnbSetupRequestXmlPath)
+
+       plmnID := setupRequest.GetPlmnId()
+       assert.Equal(t, "131014", plmnID)
+}
+
+func TestGetPlmnIdFromEnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqEnbSetupRequestXmlPath)
+
+       plmnID := setupRequest.GetPlmnId()
+       assert.Equal(t, "6359AB", plmnID)
+}
+
+func TestGetPlmnIdFromNgEnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqNgEnbSetupRequestXmlPath)
+
+       plmnID := setupRequest.GetPlmnId()
+       assert.Equal(t, "131014", plmnID)
+}
+
+func TestGetNbIdFromGnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqGnbSetupRequestXmlPath)
+
+       nbID := setupRequest.GetNbId()
+       assert.Equal(t, "001100000011000000110000", nbID)
+}
+
+func TestGetNbIdFromEnGnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqEnGnbSetupRequestXmlPath)
+
+       nbID := setupRequest.GetNbId()
+       assert.Equal(t, "11000101110001101100011111111000", nbID)
+}
+
+func TestGetNbIdFromEnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqEnbSetupRequestXmlPath)
+
+       nbID := setupRequest.GetNbId()
+       assert.Equal(t, "101010101010101010", nbID)
+}
+
+func TestGetNbIdFromNgEnbRequestSuccess(t *testing.T) {
+       setupRequest := getTestE2SetupRequest(t, e2SetupReqNgEnbSetupRequestXmlPath)
+
+       nbID := setupRequest.GetNbId()
+       assert.Equal(t, "101010101010101010", nbID)
+}