@marimo-team/islands 0.23.2-dev55 → 0.23.2-dev58

package/dist/main.js

@@ -45525,7 +45525,7 @@ ${c}
       return Logger.warn("Failed to get version from mount config"), null;
     }
   }
-  const marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.2-dev55"), showCodeInRunModeAtom = atom(true);
+  const marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.2-dev58"), showCodeInRunModeAtom = atom(true);
   atom(null);
   var VIRTUAL_FILE_REGEX = /\/@file\/([^\s"&'/]+)\.([\dA-Za-z]+)/g, VirtualFileTracker = class e {
     constructor() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.23.2-dev55",
+  "version": "0.23.2-dev58",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/core/wasm/__tests__/bridge.test.ts

@@ -0,0 +1,113 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+
+const { mockNotebookReadFile, rpcListeners } = vi.hoisted(() => ({
+  mockNotebookReadFile: vi.fn(),
+  rpcListeners: {} as Record<string, () => void>,
+}));
+
+// Mock browser globals before any imports
+vi.stubGlobal("crossOriginIsolated", false);
+vi.stubGlobal(
+  "Worker",
+  vi.fn(() => ({
+    addEventListener: vi.fn(),
+    postMessage: vi.fn(),
+    terminate: vi.fn(),
+  })),
+);
+
+class MockURL extends URL {
+  static override createObjectURL = vi.fn(() => "blob:mock-url");
+  static override revokeObjectURL = vi.fn();
+}
+vi.stubGlobal("URL", MockURL);
+
+vi.mock("@/core/wasm/rpc", () => ({
+  getWorkerRPC: () => ({
+    proxy: {
+      request: {
+        bridge: vi.fn(),
+        startSession: vi.fn(),
+        readFile: vi.fn(),
+        readNotebook: vi.fn(),
+        saveNotebook: vi.fn(),
+      },
+      send: { consumerReady: vi.fn() },
+    },
+    addMessageListener: (event: string, cb: () => void) => {
+      rpcListeners[event] = cb;
+    },
+  }),
+}));
+
+vi.mock("@/core/meta/globals", () => ({
+  getMarimoVersion: () => "0.0.0-test",
+}));
+
+vi.mock("@/core/wasm/utils", () => ({
+  isWasm: () => true,
+}));
+
+vi.mock("@/core/wasm/store", () => ({
+  fallbackFileStore: { readFile: vi.fn(), saveFile: vi.fn() },
+  notebookFileStore: { readFile: mockNotebookReadFile, saveFile: vi.fn() },
+}));
+
+// Import after all mocks are set up
+import { store } from "@/core/state/jotai";
+import { initialModeAtom } from "@/core/mode";
+import { PyodideBridge } from "../bridge";
+
+// Access INSTANCE once at module level so the constructor runs (and
+// addMessageListener populates rpcListeners) before any test executes.
+void PyodideBridge.INSTANCE;
+
+describe("PyodideBridge.readCode", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  afterEach(() => {
+    store.set(initialModeAtom, undefined);
+  });
+
+  it("reads from notebookFileStore in read mode", async () => {
+    store.set(initialModeAtom, "read");
+    // Trigger getSaveWorker — it reads the current mode and returns a stub
+    // whose readNotebook delegates to notebookFileStore.
+    rpcListeners.initialized();
+    mockNotebookReadFile.mockResolvedValue(
+      "import numpy as np\nprint(np.__version__)",
+    );
+
+    const result = await PyodideBridge.INSTANCE.readCode();
+
+    expect(result).toEqual({
+      contents: "import numpy as np\nprint(np.__version__)",
+    });
+    expect(mockNotebookReadFile).toHaveBeenCalledTimes(1);
+  });
+
+  it("returns empty string when notebookFileStore has no code in read mode", async () => {
+    store.set(initialModeAtom, "read");
+    rpcListeners.initialized();
+    mockNotebookReadFile.mockResolvedValue(null);
+
+    const result = await PyodideBridge.INSTANCE.readCode();
+
+    expect(result).toEqual({ contents: "" });
+  });
+
+  it("does not call notebookFileStore in edit mode", async () => {
+    store.set(initialModeAtom, "edit");
+    // getSaveWorker in edit mode creates a real worker (mocked); readNotebook
+    // goes to the RPC proxy, not notebookFileStore.
+    rpcListeners.initialized();
+
+    await PyodideBridge.INSTANCE.readCode();
+
+    expect(mockNotebookReadFile).not.toHaveBeenCalled();
+  });
+});

package/src/core/wasm/bridge.ts

@@ -67,10 +67,10 @@ export class PyodideBridge implements RunRequests, EditRequests {
 
   private getSaveWorker(): SaveWorker {
     if (getInitialAppMode() === "read") {
-      Logger.debug("Skipping SaveWorker in read-mode");
+      Logger.debug("Using partially disabled SaveWorker in read-mode");
       return {
         readFile: throwNotImplemented,
-        readNotebook: throwNotImplemented,
+        readNotebook: async () => (await notebookFileStore.readFile()) ?? "",
         saveNotebook: throwNotImplemented,
       };
     }