3ed2fbc18a2913292575a1776467ff5cd9544d81
[ric-plt/e2mgr.git] / E2Manager / models / ric_service_query_message_test.go
1 //
2 // Copyright 2020 AT&T Intellectual Property
3 // Copyright 2020 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 package models_test
21
22 import (
23         "e2mgr/models"
24         "e2mgr/utils"
25         "encoding/xml"
26         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
27         "github.com/stretchr/testify/assert"
28         "testing"
29 )
30
31 func getTestRicServiceQueryRanFunctions(t *testing.T) []*entities.RanFunction {
32         gnbSetupRequestXmlPath := "../tests/resources/setupRequest/setupRequest_gnb.xml"
33         xmlgnb := utils.ReadXmlFile(t, gnbSetupRequestXmlPath)
34
35         setupRequest := &models.E2SetupRequestMessage{}
36         err := xml.Unmarshal(utils.NormalizeXml(xmlgnb), &setupRequest.E2APPDU)
37         if err != nil {
38                 t.Fatal(err)
39         }
40
41         ranFunctionList := setupRequest.ExtractRanFunctionsList()
42         return ranFunctionList
43 }
44
45 func TestRicServiceQueryMessageSuccess(t *testing.T) {
46         ranFunctionList := getTestRicServiceQueryRanFunctions(t)
47
48         serviceQuery := models.NewRicServiceQueryMessage(ranFunctionList)
49         initMsg := serviceQuery.E2APPDU.InitiatingMessage
50         assert.Equal(t, "6", initMsg.ProcedureCode)
51         assert.Equal(t, "9", initMsg.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[0].Id)
52         assert.Equal(t, 3, len(initMsg.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[0].Value.RANFunctionIdList.ProtocolIESingleContainer))
53 }