Squash-merging e2ap-v2.0 branch
[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         "testing"
27
28         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
29         "github.com/stretchr/testify/assert"
30 )
31
32 func getTestRicServiceQueryRanFunctions(t *testing.T) []*entities.RanFunction {
33         gnbSetupRequestXmlPath := "../tests/resources/setupRequest/setupRequest_gnb.xml"
34         xmlgnb := utils.ReadXmlFile(t, gnbSetupRequestXmlPath)
35
36         setupRequest := &models.E2SetupRequestMessage{}
37         err := xml.Unmarshal(utils.NormalizeXml(xmlgnb), &setupRequest.E2APPDU)
38         if err != nil {
39                 t.Fatal(err)
40         }
41
42         ranFunctionList := setupRequest.ExtractRanFunctionsList()
43         return ranFunctionList
44 }
45
46 func TestRicServiceQueryMessageSuccess(t *testing.T) {
47         ranFunctionList := getTestRicServiceQueryRanFunctions(t)
48
49         serviceQuery := models.NewRicServiceQueryMessage(ranFunctionList)
50         initMsg := serviceQuery.E2APPDU.InitiatingMessage
51         assert.Equal(t, models.ProcedureCode_id_RICserviceQuery, initMsg.ProcedureCode)
52         assert.Equal(t, models.ProtocolIE_ID_id_TransactionID, initMsg.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[0].Id)
53         assert.Equal(t, models.ProtocolIE_ID_id_RANfunctionsAccepted, initMsg.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[1].Id)
54         assert.Equal(t, 3, len(initMsg.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[1].Value.(models.RICServiceQueryRANFunctionIdList).RANFunctionIdList.ProtocolIESingleContainer))
55 }
56
57 func TestTransactionIdServiceQuery(t *testing.T) {
58         ranFunctionList := getTestRicServiceQueryRanFunctions(t)
59         serviceQuery := models.NewRicServiceQueryMessage(ranFunctionList)
60         txIE := serviceQuery.E2APPDU.InitiatingMessage.Value.RICServiceQuery.ProtocolIEs.RICServiceQueryIEs[0].Value.(models.RICServiceQueryTransactionID)
61         assert.NotEmpty(t, txIE.TransactionID, "transaction ID should not be empty")
62 }