From 22caaa22cda72576b1ce625619c3cb653de211ff Mon Sep 17 00:00:00 2001
From: Hood Chatham <roberthoodchatham@gmail.com>
Date: Tue, 21 Apr 2026 14:38:33 -0700
Subject: [PATCH] chore: Run tests with memory snapshots

Should help prevent bugs like
https://github.com/cloudflare/workers-sdk/issues/12961
---
 packages/cli/tests/test_in_workerd.py | 31 +++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/packages/cli/tests/test_in_workerd.py b/packages/cli/tests/test_in_workerd.py
index 98bf92d..13fd81e 100644
--- a/packages/cli/tests/test_in_workerd.py
+++ b/packages/cli/tests/test_in_workerd.py
@@ -43,8 +43,13 @@ def embed(dir: Path, root: Path, level: int = 0):
     return modules
 
 
+@pytest.fixture(scope="module")
+def bundle_cache(tmp_path_factory):
+    yield tmp_path_factory.mktemp("bundle_cache")
+
+
 @pytest.mark.parametrize("test_dir, wd_test_file", discover_workerd_tests())
-def test_in_workerd(tmp_path, test_dir, wd_test_file, pytestconfig):
+def test_in_workerd(tmp_path, test_dir, wd_test_file, pytestconfig, bundle_cache):
     color = pytestconfig.get_terminal_writer().hasmarkup
     target = tmp_path / test_dir.name
     shutil.copytree(test_dir, target)
@@ -75,8 +80,30 @@ def test_in_workerd(tmp_path, test_dir, wd_test_file, pytestconfig):
         cwd=target,
         check=True,
     )
+    workerd_common = [
+        "node_modules/workerd/bin/workerd",
+        "test",
+        wd_test_file,
+        "--experimental",
+        "--python-snapshot-dir",
+        ".",
+        "--pyodide-bundle-disk-cache-dir",
+        bundle_cache,
+    ]
+    subprocess.run(
+        [
+            *workerd_common,
+            "--python-save-snapshot",
+        ],
+        cwd=target,
+        check=True,
+    )
     subprocess.run(
-        ["node_modules/workerd/bin/workerd", "test", wd_test_file, "--experimental"],
+        [
+            *workerd_common,
+            "--python-load-snapshot",
+            "snapshot.bin",
+        ],
         cwd=target,
         check=True,
     )