Override runtime dir for testing bubblejail

igo95862 · May 25, 2024 · 2fec729 · 2fec729
1 parent fd731b2
commit 2fec729
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tools/run_test_bubblejail.py b/tools/run_test_bubblejail.py
@@ -79,6 +79,20 @@ async def run_with_helper_script(
         run_with_helper_script,
     )
 
+    import bubblejail.bubblejail_instance
+
+    custom_run_dir = BUILD_DIR / "bubblejail_test_rundir"
+    custom_run_dir.mkdir(exist_ok=True)
+
+    def custom_runtime_dir() -> str:
+        return str(custom_run_dir)
+
+    setattr(
+        bubblejail.bubblejail_instance,
+        "get_runtime_dir",
+        custom_runtime_dir,
+    )
+
 
 def shell_main() -> None:
     history_file = BUILD_DIR / 'bubblejail_cmd_history'