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
|
#!/usr/bin/env python
"""This module tests the deployment with constraints."""
from __future__ import print_function
import os
import argparse
import logging
import sys
from deploy_stack import (
BootstrapManager,
)
from jujucharm import (
Charm,
local_charm_path,
)
from utility import (
add_basic_testing_arguments,
configure_logging,
JujuAssertionError,
temp_dir,
)
__metaclass__ = type
log = logging.getLogger("assess_constraints")
VIRT_TYPES = ['lxd']
def deploy_constraint(client, charm, series, charm_repo,
memory=None, cpu_cores=None, virt_type=None):
"""Test deploying charm with constraints."""
args = ()
if memory:
args += ("mem={}".format(memory),)
if cpu_cores:
args += ("cpu-cores={}".format(cpu_cores),)
if virt_type:
args += ("virt-type={}".format(virt_type),)
constraints = " ".join(args)
client.deploy(charm, series=series, repository=charm_repo,
constraints=constraints)
client.wait_for_workloads()
def assess_virt_type(client, virt_type):
"""Assess the virt-type option for constraints"""
if virt_type not in VIRT_TYPES:
raise JujuAssertionError(virt_type)
charm_name = 'virt-type-' + virt_type
charm_series = 'xenial'
with temp_dir() as charm_dir:
constraints_charm = Charm(charm_name,
'Test charm for constraints',
series=['xenial'])
charm_root = constraints_charm.to_repo_dir(charm_dir)
platform = 'ubuntu'
charm = local_charm_path(charm=charm_name,
juju_ver=client.version,
series=charm_series,
repository=os.path.dirname(charm_root),
platform=platform)
deploy_constraint(client, charm, charm_series,
charm_dir, virt_type=virt_type)
def assess_constraints(client, test_kvm=False):
"""Assess deployment with constraints"""
if test_kvm:
VIRT_TYPES.append("kvm")
for virt_type in VIRT_TYPES:
assess_virt_type(client, virt_type)
try:
assess_virt_type(client, 'aws')
except JujuAssertionError:
log.info("Correctly rejected virt-type aws")
else:
raise JujuAssertionError("FAIL: Client deployed with virt-type aws")
if test_kvm:
VIRT_TYPES.remove("kvm")
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(description="Test constraints")
add_basic_testing_arguments(parser)
return parser.parse_args(argv)
def main(argv=None):
args = parse_args(argv)
configure_logging(args.verbose)
bs_manager = BootstrapManager.from_args(args)
test_kvm = '--with-virttype-kvm' in args
with bs_manager.booted_context(args.upload_tools):
assess_constraints(bs_manager.client, test_kvm)
return 0
if __name__ == '__main__':
sys.exit(main())
|