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
|
#!/usr/bin/env python
from argparse import ArgumentParser
from contextlib import contextmanager
import logging
import sys
from tempfile import NamedTemporaryFile
import os
import yaml
from deploy_stack import (
boot_context,
check_token,
configure_logging,
deploy_dummy_stack,
dump_env_logs,
get_random_string,
safe_print_status,
)
from jujupy import (
EnvJujuClient,
make_safe_config,
SimpleEnvironment,
)
from utility import (
add_basic_testing_arguments,
ensure_dir,
print_now,
)
def make_hosted_env_client(client, suffix):
env_name = '{}-{}'.format(client.env.environment, suffix)
hosted_environment = SimpleEnvironment(env_name, dict(client.env.config),
client.env.juju_home)
hosted_env_client = client.clone(hosted_environment)
return hosted_env_client
def test_jes_deploy(client, charm_prefix, log_dir, base_env):
"""Deploy the dummy stack in two hosted environments."""
# deploy into system env
deploy_dummy_stack(client, charm_prefix)
# deploy into hosted envs
with hosted_environment(client, log_dir, 'env1') as env1_client:
deploy_dummy_stack(env1_client, charm_prefix)
with hosted_environment(client, log_dir, 'env2') as env2_client:
deploy_dummy_stack(env2_client, charm_prefix)
# check all the services can talk
check_services(client)
check_services(env1_client)
check_services(env2_client)
@contextmanager
def jes_setup(args):
"""
Sets up the juju client and its environment.
Returns return client, charm_prefix and base_env.
"""
base_env = args.env
configure_logging(args.verbose)
# TODO(gz): Logic from deploy_stack, and precise is a bad default series?
series = args.series
if series is None:
series = 'precise'
charm_prefix = 'local:{}/'.format(series)
client = EnvJujuClient.by_version(
SimpleEnvironment.from_config(base_env), args.juju_bin, args.debug,
)
if not client.is_jes_enabled():
client.enable_jes()
with boot_context(
args.temp_env_name,
client,
args.bootstrap_host,
args.machine,
args.series,
args.agent_url,
args.agent_stream,
args.logs, args.keep_env,
upload_tools=False,
region=args.region,
):
yield client, charm_prefix, base_env
def env_token(env_name):
return env_name + get_random_string()
@contextmanager
def hosted_environment(system_client, log_dir, suffix):
client = make_hosted_env_client(system_client, suffix)
try:
with NamedTemporaryFile() as config_file:
config = make_safe_config(client)
yaml.dump(config, config_file)
config_file.flush()
client.create_environment(system_client, config_file.name)
yield client
except:
logging.exception(
'Exception while environment "{}" active'.format(
client.env.environment))
sys.exit(1)
finally:
safe_print_status(client)
hosted_log_dir = os.path.join(log_dir, suffix)
ensure_dir(hosted_log_dir)
dump_env_logs(client, None, hosted_log_dir)
client.destroy_model()
def check_services(client):
token = env_token(client.env.environment)
client.set_config('dummy-source', {'token': token})
print_now("checking services in " + client.env.environment)
check_token(client, token)
def main():
parser = ArgumentParser()
add_basic_testing_arguments(parser, using_jes=True)
args = parser.parse_args()
with jes_setup(args) as (client, charm_prefix, base_env):
test_jes_deploy(client, charm_prefix, args.logs, base_env)
if __name__ == '__main__':
main()
|