Coverage for osm_mon/collector/infra_collectors/onos.py: 34%

1# Copyright 2018 Whitestack, LLC 

2# ************************************************************* 

3 

4# This file is part of OSM Monitoring module 

5# All Rights Reserved to Whitestack, LLC 

6 

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

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

9# a copy of the License at 

10 

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

12 

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

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

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

16# License for the specific language governing permissions and limitations 

17# under the License. 

18 

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

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

21## 

22import logging 

23from typing import List 

24 

25import requests 

26from requests.auth import HTTPBasicAuth 

27 

28from osm_mon.collector.infra_collectors.base_sdnc import BaseSdncInfraCollector 

29from osm_mon.collector.metric import Metric 

30from osm_mon.core.common_db import CommonDbClient 

31from osm_mon.core.config import Config 

32 

33log = logging.getLogger(__name__) 

34 

35 

36class OnosInfraCollector(BaseSdncInfraCollector): 

37 def __init__(self, config: Config, sdnc_id: str): 

38 super().__init__(config, sdnc_id) 

39 self.common_db = CommonDbClient(config) 

40 self.sdnc = self.common_db.get_sdnc(sdnc_id) 

41 

42 def _obtain_url(self, sdnc_dict): 

43 url = sdnc_dict.get("url") 

44 if url: 

45 return url 

46 else: 

47 if not sdnc_dict.get("ip") or not sdnc_dict.get("port"): 

48 raise Exception("You must provide a URL to contact the SDN Controller") 

49 else: 

50 return "http://{}:{}/onos/v1/devices".format( 

51 sdnc_dict["ip"], sdnc_dict["port"] 

52 ) 

53 

54 def collect(self) -> List[Metric]: 

55 metrics = [] 

56 sdnc_status = self.is_sdnc_ok() 

57 if self.sdnc["_admin"]["projects_read"]: 

58 sdnc_project_id = self.sdnc["_admin"]["projects_read"][0] 

59 else: 

60 sdnc_project_id = "" 

61 sdnc_tags = {"sdnc_id": self.sdnc["_id"], "project_id": sdnc_project_id} 

62 sdnc_status_metric = Metric(sdnc_tags, "sdnc_status", sdnc_status) 

63 metrics.append(sdnc_status_metric) 

64 

65 return metrics 

66 

67 def is_sdnc_ok(self) -> bool: 

68 try: 

69 url = self._obtain_url(self.sdnc) 

70 user = self.sdnc["user"] 

71 password = self.common_db.decrypt_sdnc_password( 

72 self.sdnc["password"], self.sdnc["schema_version"], self.sdnc["_id"] 

73 ) 

74 

75 requests.get(url, auth=HTTPBasicAuth(user, password)) 

76 return True 

77 except Exception: 

78 log.exception("SDNC status is not OK!") 

79 return False