Coverage for osm_policy_module/utils/vnfd.py: 100%

1# -*- coding: utf-8 -*- 

2 

3# Copyright 2018 Whitestack, LLC 

4# ************************************************************* 

5 

6# This file is part of OSM Monitoring module 

7# All Rights Reserved to Whitestack, LLC 

8 

9# Licensed under the Apache License, Version 2.0 (the "License"); you may 

10# not use this file except in compliance with the License. You may obtain 

11# a copy of the License at 

12 

13# http://www.apache.org/licenses/LICENSE-2.0 

14 

15# Unless required by applicable law or agreed to in writing, software 

16# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

17# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

18# License for the specific language governing permissions and limitations 

19# under the License. 

20 

21# For those usages not covered by the Apache License, Version 2.0 please 

22# contact: bdiaz@whitestack.com or glavado@whitestack.com 

23## 

24from osm_policy_module.core.exceptions import ManagementVduNotFound 

25 

26 

27class VnfdUtils: 

28 @staticmethod 

29 def get_mgmt_vdu(vnfd: dict): 

30 if "mgmt-cp" in vnfd: 

31 mgmt_cp = vnfd["mgmt-cp"] 

32 mgmt_cp_vdu_id = None 

33 for cpd in vnfd.get("ext-cpd", ()): 

34 if cpd.get("id") == mgmt_cp: 

35 mgmt_cp_vdu_id = cpd.get("int-cpd", {}).get("vdu-id") 

36 for vdu in vnfd.get("vdu", ()): 

37 if vdu.get("id") == mgmt_cp_vdu_id: 

38 return vdu 

39 raise ManagementVduNotFound("Management vdu not found in vnfd %s", vnfd["id"])