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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
#!/usr/bin/env python
"""Script to run smoke tests on aws cli packaged installers"""
import argparse
import os
import re
import shutil
import sys
import tempfile
SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(SCRIPTS_DIR)
from utils import extract_zip, run, tmp_dir
REPO_ROOT = os.path.dirname(SCRIPTS_DIR)
DIST_DIR = os.path.join(REPO_ROOT, 'dist')
SMOKE_TEST_PATH = os.path.join(
REPO_ROOT, 'tests', 'integration', 'test_smoke.py'
)
UNINSTALL_MAC_PKG_PATH = os.path.join(
SCRIPTS_DIR, 'installers', 'uninstall-mac-pkg'
)
EXE_NAME = 'aws'
class InstallerTester:
DEFAULT_INSTALLER_LOCATION = None
def __init__(self, installer_location=None):
self._installer_location = installer_location
if self._installer_location is None:
self._installer_location = self.DEFAULT_INSTALLER_LOCATION
def __call__(self):
try:
self.setup()
return self.test()
finally:
self.cleanup()
def get_aws_cmd(self):
raise NotImplementedError('get_aws_cmd()')
def setup(self):
pass
def test(self):
env = os.environ.copy()
aws_cmd = self.get_aws_cmd()
self._assert_aws_cmd(aws_cmd)
sys.stdout.write('Setting test command to "%s"\n' % aws_cmd)
env['AWS_TEST_COMMAND'] = aws_cmd
run('nosetests -v -s %s' % SMOKE_TEST_PATH, env=env)
return 0
def cleanup(self):
pass
def _assert_aws_cmd(self, aws_cmd):
# Do a quick --version check to make sure we can execute
# the provided command
run('%s --version' % aws_cmd)
class ExeTester(InstallerTester):
DEFAULT_INSTALLER_LOCATION = os.path.join(DIST_DIR, 'awscli-exe.zip')
def __init__(self, installer_location=None):
self._installation_dir = None
self._bin_dir = None
super(ExeTester, self).__init__(installer_location=installer_location)
def setup(self):
self._installation_dir = tempfile.mkdtemp()
install_dir = os.path.join(self._installation_dir, 'aws-cli')
self._bin_dir = os.path.join(self._installation_dir, 'bin')
self._run_install_script(install_dir, self._bin_dir)
def _run_install_script(self, install_dir, bin_dir):
with tmp_dir() as workdir:
extract_zip(self._installer_location, workdir)
install_script = os.path.join(workdir, 'aws', 'install')
run(
'%s --install-dir %s --bin-dir %s'
% (install_script, install_dir, bin_dir)
)
def cleanup(self):
shutil.rmtree(self._installation_dir)
def get_aws_cmd(self):
return os.path.join(self._bin_dir, EXE_NAME)
class PkgTester(InstallerTester):
DEFAULT_INSTALLER_LOCATION = os.path.join(
DIST_DIR, 'AWS-CLI-Installer.pkg'
)
_PKG_ID = 'com.amazon.aws.cli2'
def get_aws_cmd(self):
value = run('%s %s check' % (sys.executable, UNINSTALL_MAC_PKG_PATH))
match = re.search('installed at (.+)?\n', value)
base_path = match.group(1)
cmd_path = os.path.join(base_path, EXE_NAME)
return cmd_path
def setup(self):
run('sudo installer -pkg %s -target /' % self._installer_location)
def cleanup(self):
run('sudo %s %s uninstall' % (sys.executable, UNINSTALL_MAC_PKG_PATH))
def __call__(self):
assert (
os.geteuid() == 0
), 'Mac PKG installer must be run as root (with sudo).'
super(PkgTester, self).__call__()
def main():
installer_to_tester_cls = {
'exe': ExeTester,
'pkg': PkgTester,
}
parser = argparse.ArgumentParser(usage=__doc__)
parser.add_argument(
'--installer-type',
required=True,
choices=installer_to_tester_cls.keys(),
help='The type of installer to test',
)
parser.add_argument(
'--installer-path',
help=(
'The path to the installer to test. By default, installers are '
'used from the dist directory.'
),
)
args = parser.parse_args()
tester = installer_to_tester_cls[args.installer_type](args.installer_path)
return tester()
if __name__ == '__main__':
try:
sys.exit(main())
except AssertionError as e:
print(e)
sys.exit(1)
|
