logo

bake/tests/conftest.py

@@ -1,12 +1,11 @@
 import os
 
 os.environ["PYTHONUNBUFFERED"] = "1"
+os.environ.pop("BAKEFILE_PATH", "")
 
 import pytest
 import delegator
 
-import bake.cli as bake_cli
-
 
 @pytest.fixture
 def bake():
@@ -30,3 +29,27 @@ def bake():
         return c
 
     return run
+
+
+@pytest.fixture
+def bake_i():
+    def run(*args, fixture="default", assert_ok=True, block=True):
+        bakefile = os.path.join(
+            os.path.dirname(__file__), "fixtures", f"{fixture}.Bakefile"
+        )
+        cmd = " ".join(["bake", "-i", "-b", bakefile] + list(args))
+        print(f"$ {cmd}")
+
+        c = delegator.run(cmd, block=block)
+
+        if block and assert_ok:
+            assert c.return_code == 0
+
+        if block:
+
+            print(c.out)
+            print(c.err)
+
+        return c
+
+    return run