summaryrefslogtreecommitdiff
path: root/tests/cloud_tests/platforms/__init__.py
diff options
context:
space:
mode:
authorChad Smith <chad.smith@canonical.com>2018-02-02 11:11:36 -0700
committerChad Smith <chad.smith@canonical.com>2018-02-02 11:11:36 -0700
commit78013bc65030421699b5feb66bc8b7a205abfbc0 (patch)
tree2ebf7111129f4aaf8a833ba6d226d4513ed59388 /tests/cloud_tests/platforms/__init__.py
parent192261fe38a32edbd1f605ba25bbb6f4822a0720 (diff)
parentf7deaf15acf382d62554e2b1d70daa9a9109d542 (diff)
downloadvyos-cloud-init-78013bc65030421699b5feb66bc8b7a205abfbc0.tar.gz
vyos-cloud-init-78013bc65030421699b5feb66bc8b7a205abfbc0.zip
merge from master at 17.2-30-gf7deaf15
Diffstat (limited to 'tests/cloud_tests/platforms/__init__.py')
-rw-r--r--tests/cloud_tests/platforms/__init__.py22
1 files changed, 20 insertions, 2 deletions
diff --git a/tests/cloud_tests/platforms/__init__.py b/tests/cloud_tests/platforms/__init__.py
index 3490fe87..a01e51ac 100644
--- a/tests/cloud_tests/platforms/__init__.py
+++ b/tests/cloud_tests/platforms/__init__.py
@@ -2,15 +2,27 @@
"""Main init."""
-from tests.cloud_tests.platforms import lxd
-from tests.cloud_tests.platforms import nocloudkvm
+from .ec2 import platform as ec2
+from .lxd import platform as lxd
+from .nocloudkvm import platform as nocloudkvm
PLATFORMS = {
+ 'ec2': ec2.EC2Platform,
'nocloud-kvm': nocloudkvm.NoCloudKVMPlatform,
'lxd': lxd.LXDPlatform,
}
+def get_image(platform, config):
+ """Get image from platform object using os_name."""
+ return platform.get_image(config)
+
+
+def get_instance(snapshot, *args, **kwargs):
+ """Get instance from snapshot."""
+ return snapshot.launch(*args, **kwargs)
+
+
def get_platform(platform_name, config):
"""Get the platform object for 'platform_name' and init."""
platform_cls = PLATFORMS.get(platform_name)
@@ -18,4 +30,10 @@ def get_platform(platform_name, config):
raise ValueError('invalid platform name: {}'.format(platform_name))
return platform_cls(config)
+
+def get_snapshot(image):
+ """Get snapshot from image."""
+ return image.snapshot()
+
+
# vi: ts=4 expandtab