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
|
from argparse import Namespace
import os
from unittest import TestCase
from mock import (
MagicMock,
patch,
)
from assess_heterogeneous_control import (
assess_heterogeneous,
get_clients,
parse_args,
test_control_heterogeneous,
)
from jujupy import (
_temp_env,
)
from tests.test_deploy_stack import FakeBootstrapManager
from tests.test_jujupy import FakeJujuClient
__metaclass__ = type
class TestParseArgs(TestCase):
def test_parse_args(self):
args = parse_args(['a', 'b', 'c', 'd', 'e'])
self.assertEqual(args, Namespace(
initial='a', other='b', base_environment='c',
environment_name='d', log_dir='e', debug=False,
upload_tools=False, agent_url=None, agent_stream=None, series=None,
user=os.environ.get('JENKINS_USER'),
password=os.environ.get('JENKINS_PASSWORD')))
def test_parse_args_agent_url(self):
args = parse_args(['a', 'b', 'c', 'd', 'e', '--agent-url', 'foo',
'--user', 'my name', '--password', 'fake pass'])
self.assertEqual(args.agent_url, 'foo')
self.assertEqual(args.user, 'my name')
self.assertEqual(args.password, 'fake pass')
def test_parse_args_agent_stream(self):
args = parse_args(['a', 'b', 'c', 'd', 'e',
'--agent-stream', 'proposed',
'--user', 'my name', '--password', 'fake pass'])
self.assertEqual(args.agent_stream, 'proposed')
self.assertEqual(args.user, 'my name')
self.assertEqual(args.password, 'fake pass')
def test_parse_args_series(self):
args = parse_args(['a', 'b', 'c', 'd', 'e', '--series', 'trusty',
'--user', 'my name', '--password', 'fake pass'])
self.assertEqual(args.series, 'trusty')
self.assertEqual(args.user, 'my name')
self.assertEqual(args.password, 'fake pass')
class TestGetClients(TestCase):
def test_get_clients(self):
boo = {
('foo', '--version'): '1.18.73',
('bar', '--version'): '1.18.74',
('juju', '--version'): '1.18.75',
('which', 'juju'): '/usr/bun/juju'
}
with _temp_env({'environments': {'baz': {}}}):
with patch('subprocess.check_output', lambda x: boo[x]):
initial, other, released = get_clients('foo', 'bar', 'baz',
True, 'quxx')
self.assertEqual(initial.env, other.env)
self.assertEqual(initial.env, released.env)
self.assertNotIn('tools-metadata-url', initial.env.config)
self.assertEqual(initial.full_path, os.path.abspath('foo'))
self.assertEqual(other.full_path, os.path.abspath('bar'))
self.assertEqual(released.full_path, '/usr/bun/juju')
def test_get_clients_no_agent(self):
with _temp_env({'environments': {'baz': {}}}):
with patch('subprocess.check_output', return_value='1.18.73'):
initial, other, released = get_clients('foo', 'bar', 'baz',
True, None)
self.assertTrue('tools-metadata-url' not in initial.env.config)
class TestAssessHeterogeneous(TestCase):
@patch('assess_heterogeneous_control.BootstrapManager')
@patch('assess_heterogeneous_control.test_control_heterogeneous',
autospec=True)
@patch('assess_heterogeneous_control.get_clients', autospec=True)
def test_assess_heterogeneous(self, gc_mock, ch_mock, bm_mock):
initial = MagicMock()
gc_mock.return_value = (
initial, 'other_client', 'released_client')
assess_heterogeneous(
'initial', 'other', 'base_env', 'environment_name', 'log_dir',
False, False, 'agent_url', 'agent_stream', 'series')
gc_mock.assert_called_once_with(
'initial', 'other', 'base_env', False, 'agent_url')
is_jes_enabled = initial.is_jes_enabled.return_value
bm_mock.assert_called_once_with(
'environment_name', initial, 'released_client',
agent_stream='agent_stream', agent_url='agent_url',
bootstrap_host=None, jes_enabled=is_jes_enabled, keep_env=False,
log_dir='log_dir', machines=[], permanent=is_jes_enabled,
region=None, series='series')
ch_mock.assert_called_once_with(
bm_mock.return_value, 'other_client', False)
class TestTestControlHeterogeneous(TestCase):
def test_test_control_heterogeneous(self):
client = FakeJujuClient()
bs_manager = FakeBootstrapManager(client)
test_control_heterogeneous(bs_manager, client, True)
self.assertEqual(client._backing_state.exposed,
{'sink2', 'dummy-sink'})
self.assertEqual(client._backing_state.machines, {'0', '1', '2'})
self.assertEqual(client.juju_home, 'foo')
def test_same_home_and_env(self):
initial_client = FakeJujuClient()
other_client = FakeJujuClient()
other_client._backing_state = initial_client._backing_state
bs_manager = FakeBootstrapManager(initial_client)
bs_manager.permanent = True
test_control_heterogeneous(bs_manager, other_client, True)
self.assertEqual(initial_client.juju_home, other_client.juju_home)
self.assertEqual(
initial_client.env.environment, other_client.env.environment)
|