test: make path to Debian cloud image be configurable

2 files changed, 17 insertions, 0 deletions

diff --git a/subplot/vmadm.py b/subplot/vmadm.py
index 48737a3..2daa81c 100644
--- a/subplot/vmadm.py
+++ b/subplot/vmadm.py
@@ -1,5 +1,6 @@
 import logging
 import os
+import shutil
 import yaml
 
 
@@ -21,6 +22,19 @@ def install_vmadm(ctx):
     os.mkdir("expected")
 
 
+def ensure_base_image(ctx):
+    assert_eq = globals()["assert_eq"]
+    assert_ne = globals()["assert_ne"]
+
+    base = os.environ.get("BASE_IMAGE", "")
+    _, ext = os.path.splitext(base)
+
+    assert_ne(base, "")
+    assert_eq(ext, ".qcow2")
+
+    shutil.copy(base, "base.qcow2")
+
+
 def invoke_cloud_init(ctx, filename=None, dirname=None):
     runcmd_run = globals()["runcmd_run"]
     runcmd_exit_code_is_zero = globals()["runcmd_exit_code_is_zero"]
diff --git a/subplot/vmadm.yaml b/subplot/vmadm.yaml
index 374391b..a8810ae 100644
--- a/subplot/vmadm.yaml
+++ b/subplot/vmadm.yaml
@@ -1,6 +1,9 @@
 - given: "an installed vmadm"
   function: install_vmadm
 
+- given: "a Debian 10 OpenStack cloud image"
+  function: ensure_base_image
+
 - when: "I invoke vmadm cloud-init {filename} {dirname}"
   function: invoke_cloud_init