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
137
138
139
140
141
142
143
144
145
146
147
148
149
|
#!/usr/bin/env python
"""
Charms have the capability to declare that they support more than
one series. Previously a separate copy of the charm was required for
each series. An important constraint here is that for a given charm,
all of the listed series must be for the same distro/OS; it is not
allowed to offer a single charm for Ubuntu and CentOS for example.
Supported series are added to charm metadata as follows:
name: mycharm
summary: "Great software"
description: It works
series:
- trusty
- precise
- wily
The default series is the first in the list:
juju deploy mycharm
should deploy a mycharm service running on trusty.
A different, non-default series may be specified:
juju deploy mycharm --series precise
It is possible to force the charm to deploy using an unsupported series
(so long as the underlying OS is compatible):
juju deploy mycharm --series xenial --force
"""
from __future__ import print_function
import argparse
from collections import namedtuple
import logging
import os
import subprocess
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,
)
from assess_heterogeneous_control import check_series
__metaclass__ = type
log = logging.getLogger("assess_multi_series_charms")
Test = namedtuple("Test", ["series", "service", "force", "success", "machine",
"juju1x_supported"])
def assess_multi_series_charms(client):
"""Assess multi series charms.
:param client: Juju client.
:type client: jujupy.EnvJujuClient
:return: None
"""
tests = [
Test(series="precise", service='test0', force=False, success=False,
machine=None, juju1x_supported=False),
Test(series=None, service='test1', force=False, success=True,
machine='0', juju1x_supported=True),
Test(series="trusty", service='test2', force=False, success=True,
machine='1', juju1x_supported=True),
Test(series="xenial", service='test3', force=False, success=True,
machine='2', juju1x_supported=False),
Test(series="precise", service='test4', force=True, success=True,
machine='3', juju1x_supported=False),
]
with temp_dir() as repository:
charm_name = 'dummy'
charm = Charm(charm_name, 'Test charm', series=['trusty', 'xenial'])
charm_dir = charm.to_repo_dir(repository)
charm_path = local_charm_path(
charm=charm_name, juju_ver=client.version, series='trusty',
repository=os.path.dirname(charm_dir))
for test in tests:
if client.is_juju1x() and not test.juju1x_supported:
continue
log.info(
"Assessing multi series charms: test: {} charm_dir:{}".format(
test, charm_path))
assert_deploy(client, test, charm_path, repository=repository)
if test.machine:
check_series(client, machine=test.machine, series=test.series)
def assert_deploy(client, test, charm_path, repository=None):
"""Deploy a charm and assert a success or fail.
:param client: Juju client
:type client: jujupy.EnvJujuClient
:param test: Deploy test data.
:type test: Test
:param charm_dir:
:type charm_dir: str
:param repository: Direcotry path to the repository
:type repository: str
:return: None
"""
if test.success:
client.deploy(charm=charm_path, series=test.series,
service=test.service, force=test.force,
repository=repository)
client.wait_for_started()
else:
try:
client.deploy(charm=charm_path, series=test.series,
service=test.service, force=test.force,
repository=repository)
except subprocess.CalledProcessError:
return
raise JujuAssertionError('Assert deploy failed for {}'.format(test))
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description="Test multi series charm feature")
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)
with bs_manager.booted_context(args.upload_tools):
assess_multi_series_charms(bs_manager.client)
return 0
if __name__ == '__main__':
sys.exit(main())
|