🐛 Fix test_template and test_template_file

alan-turing-institute · Jan 25, 2024 · f7cc4f6 · f7cc4f6
1 parent 4e5ec60
commit f7cc4f6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests_/commands/test_config.py b/tests_/commands/test_config.py
@@ -6,7 +6,8 @@ def test_template(self, runner):
         result = runner.invoke(config_command_group, ["template"])
         assert result.exit_code == 0
         assert "subscription_id: Azure subscription ID" in result.stdout
-        assert "sres: {}" in result.stdout
+        assert "shm:" in result.stdout
+        assert "sres:" in result.stdout
 
     def test_template_file(self, runner, tmp_path):
         template_file = (tmp_path / "template.yaml").absolute()
@@ -17,7 +18,8 @@ def test_template_file(self, runner, tmp_path):
         with open(template_file) as f:
             template_text = f.read()
         assert "subscription_id: Azure subscription ID" in template_text
-        assert "sres: {}" in template_text
+        assert "shm:" in template_text
+        assert "sres:" in template_text
 
 
 class TestUpload: