CLN: split #1928, refactor render test (#2246)

nebari-dev · Apr 5, 2024 · 0f95420 · 0f95420
2 parents 994b57b + 7d59633
commit 0f95420
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/tests/tests_unit/test_render.py b/tests/tests_unit/test_render.py
@@ -1,7 +1,6 @@
 import os
 
 from _nebari.stages.bootstrap import CiEnum
-from nebari import schema
 from nebari.plugins import nebari_plugin_manager
 
 
@@ -22,18 +21,12 @@ def test_render_config(nebari_render):
         "03-kubernetes-initialize",
     }.issubset(os.listdir(output_directory / "stages"))
 
-    if config.provider == schema.ProviderEnum.do:
-        assert (output_directory / "stages" / "01-terraform-state/do").is_dir()
-        assert (output_directory / "stages" / "02-infrastructure/do").is_dir()
-    elif config.provider == schema.ProviderEnum.aws:
-        assert (output_directory / "stages" / "01-terraform-state/aws").is_dir()
-        assert (output_directory / "stages" / "02-infrastructure/aws").is_dir()
-    elif config.provider == schema.ProviderEnum.gcp:
-        assert (output_directory / "stages" / "01-terraform-state/gcp").is_dir()
-        assert (output_directory / "stages" / "02-infrastructure/gcp").is_dir()
-    elif config.provider == schema.ProviderEnum.azure:
-        assert (output_directory / "stages" / "01-terraform-state/azure").is_dir()
-        assert (output_directory / "stages" / "02-infrastructure/azure").is_dir()
+    assert (
+        output_directory / "stages" / f"01-terraform-state/{config.provider.value}"
+    ).is_dir()
+    assert (
+        output_directory / "stages" / f"02-infrastructure/{config.provider.value}"
+    ).is_dir()
 
     if config.ci_cd.type == CiEnum.github_actions:
         assert (output_directory / ".github/workflows/").is_dir()