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
135
136
|
#!/usr/bin/env python
"""Assess the model-defaults command."""
from __future__ import print_function
import argparse
import logging
import sys
from deploy_stack import (
BootstrapManager,
)
from utility import (
add_basic_testing_arguments,
configure_logging,
JujuAssertionError,
)
__metaclass__ = type
log = logging.getLogger('assess_model_defaults')
def assemble_model_default(model_key, default, controller=None, regions=None):
"""Create a dict that contrains the formatted model-defaults data."""
# Ordering in the regions argument is lost.
defaults = {'default': default}
if controller is not None:
defaults['controller'] = controller
if regions is not None:
defaults['regions'] = [
{'name': region, 'value': region_default}
for (region, region_default) in regions.items()]
return {model_key: defaults}
def juju_assert_equal(lhs, rhs, msg):
if lhs != rhs:
raise JujuAssertionError(msg, lhs, rhs)
def get_new_model_config(client, region=None, model_name=None):
"""Create a new model, get it's config and then destroy it.
:param client: Client to use create the new model on.
:param region: If given and not None, create the new model in that region
otherwise create it in the client region.
:param model_name: Name of the new model.
"""
if model_name is None:
model_name = 'temp-model'
new_env = client.env.clone(model_name)
if region is not None:
new_env.set_region(region)
new_model = client.add_model(new_env)
config_data = new_model.get_model_config()
new_model.destroy_model()
return config_data
def assess_model_defaults_case(client, model_key, value, expected_default,
cloud=None, region=None):
"""Check the setting and unsetting of a region field."""
base_line = client.get_model_defaults(model_key)
client.set_model_defaults(model_key, value, cloud, region)
juju_assert_equal(expected_default, client.get_model_defaults(model_key),
'Mismatch after setting model-default.')
config = get_new_model_config(client, region)
juju_assert_equal(value, config[model_key]['value'],
'New model did not use the default.')
client.unset_model_defaults(model_key, cloud, region)
juju_assert_equal(base_line, client.get_model_defaults(model_key),
'Mismatch after resetting model-default.')
def assess_model_defaults_no_region(client, model_key, value):
"""Check the setting and unsetting of the controller field."""
default = client.get_model_defaults(model_key)[model_key]['default']
assess_model_defaults_case(
client, model_key, value,
assemble_model_default(model_key, default, str(value)))
def assess_model_defaults_region(client, model_key, value,
cloud=None, region=None):
"""Check the setting and unsetting of a region field."""
default = client.get_model_defaults(model_key)[model_key]['default']
assess_model_defaults_case(
client, model_key, value,
assemble_model_default(model_key, default, None, {region: str(value)}),
cloud, region)
def assess_model_defaults(client, other_region):
log.info('Checking controller model-defaults.')
assess_model_defaults_no_region(
client, 'automatically-retry-hooks', False)
region = client.env.get_region()
if region is not None:
log.info('Checking region model-defaults.')
assess_model_defaults_region(
client, 'default-series', 'trusty', region=region)
if other_region is not None:
log.info('Checking other region model-defaults.')
assess_model_defaults_region(
client, 'default-series', 'trusty', region=other_region)
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description='Assess the model-defaults command.')
add_basic_testing_arguments(parser)
parser.add_argument('--other-region',
help='Set the model default for a different region.')
return parser.parse_args(argv)
def main(argv=None):
args = parse_args(argv)
configure_logging(args.verbose)
bs_manager = BootstrapManager.from_args(args)
if (args.other_region is not None and
args.other_region == bs_manager.client.env.get_region()):
raise ValueError('Other region is a repeat of region.')
with bs_manager.booted_context(args.upload_tools):
assess_model_defaults(bs_manager.client, args.other_region)
return 0
if __name__ == '__main__':
sys.exit(main())
|