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
|
# This file is part of cloud-init. See LICENSE file for license information.
"""Base LXD platform."""
from pylxd import (Client, exceptions)
from tests.cloud_tests.images import lxd as lxd_image
from tests.cloud_tests.instances import lxd as lxd_instance
from tests.cloud_tests.platforms import base
from tests.cloud_tests import util
DEFAULT_SSTREAMS_SERVER = "https://images.linuxcontainers.org:8443"
class LXDPlatform(base.Platform):
"""LXD test platform."""
platform_name = 'lxd'
def __init__(self, config):
"""Set up platform."""
super(LXDPlatform, self).__init__(config)
# TODO: allow configuration of remote lxd host via env variables
# set up lxd connection
self.client = Client()
def get_image(self, img_conf):
"""Get image using specified image configuration.
@param img_conf: configuration for image
@return_value: cloud_tests.images instance
"""
pylxd_image = self.client.images.create_from_simplestreams(
img_conf.get('sstreams_server', DEFAULT_SSTREAMS_SERVER),
img_conf['alias'])
image = lxd_image.LXDImage(self, img_conf, pylxd_image)
if img_conf.get('override_templates', False):
image.update_templates(self.config.get('template_overrides', {}),
self.config.get('template_files', {}))
return image
def launch_container(self, properties, config, features,
image=None, container=None, ephemeral=False,
container_config=None, block=True, image_desc=None,
use_desc=None):
"""Launch a container.
@param properties: image properties
@param config: image configuration
@param features: image features
@param image: image fingerprint to launch from
@param container: container to copy
@param ephemeral: delete image after first shutdown
@param container_config: config options for instance as dict
@param block: wait until container created
@param image_desc: description of image being launched
@param use_desc: description of container's use
@return_value: cloud_tests.instances instance
"""
if not (image or container):
raise ValueError("either image or container must be specified")
container = self.client.containers.create({
'name': util.gen_instance_name(image_desc=image_desc,
use_desc=use_desc,
used_list=self.list_containers()),
'ephemeral': bool(ephemeral),
'config': (container_config
if isinstance(container_config, dict) else {}),
'source': ({'type': 'image', 'fingerprint': image} if image else
{'type': 'copy', 'source': container})
}, wait=block)
return lxd_instance.LXDInstance(self, container.name, properties,
config, features, container)
def container_exists(self, container_name):
"""Check if container with name 'container_name' exists.
@return_value: True if exists else False
"""
res = True
try:
self.client.containers.get(container_name)
except exceptions.LXDAPIException as e:
res = False
if e.response.status_code != 404:
raise
return res
def list_containers(self):
"""List names of all containers.
@return_value: list of names
"""
return [container.name for container in self.client.containers.all()]
def query_image_by_alias(self, alias):
"""Get image by alias in local image store.
@param alias: alias of image
@return_value: pylxd image (not cloud_tests.images instance)
"""
return self.client.images.get_by_alias(alias)
def destroy(self):
"""Clean up platform data."""
super(LXDPlatform, self).destroy()
# vi: ts=4 expandtab
|
