testdrive: add possible params

MaterializeInc · Jan 10, 2024 · 8194915 · 8194915
1 parent d22c443
commit 8194915
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/misc/python/materialize/mzcompose/composition.py b/misc/python/materialize/mzcompose/composition.py
@@ -1021,6 +1021,8 @@ def testdrive(
         args: list[str] = [],
         caller: Traceback | None = None,
         mz_service: str | None = None,
+        default_timeout: str = "300s",
+        no_reset: bool = False,
     ) -> None:
         """Run a string as a testdrive script.
 
@@ -1036,6 +1038,12 @@ def testdrive(
         caller = caller or getframeinfo(stack()[1][0])
         args = args + [f"--source={caller.filename}:{caller.lineno}"]
 
+        if default_timeout is not None:
+            args += [f"--default-timeout={default_timeout}"]
+
+        if no_reset:
+            args += ["--no-reset"]
+
         if mz_service is not None:
             args += [
                 f"--materialize-url=postgres://materialize@{mz_service}:6875",