7
from deploy_stack import BootstrapManager
8
from fakejuju import fake_juju_client
13
from quickstart_deploy import QuickstartTest
18
from tests.test_deploy_stack import FakeBootstrapManager
19
from utility import temp_dir
22
def make_bootstrap_manager(client, log_dir='log_dir'):
23
return BootstrapManager(
24
'env', client, client, None, [], 'series', 'agent_url',
25
'agent_stream', 'region', log_dir, 'keep_env', 'permanent',
29
class TestQuickstartTest(FakeHomeTestCase):
31
def test_run_finally(self):
32
do_finally = MagicMock()
34
def fake_iter_steps():
36
yield {'bootstrap_host': 'foo'}
40
client = EnvJujuClient(
41
JujuData('foo', {'type': 'local'}), '1.234-76', None)
42
bs_manager = make_bootstrap_manager(client)
43
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
44
with patch.object(quickstart, 'iter_steps',
45
side_effect=fake_iter_steps):
47
do_finally.assert_called_once_with()
49
def test_run_exception(self):
50
tear_down = MagicMock()
52
def fake_iter_steps():
54
yield {'bootstrap_host': 'foo'}
58
client = EnvJujuClient(
59
JujuData('foo', {'type': 'local'}), '1.234-76', None)
60
bs_manager = make_bootstrap_manager(client)
61
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
62
with patch.object(quickstart, 'iter_steps',
63
side_effect=fake_iter_steps):
64
with self.assertRaises(BaseException):
65
with patch('logging.info', side_effect=Exception):
67
tear_down.assert_called_once_with()
69
def test_iter_steps(self):
70
log_dir = use_context(self, temp_dir())
71
client = EnvJujuClient(
72
JujuData('foo', {'type': 'local'}), '1.234-76', None)
73
bs_manager = make_bootstrap_manager(client, log_dir=log_dir)
74
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
75
steps = quickstart.iter_steps()
76
with patch.object(client, 'quickstart') as qs_mock:
78
with patch('jujupy.check_free_disk_space', autospec=True):
79
with patch('deploy_stack.tear_down', autospec=True) as td_mock:
81
td_mock.assert_called_once_with(client, 'jes_enabled', try_jes=True)
82
qs_mock.assert_called_once_with('/tmp/bundle.yaml')
83
expected = {'juju-quickstart': 'Returned from quickstart'}
84
self.assertEqual(expected, step)
85
with patch('deploy_stack.get_machine_dns_name',
86
return_value='mocked_name') as dns_mock:
88
with patch.object(client, 'get_controller_client') as gcc_mock:
90
dns_mock.assert_called_once_with(gcc_mock.return_value, '0')
91
self.assertEqual('mocked_name', step['bootstrap_host'])
92
with patch.object(client, 'wait_for_deploy_started') as wds_mock:
95
wds_mock.assert_called_once_with(2)
96
self.assertEqual('Deploy stated', step['deploy_started'])
97
with patch.object(client, 'wait_for_started') as ws_mock:
100
ws_mock.assert_called_once_with(ANY)
101
self.assertEqual('All Agents started', step['agents_started'])
102
with patch('deploy_stack.safe_print_status'):
103
with patch('deploy_stack.tear_down'):
104
with patch('quickstart_deploy.BootstrapManager.dump_all_logs'):
105
with patch('jujupy.EnvJujuClient.iter_model_clients',
109
def test_iter_steps_context(self):
110
client = fake_juju_client()
111
bs_manager = FakeBootstrapManager(client)
112
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
113
step_iter = quickstart.iter_steps()
114
self.assertIs(False, bs_manager.entered_top)
115
self.assertIs(False, bs_manager.exited_top)
116
self.assertIs(False, bs_manager.entered_bootstrap)
117
self.assertIs(False, bs_manager.exited_bootstrap)
119
models = client._backend.controller_state.models
120
backing_state = models[client.model_name]
121
self.assertEqual('/tmp/bundle.yaml', backing_state.current_bundle)
122
self.assertIs(True, bs_manager.entered_top)
123
self.assertIs(True, bs_manager.entered_bootstrap)
124
self.assertIs(False, bs_manager.exited_bootstrap)
125
self.assertIs(False, bs_manager.entered_runtime)
126
self.assertIs(False, bs_manager.exited_runtime)
128
self.assertIs(True, bs_manager.exited_bootstrap)
129
self.assertIs(True, bs_manager.entered_runtime)
130
self.assertIs(False, bs_manager.exited_runtime)
131
with patch.object(client, 'wait_for_deploy_started') as wfds_mock:
133
wfds_mock.assert_called_once_with(2)
134
self.assertIs(False, bs_manager.exited_runtime)
135
with patch.object(client, 'wait_for_started') as wfs_mock:
137
wfs_mock.assert_called_once_with(3600)
138
self.assertIs(False, bs_manager.exited_runtime)
139
with self.assertRaises(StopIteration):
141
self.assertIs(True, bs_manager.exited_runtime)
142
self.assertIs(True, bs_manager.exited_top)
144
def test_iter_steps_quickstart_fail(self):
145
client = EnvJujuClient(
146
JujuData('foo', {'type': 'local'}), '1.234-76', None)
147
bs_manager = FakeBootstrapManager(client)
148
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
149
step_iter = quickstart.iter_steps()
150
with patch.object(client, 'quickstart', side_effect=Exception):
151
with self.assertRaises(Exception):
153
self.assertIs(False, bs_manager.entered_runtime)
154
self.assertIs(True, bs_manager.exited_bootstrap)
155
self.assertIs(True, bs_manager.exited_top)
157
def test_iter_steps_wait_fail(self):
158
client = fake_juju_client()
159
bs_manager = FakeBootstrapManager(client)
160
quickstart = QuickstartTest(bs_manager, '/tmp/bundle.yaml', 2)
161
step_iter = quickstart.iter_steps()
164
with patch.object(client, 'wait_for_deploy_started',
165
side_effect=Exception):
166
with self.assertRaises(Exception):
168
self.assertIs(True, bs_manager.exited_runtime)
169
self.assertIs(True, bs_manager.exited_bootstrap)
170
self.assertIs(True, bs_manager.exited_top)