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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
##############################################################################
# Copyright (c) 2017 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
##############################################################################
import abc
import getpass
import grp
import os
import pwd
import six
import stat
import subprocess
from doctor_tests.common.utils import get_doctor_test_root_dir
from doctor_tests.identity_auth import get_session
from doctor_tests.os_clients import nova_client
@six.add_metaclass(abc.ABCMeta)
class BaseInstaller(object):
def __init__(self, conf, log):
self.conf = conf
self.log = log
self.servers = list()
@abc.abstractproperty
def node_user_name(self):
"""user name for login to cloud node"""
@abc.abstractmethod
def get_ssh_key_from_installer(self):
pass
@abc.abstractmethod
def get_host_ip_from_hostname(self, hostname):
pass
@abc.abstractmethod
def setup(self):
pass
@abc.abstractmethod
def cleanup(self):
pass
def create_flavor(self):
self.nova = \
nova_client(self.conf.nova_version,
get_session())
flavors = {flavor.name: flavor for flavor in self.nova.flavors.list()}
if self.conf.flavor not in flavors:
self.nova.flavors.create(self.conf.flavor, 512, 1, 1)
def setup_stunnel(self):
self.log.info('Setup ssh stunnel in %s installer......'
% self.conf.installer.type)
for node_ip in self.controllers:
cmd = ("ssh -o UserKnownHostsFile=/dev/null"
" -o StrictHostKeyChecking=no"
" -i %s %s@%s -R %s:localhost:%s"
" sleep 600 > ssh_tunnel.%s"
" 2>&1 < /dev/null &"
% (self.key_file,
self.node_user_name,
node_ip,
self.conf.consumer.port,
self.conf.consumer.port,
node_ip))
server = subprocess.Popen(cmd, shell=True)
self.servers.append(server)
server.communicate()
def _get_ssh_key(self, client, key_path):
self.log.info('Get SSH keys from %s installer......'
% self.conf.installer.type)
if self.key_file is not None:
self.log.info('Already have SSH keys from %s installer......'
% self.conf.installer.type)
return self.key_file
ssh_key = '{0}/{1}'.format(get_doctor_test_root_dir(), 'instack_key')
client.scp(key_path, ssh_key, method='get')
user = getpass.getuser()
uid = pwd.getpwnam(user).pw_uid
gid = grp.getgrnam(user).gr_gid
os.chown(ssh_key, uid, gid)
os.chmod(ssh_key, stat.S_IREAD)
return ssh_key
def _run_cmd_remote(self, client, command):
self.log.info('Run command=%s in %s installer......'
% (command, self.conf.installer.type))
ret, output = client.ssh(command)
if ret:
raise Exception('Exec command in %s installer failed,'
'ret=%s, output=%s'
% (self.conf.installer.type,
ret, output))
self.log.info('Output=%s command=%s in %s installer'
% (output, command, self.conf.installer.type))
return output
def _run_apply_patches(self, client, restart_cmd, script_name):
installer_dir = os.path.dirname(os.path.realpath(__file__))
script_abs_path = '{0}/{1}/{2}'.format(installer_dir,
'common', script_name)
client.scp(script_abs_path, script_name)
cmd = 'sudo python %s' % script_name
ret, output = client.ssh(cmd)
if ret:
raise Exception('Do the command in controller'
' node failed, ret=%s, cmd=%s, output=%s'
% (ret, cmd, output))
client.ssh(restart_cmd)
|
