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
|
#!/usr/bin/env python
from __future__ import print_function
from argparse import ArgumentParser
import os.path
from jujupy import (
EnvJujuClient,
SimpleEnvironment,
until_timeout,
)
from utility import scoped_environ
def assess_bootstrap(juju, env, debug):
with scoped_environ():
juju_bin = os.path.dirname(os.path.abspath(juju))
os.environ['PATH'] = '{}:{}'.format(juju_bin, os.environ['PATH'])
client = EnvJujuClient.by_version(SimpleEnvironment.from_config(env),
juju, debug)
client.destroy_environment()
try:
client.bootstrap()
for ignored in until_timeout(30):
client.get_status(1)
print('Environment successfully bootstrapped.')
finally:
client.destroy_environment()
def parse_args(argv=None):
parser = ArgumentParser()
parser.add_argument('juju', help="The Juju client to use.")
parser.add_argument('env', help="The environment to test with.")
parser.add_argument('--debug', action="store_true", default=False,
help='Use --debug juju logging.')
return parser.parse_args(argv)
def main():
args = parse_args()
assess_bootstrap(**args.__dict__)
if __name__ == '__main__':
main()
|