| # Copyright 2016 Mirantis, Inc. |
| # |
| # 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. |
| |
| import os |
| |
| from tcp_tests.managers.execute_commands import ExecuteCommandsMixin |
| from tcp_tests.managers.clients.prometheus import prometheus_client |
| from tcp_tests import logger |
| |
| LOG = logger.logger |
| |
| |
| class SLManager(ExecuteCommandsMixin): |
| """docstring for OpenstackManager""" |
| |
| __config = None |
| __underlay = None |
| |
| def __init__(self, config, underlay, salt): |
| self.__config = config |
| self.__underlay = underlay |
| self._salt = salt |
| self._p_client = None |
| super(SLManager, self).__init__( |
| config=config, underlay=underlay) |
| |
| def install(self, commands): |
| self.execute_commands(commands, |
| label='Install SL services') |
| self.__config.stack_light.sl_installed = True |
| self.__config.stack_light.sl_vip_host = self.get_sl_vip() |
| |
| def get_sl_vip(self): |
| sl_vip_address_pillars = self._salt.get_pillar( |
| tgt='I@keepalived:cluster:enabled:true and not ctl*', |
| pillar='keepalived:cluster:instance:prometheus_server_vip:address') |
| sl_vip_ip = set([ip |
| for item in sl_vip_address_pillars |
| for node,ip in item.items() if ip]) |
| assert len(sl_vip_ip) == 1, ( |
| "Found more than one SL VIP in pillars:{0}, " |
| "expected one!").format(sl_vip_ip) |
| sl_vip_ip_host = sl_vip_ip.pop() |
| return sl_vip_ip_host |
| |
| @property |
| def api(self): |
| if self._p_client is None: |
| self._p_client = prometheus_client.PrometheusClient( |
| host=self.__config.stack_light.sl_vip_host, |
| port=self.__config.stack_light.sl_prometheus_port, |
| proto=self.__config.stack_light.sl_prometheus_proto) |
| return self._p_client |
| |
| def get_monitoring_nodes(self): |
| return [node_name for node_name |
| in self.__underlay.node_names() if 'mon' in node_name] |
| |
| def get_service_info_from_node(self, node_name): |
| service_stat_dict = {} |
| with self.__underlay.remote(node_name=node_name) as node_remote: |
| result = node_remote.execute( |
| "docker service ls --format '{{.Name}}:{{.Replicas}}'") |
| LOG.debug("Service ls result {0} from node {1}".format( |
| result['stdout'], node_name)) |
| for line in result['stdout']: |
| tmp = line.split(':') |
| service_stat_dict.update({tmp[0]: tmp[1]}) |
| return service_stat_dict |
| |
| def run_sl_functional_tests(self, node_to_run, path_tests_to_run): |
| target_node_name = [node_name for node_name |
| in self.__underlay.node_names() |
| if node_to_run in node_name] |
| with self.__underlay.remote(node_name=target_node_name[0]) as node_remote: |
| cmd = "python -k {}".format(path_tests_to_run) |
| result = node_remote.execute(cmd) |
| LOG.debug("Test execution result is {}".format(result)) |
| return result |
| |
| def download_sl_test_report(self, stored_node, file_path): |
| target_node_name = [node_name for node_name |
| in self.__underlay.node_names() |
| if stored_node in node_name] |
| with self.__underlay.remote(node_name=target_node_name[0]) as r: |
| r.download( |
| destination=file_path, |
| target=os.getcwd()) |