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
|
from argparse import Namespace
import os
from tempfile import NamedTemporaryFile
from mock import patch, call
from run_download_juju import (
create_workspace_yaml,
get_revisions,
parse_args,
main
)
from jujupy import ensure_dir
from tests import (
TestCase,
)
from test_schedule_hetero_control import make_build_var_file
from tests import parse_error
from utility import temp_dir
class TestGetCandidateInfo(TestCase):
def test_get_candidate_info(self):
with temp_dir() as dir_path:
candidate_dir = os.path.join(dir_path, 'candidate')
ensure_dir(candidate_dir)
for ver, rev in [['1.24.3', '2870'], ['1.24.5', '2999']]:
ver_dir = os.path.join(candidate_dir, ver)
ensure_dir(ver_dir)
make_build_var_file(ver_dir, version=ver, revision_build=rev)
rev = get_revisions(dir_path)
self.assertEqual(sorted(rev), ['2870', '2999'])
def test_create_workspace_yaml(self):
with NamedTemporaryFile() as temp_file:
create_workspace_yaml("/my/home", "/script/path", temp_file)
with open(temp_file.name) as yaml_file:
content = yaml_file.read()
self.assertEqual(content, self.expected())
def test_create_workspace_yaml_with_rev(self):
with NamedTemporaryFile() as temp_file:
create_workspace_yaml(
"/my/home", "/script/path", temp_file, ['2870', '2999'])
with open(temp_file.name) as yaml_file:
content = yaml_file.read()
self.assertEqual(content, self.expected('-c 2870 2999'))
def test_parse_args_default(self):
os.environ['JUJU_HOME'] = '/juju/home'
args = parse_args([])
expected_args = Namespace(
osx_host='jenkins@osx-slave.vapour.ws',
win_host='Administrator@win-slave.vapour.ws',
juju_home='/juju/home')
self.assertEqual(args, expected_args)
def test_parse_args(self):
args = parse_args(['-o', 'j@my-osx-host', '-w', 'j@my-win-host',
'-j' '/juju/home'])
expected_args = Namespace(
osx_host='j@my-osx-host', win_host='j@my-win-host',
juju_home='/juju/home')
self.assertEqual(args, expected_args)
def test_parse_args_juju_home_unset(self):
with parse_error(self) as stderr:
parse_args([])
self.assertRegexpMatches(stderr.getvalue(), 'Invalid JUJU_HOME value')
def test_main(self):
with temp_dir() as home:
os.environ['HOME'] = home
juju_home = os.path.join(home, 'juju')
os.mkdir(juju_home)
os.environ['JUJU_HOME'] = juju_home
temp_file = NamedTemporaryFile(delete=False)
with patch('run_download_juju.NamedTemporaryFile', autospec=True,
return_value=temp_file) as ntf_mock:
with patch('subprocess.check_output') as sco_mock:
main([])
calls = [
call(['workspace-run', temp_file.name,
'Administrator@win-slave.vapour.ws']),
call(['workspace-run', temp_file.name,
'jenkins@osx-slave.vapour.ws'])]
self.assertEqual(sco_mock.call_args_list, calls)
ntf_mock.assert_called_once_with()
def test_main_expected_arguments(self):
os.environ['HOME'] = '/home/fake-juju-user'
with temp_dir() as juju_home:
os.environ['JUJU_HOME'] = juju_home
temp_file = NamedTemporaryFile(delete=False)
with patch('run_download_juju.get_revisions',
autospec=True) as gr_mock:
with patch('run_download_juju.create_workspace_yaml',
autospec=True) as cwy_mock:
with patch('run_download_juju.NamedTemporaryFile',
autospec=True,
return_value=temp_file) as ntf_mock:
with patch('subprocess.check_output') as sco_mock:
main([])
sco_calls = [
call(['workspace-run', temp_file.name,
'Administrator@win-slave.vapour.ws']),
call(['workspace-run', temp_file.name,
'jenkins@osx-slave.vapour.ws'])]
cwy_calls = [
call(juju_home,
('C:\\\Users\\\Administrator\\\juju-ci-tools\\\download_'
'juju.py'),
temp_file, gr_mock.return_value),
call(juju_home, '$HOME/juju-ci-tools/download_juju.py',
temp_file, gr_mock.return_value)]
self.assertEqual(sco_mock.call_args_list, sco_calls)
ntf_mock.assert_called_once_with()
gr_mock.assert_called_once_with(os.environ['HOME'])
self.assertEqual(cwy_mock.call_args_list, cwy_calls)
def expected(self, rev="''"):
return ("command: [python, /script/path, cloud-city, -r, -v, {}]\n"
"install:\n"
" cloud-city: [/my/home/ec2rc]\n".format(rev))
|