test: verify that "vmadm config" works

Sponsored-by: author

2 files changed, 27 insertions, 0 deletions

diff --git a/subplot/vmadm.py b/subplot/vmadm.py
index dcdcbb6..42f857a 100644
--- a/subplot/vmadm.py
+++ b/subplot/vmadm.py
@@ -1,3 +1,5 @@
+import io
+import json
 import logging
 import os
 import shutil
@@ -105,3 +107,25 @@ def run_hostname_over_ssh(ctx, config=None, target=None, args=None):
     runcmd_run(ctx, ["chmod", "-R", "u=rwX,go=", ".ssh"])
     runcmd_run(ctx, ["ssh", "-F", config, target] + args.split())
     runcmd_exit_code_is_zero(ctx)
+
+
+def stdout_json_matches(ctx, filename=None):
+    runcmd_get_stdout = globals()["runcmd_get_stdout"]
+    assert_dict_eq = globals()["assert_dict_eq"]
+
+    stdout = io.StringIO(runcmd_get_stdout(ctx))
+    actual = yaml.safe_load(stdout)
+
+    with open(filename) as f:
+        expected = json.load(f)
+
+    for key in expected:
+        value = expected[key]
+        if isinstance(value, str):
+            expected[key] = os.path.expanduser(value)
+        elif isinstance(value, list) and value and isinstance(value[0], str):
+            expected[key] = [os.path.expanduser(item) for item in value]
+
+    logging.debug(f"actual: {actual}")
+    logging.debug(f"expect: {expected}")
+    assert_dict_eq(actual, expected)
diff --git a/subplot/vmadm.yaml b/subplot/vmadm.yaml
index fb778dd..c6c5ad1 100644
--- a/subplot/vmadm.yaml
+++ b/subplot/vmadm.yaml
@@ -25,3 +25,6 @@
 
 - then: "directories {actual} and {expected} are identical"
   function: directories_match
+
+- then: "stdout, as JSON, matches file {filename} with tilde expansion"
+  function: stdout_json_matches