3
from argparse import ArgumentParser
4
from contextlib import contextmanager
7
from tempfile import NamedTemporaryFile
12
from deploy_stack import (
27
add_basic_testing_arguments,
33
def make_hosted_env_client(client, suffix):
34
env_name = '{}-{}'.format(client.env.environment, suffix)
35
hosted_environment = SimpleEnvironment(env_name, dict(client.env.config),
37
hosted_env_client = client.clone(hosted_environment)
38
return hosted_env_client
41
def test_jes_deploy(client, charm_series, log_dir, base_env):
42
"""Deploy the dummy stack in two hosted environments."""
43
# deploy into system env
44
deploy_dummy_stack(client, charm_series)
46
# deploy into hosted envs
47
with hosted_environment(client, log_dir, 'env1') as env1_client:
48
deploy_dummy_stack(env1_client, charm_series)
49
with hosted_environment(client, log_dir, 'env2') as env2_client:
50
deploy_dummy_stack(env2_client, charm_series)
51
# check all the services can talk
52
check_services(client)
53
check_services(env1_client)
54
check_services(env2_client)
60
Sets up the juju client and its environment.
62
Returns return client, charm_prefix and base_env.
65
configure_logging(args.verbose)
66
# TODO(gz): Logic from deploy_stack, and precise is a bad default series?
71
client = EnvJujuClient.by_version(
72
SimpleEnvironment.from_config(base_env), args.juju_bin, args.debug,
74
if not client.is_jes_enabled():
84
args.logs, args.keep_env,
88
yield client, charm_series, base_env
91
def env_token(env_name):
92
return env_name + get_random_string()
96
def hosted_environment(system_client, log_dir, suffix):
97
client = make_hosted_env_client(system_client, suffix)
99
with NamedTemporaryFile() as config_file:
100
config = make_safe_config(client)
101
yaml.dump(config, config_file)
103
client.create_environment(system_client, config_file.name)
107
'Exception while environment "{}" active'.format(
108
client.env.environment))
111
safe_print_status(client)
112
hosted_log_dir = os.path.join(log_dir, suffix)
113
ensure_dir(hosted_log_dir)
114
dump_env_logs(client, None, hosted_log_dir)
115
client.destroy_model()
118
def check_services(client):
119
token = env_token(client.env.environment)
120
client.set_config('dummy-source', {'token': token})
121
print_now("checking services in " + client.env.environment)
122
check_token(client, token)
126
parser = ArgumentParser()
127
add_basic_testing_arguments(parser, using_jes=True)
128
args = parser.parse_args()
129
with jes_setup(args) as (client, charm_series, base_env):
130
test_jes_deploy(client, charm_series, args.logs, base_env)
133
if __name__ == '__main__':