Fix tests

charmoniumQ · Jan 18, 2025 · 972488b · 972488b
1 parent c65d870
commit 972488b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_record.py b/tests/test_record.py
@@ -45,7 +45,8 @@ def bash(*cmds: str) -> list[str]:
 modes = [
     ["probe", "record"],
     ["probe", "record", "--debug"],
-    ["probe", "record", "--copy-files"],
+    ["probe", "record", "--copy-files-lazily"],
+    ["probe", "record", "--copy-files-eagerly"],
 ]
 
 
@@ -78,7 +79,7 @@ def test_cmds(mode: list[str], command: list[str]) -> None:
     print(shlex.join(cmd))
     subprocess.run(cmd, check=True, cwd=tmpdir)
 
-    if "--copy-files" in mode:
+    if any("copy-files" in arg for arg in mode):
 
         cmd = ["probe", "export", "oci-image", "probe-command-test:latest"]
         print(shlex.join(cmd))