summaryrefslogtreecommitdiffstats
path: root/doctor_tests/installer/mcp.py
blob: 8ba9f000142b374e65a3ee81f15cfc5ecf298661 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
##############################################################################
# Copyright (c) 2018 ZTE Corporation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from os.path import isfile

from doctor_tests.common.utils import SSHClient
from doctor_tests.installer.base import BaseInstaller


class McpInstaller(BaseInstaller):
    node_user_name = 'ubuntu'
    cm_set_script = 'set_ceilometer.py'
    cm_restore_script = 'restore_ceilometer.py'

    def __init__(self, conf, log):
        super(McpInstaller, self).__init__(conf, log)
        self.key_file = self.get_ssh_key_from_installer()
        self.client = SSHClient(self.conf.installer.ip,
                                self.node_user_name,
                                key_filename=self.key_file)
        self.controllers = list()
        self.controller_clients = list()

    def setup(self):
        self.log.info('Setup MCP installer start......')

        self.controllers = self.get_controller_ips()
        self.create_flavor()
        self.set_apply_patches()
        self.setup_stunnel()

    def cleanup(self):
        self.restore_apply_patches()
        for server in self.servers:
            server.terminate()

    def get_ssh_key_from_installer(self):
        self.log.info('Get SSH keys from MCP......')

        # Assuming mcp.rsa is already mapped to functest container
        # if not, only the test runs on jumphost can get the ssh_key
        # default in path /var/lib/opnfv/mcp.rsa
        ssh_key = '/root/.ssh/id_rsa'
        mcp_key = '/var/lib/opnfv/mcp.rsa'
        return ssh_key if isfile(ssh_key) else mcp_key

    def get_controller_ips(self):
        self.log.info('Get controller ips from Mcp installer......')

        command = "sudo salt --out yaml 'ctl*' " \
                  "pillar.get _param:openstack_control_address |" \
                  "awk '{print $2}'"
        controllers = self._run_cmd_remote(self.client, command)
        self.log.info('Get controller_ips:%s from Mcp installer'
                      % controllers)
        return controllers

    def get_host_ip_from_hostname(self, hostname):
        command = "sudo salt --out yaml '%s*' " \
                  "pillar.get _param:single_address |" \
                  "awk '{print $2}'" % hostname
        host_ips = self._run_cmd_remote(self.client, command)
        return host_ips[0]

    def set_apply_patches(self):
        self.log.info('Set apply patches start......')

        restart_cm_cmd = 'sudo service ceilometer-agent-notification restart'
        for node_ip in self.controllers:
            client = SSHClient(node_ip, self.node_user_name,
                               key_filename=self.key_file)
            self.controller_clients.append(client)
            self._run_apply_patches(client,
                                    restart_cm_cmd,
                                    self.cm_set_script)

    def restore_apply_patches(self):
        self.log.info('restore apply patches start......')

        restart_cm_cmd = 'sudo service ceilometer-agent-notification restart'
        for client in self.controller_clients:
            self._run_apply_patches(client,
                                    restart_cm_cmd,
                                    self.cm_restore_script)
"o">._add_user_role_in_project(is_admin=True) self.log.info('user create end......') def _create_project(self): """create test project""" self.projects = {project.name: project for project in self.keystone.projects.list( domain=self.conf.doctor_domain_id)} if self.conf.doctor_project not in self.projects: self.log.info('create project......') test_project = \ self.keystone.projects.create( self.conf.doctor_project, self.conf.doctor_domain_id) self.projects[test_project.name] = test_project else: self.log.info('project %s already created......' % self.conf.doctor_project) self.log.info('test project %s' % str(self.projects[self.conf.doctor_project])) def _create_user(self): """create test user""" self.users = {user.name: user for user in self.keystone.users.list( domain=self.conf.doctor_domain_id)} if self.conf.doctor_user not in self.users: self.log.info('create user......') test_user = self.keystone.users.create( self.conf.doctor_user, password=self.conf.doctor_passwd, domain=self.conf.doctor_domain_id) self.users[test_user.name] = test_user else: self.log.info('user %s already created......' % self.conf.doctor_user) self.log.info('test user %s' % str(self.users[self.conf.doctor_user])) def _create_role(self): """create test role""" self.roles = {role.name: role for role in self.keystone.roles.list()} if self.conf.doctor_role not in self.roles: self.log.info('create role......') test_role = self.keystone.roles.create( self.conf.doctor_role) self.roles[test_role.name] = test_role else: self.use_exist_role = True self.log.info('role %s already created......' % self.conf.doctor_role) self.log.info('test role %s' % str(self.roles[self.conf.doctor_role])) def _add_user_role_in_project(self, is_admin=False): """add test user with test role in test project""" project = self.projects.get(self.conf.doctor_project) user_name = 'admin' if is_admin else self.conf.doctor_user user = self.users.get(user_name) role_name = 'admin' if is_admin else self.conf.doctor_role role = self.roles.get(role_name) roles_for_user = self.roles_for_admin \ if is_admin else self.roles_for_user try: self.keystone.roles.check(role, user=user, project=project) self.log.info('Already grant a role:%s to user: %s on' ' project: %s' % (role_name, user_name, self.conf.doctor_project)) except ks_exceptions.NotFound: self.keystone.roles.grant(role, user=user, project=project) roles_for_user[role_name] = role def delete(self): """delete the test user, project and role""" self.log.info('user delete start......') project = self.projects.get(self.conf.doctor_project) user = self.users.get(self.conf.doctor_user) role = self.roles.get(self.conf.doctor_role) if project: if 'admin' in self.roles_for_admin: self.keystone.roles.revoke( self.roles['admin'], user=self.users['admin'], project=project) if user: if role and self.conf.doctor_role in self.roles_for_user: self.keystone.roles.revoke( role, user=user, project=project) if not self.use_exist_role: self.keystone.roles.delete(role) self.keystone.users.delete(user) self.keystone.projects.delete(project) self.log.info('user delete end......') def update_quota(self): self.log.info('user quota update start......') project = self.projects.get(self.conf.doctor_project) user = self.users.get(self.conf.doctor_user) if project and user: self.quota = self.nova.quotas.get(project.id, user_id=user.id) if self.conf.quota_instances > self.quota.instances: self.nova.quotas.update( project.id, instances=self.conf.quota_instances, user_id=user.id) if self.conf.quota_cores > self.quota.cores: self.nova.quotas.update(project.id, cores=self.conf.quota_cores, user_id=user.id) self.log.info('user quota update end......') else: raise Exception('No project or role for update quota')