@marimo-team/frontend 0.25.3-dev13 → 0.25.3-dev135

package/src/core/state/__tests__/jotai.test.ts

@@ -1,25 +0,0 @@
-/* Copyright 2026 Marimo. All rights reserved. */
-
-import { atom } from "jotai";
-import { describe, expect, it } from "vitest";
-import { store, waitFor } from "../jotai";
-
-describe("waitFor function", () => {
-  it("should resolve when the atom satisfies the predicate", async () => {
-    const testAtom = atom(0);
-    store.set(testAtom, 10);
-
-    const result = await waitFor(testAtom, (value) => value === 10);
-    expect(result).toBe(10);
-  });
-
-  it("should resolve when the atom changes to satisfy the predicate", async () => {
-    const testAtom = atom(0);
-
-    setTimeout(() => store.set(testAtom, 15), 5);
-    setTimeout(() => store.set(testAtom, 20), 10);
-
-    const result = await waitFor(testAtom, (value) => value === 20);
-    expect(result).toBe(20);
-  });
-});

package/src/core/websocket/__tests__/createConnectionTransport.test.ts

@@ -1,49 +0,0 @@
-/* Copyright 2026 Marimo. All rights reserved. */
-
-import { afterEach, describe, expect, it, vi } from "vitest";
-import { BasicTransport } from "../transports/basic";
-import { SseTransport } from "../transports/sse";
-import { WsTransport } from "../transports/ws";
-import { createConnectionTransport } from "../transports/create";
-
-vi.mock("../../wasm/utils", () => ({
-  isWasm: () => false,
-}));
-
-describe("createConnectionTransport", () => {
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
-  const options = {
-    url: () => "http://localhost/ws",
-    headers: () => ({}),
-  };
-
-  it("uses BasicTransport for static notebooks", () => {
-    const transport = createConnectionTransport({
-      ...options,
-      static: true,
-      transportType: "websocket",
-    });
-    expect(transport).toBeInstanceOf(BasicTransport);
-  });
-
-  it("uses WsTransport by default", () => {
-    const transport = createConnectionTransport({
-      ...options,
-      static: false,
-      transportType: "websocket",
-    });
-    expect(transport).toBeInstanceOf(WsTransport);
-  });
-
-  it("uses SseTransport when the transport type is sse", () => {
-    const transport = createConnectionTransport({
-      ...options,
-      static: false,
-      transportType: "sse",
-    });
-    expect(transport).toBeInstanceOf(SseTransport);
-  });
-});

package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx

@@ -1,140 +0,0 @@
-/* Copyright 2026 Marimo. All rights reserved. */
-// @vitest-environment jsdom
-
-import { act, renderHook } from "@testing-library/react";
-import { createStore, Provider } from "jotai";
-import type { PropsWithChildren } from "react";
-import { beforeEach, describe, expect, it, vi } from "vitest";
-import { ConnectionState } from "../types";
-import { createConnectionTransport } from "../transports/create";
-import type { IConnectionTransport } from "../transports/transport";
-import { useNotebookClient } from "../useNotebookClient";
-
-vi.mock("../transports/create", () => ({ createConnectionTransport: vi.fn() }));
-
-function makeTransport() {
-  const events = new EventTarget();
-  const transport: IConnectionTransport = {
-    readyState: WebSocket.CLOSED,
-    reconnect: vi.fn(() => {
-      transport.readyState = WebSocket.OPEN;
-      events.dispatchEvent(new Event("open"));
-    }),
-    close: vi.fn(() => {
-      transport.readyState = WebSocket.CLOSED;
-      events.dispatchEvent(new CloseEvent("close"));
-    }),
-    send: vi.fn(),
-    addEventListener: events.addEventListener.bind(events),
-    removeEventListener: events.removeEventListener.bind(events),
-  };
-  return {
-    transport,
-    notify: () =>
-      events.dispatchEvent(
-        new MessageEvent("message", {
-          data: JSON.stringify({
-            event: "marimo",
-            data: { marimo: { op: "reconnected" } },
-          }),
-        }),
-      ),
-  };
-}
-
-function setup(strict = false) {
-  const store = createStore();
-  const options = {
-    sessionId: "first",
-    static: false,
-    transportType: "websocket" as const,
-    messageFormat: "host" as const,
-    url: () => "ws://localhost/ws",
-    headers: () => ({}),
-    waitUntilReady: async () => undefined,
-    checkHealth: async () => true,
-    onNotification: vi.fn(),
-    onMessageError: vi.fn(),
-  };
-  const wrapper = ({ children }: PropsWithChildren) => (
-    <Provider store={store}>{children}</Provider>
-  );
-  const hook = renderHook((props) => useNotebookClient(props), {
-    initialProps: options,
-    reactStrictMode: strict,
-    wrapper,
-  });
-  return { ...hook, options, store };
-}
-
-beforeEach(() => {
-  vi.mocked(createConnectionTransport).mockReset();
-});
-
-describe("useNotebookClient", () => {
-  it("uses current callbacks and URL providers without reconnecting on a rerender", async () => {
-    const first = makeTransport();
-    vi.mocked(createConnectionTransport).mockReturnValue(first.transport);
-    const { rerender, options, unmount } = setup();
-    await act(async () => Promise.resolve());
-    const onNotification = vi.fn();
-    rerender({ ...options, onNotification, url: () => "ws://redirected/ws" });
-    act(() => first.notify());
-    expect(options.onNotification).not.toHaveBeenCalled();
-    expect(onNotification).toHaveBeenCalledExactlyOnceWith({
-      data: { op: "reconnected" },
-    });
-    expect(createConnectionTransport).toHaveBeenCalledOnce();
-    expect(vi.mocked(createConnectionTransport).mock.calls[0][0].url()).toBe(
-      "ws://redirected/ws",
-    );
-    unmount();
-  });
-
-  it("replaces the transport when session identity changes and ignores old messages", async () => {
-    const first = makeTransport();
-    const second = makeTransport();
-    vi.mocked(createConnectionTransport)
-      .mockReturnValueOnce(first.transport)
-      .mockReturnValueOnce(second.transport);
-    const { rerender, options, unmount } = setup();
-    await act(async () => Promise.resolve());
-    rerender({ ...options, sessionId: "second" });
-    await act(async () => Promise.resolve());
-    expect(first.transport.close).toHaveBeenCalledOnce();
-    act(() => {
-      first.notify();
-      second.notify();
-    });
-    expect(options.onNotification).toHaveBeenCalledOnce();
-    unmount();
-  });
-
-  it("cleans up every transport under Strict Mode and only connects the active lifetime", async () => {
-    const transports: ReturnType<typeof makeTransport>[] = [];
-    vi.mocked(createConnectionTransport).mockImplementation(() => {
-      const next = makeTransport();
-      transports.push(next);
-      return next.transport;
-    });
-    const { result, store, options, unmount } = setup(true);
-    await act(async () => Promise.resolve());
-    expect(transports).toHaveLength(2);
-    expect(transports[0].transport.reconnect).not.toHaveBeenCalled();
-    expect(transports[0].transport.close).toHaveBeenCalledOnce();
-    expect(transports[1].transport.reconnect).toHaveBeenCalledOnce();
-    expect(store.get(result.current.connectionAtom)).toEqual({
-      state: ConnectionState.OPEN,
-    });
-    act(() => {
-      for (const transport of transports) {
-        transport.notify();
-      }
-    });
-    expect(options.onNotification).toHaveBeenCalledOnce();
-    unmount();
-    expect(transports[1].transport.close).toHaveBeenCalledOnce();
-    await result.current.reconnect();
-    expect(transports[1].transport.reconnect).toHaveBeenCalledOnce();
-  });
-});

package/src/core/websocket/transports/__tests__/basic.test.ts

@@ -1,275 +0,0 @@
-/* Copyright 2026 Marimo. All rights reserved. */
-
-import { describe, expect, it, vi } from "vitest";
-import { BasicTransport } from "../basic";
-
-describe("BasicTransport", () => {
-  describe("close", () => {
-    it("should trigger close event", () => {
-      const transport = BasicTransport.empty();
-      const closeCallback = vi.fn();
-
-      transport.addEventListener("close", closeCallback);
-      transport.close();
-
-      expect(closeCallback).toHaveBeenCalledTimes(1);
-      expect(closeCallback).toHaveBeenCalledWith(expect.any(Event));
-    });
-  });
-
-  describe("reconnect", () => {
-    it("should trigger close and open events", async () => {
-      const transport = BasicTransport.empty();
-      const openCallback = vi.fn();
-      const closeCallback = vi.fn();
-
-      transport.addEventListener("open", openCallback);
-      transport.addEventListener("close", closeCallback);
-
-      // Clear the initial open callback
-      openCallback.mockClear();
-
-      transport.reconnect();
-
-      // Wait for connect to complete
-      await new Promise((resolve) => setTimeout(resolve, 10));
-
-      expect(closeCallback).toHaveBeenCalledTimes(1);
-      expect(openCallback).toHaveBeenCalledTimes(1);
-    });
-  });
-
-  describe("send", () => {
-    it("should trigger message event with string data", async () => {
-      const transport = BasicTransport.empty();
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("message", messageCallback);
-
-      transport.send("test message");
-
-      expect(messageCallback).toHaveBeenCalledTimes(1);
-      expect(messageCallback).toHaveBeenCalledWith(
-        expect.objectContaining({
-          data: "test message",
-        }),
-      );
-    });
-
-    it("should trigger message event with ArrayBuffer data", async () => {
-      const transport = BasicTransport.empty();
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("message", messageCallback);
-
-      const buffer = new ArrayBuffer(8);
-      transport.send(buffer);
-
-      expect(messageCallback).toHaveBeenCalledTimes(1);
-      expect(messageCallback).toHaveBeenCalledWith(
-        expect.objectContaining({
-          data: buffer,
-        }),
-      );
-    });
-  });
-
-  describe("addEventListener", () => {
-    it("should add open event listener and call it immediately", () => {
-      const transport = BasicTransport.empty();
-      const callback = vi.fn();
-
-      transport.addEventListener("open", callback);
-
-      expect(callback).toHaveBeenCalledTimes(1);
-      expect(callback).toHaveBeenCalledWith(expect.any(Event));
-    });
-
-    it("should add message event listener and start producer", () => {
-      const producer = vi.fn();
-      const transport = BasicTransport.withProducerCallback(producer);
-      const callback = vi.fn();
-
-      transport.addEventListener("message", callback);
-
-      // Producer should be called once
-      expect(producer).toHaveBeenCalledTimes(1);
-      expect(producer).toHaveBeenCalledWith(expect.any(Function));
-    });
-  });
-
-  describe("removeEventListener", () => {
-    it("should remove event listener", async () => {
-      const transport = BasicTransport.empty();
-      const callback = vi.fn();
-
-      transport.addEventListener("message", callback);
-      transport.removeEventListener("message", callback);
-
-      transport.send("test");
-
-      expect(callback).not.toHaveBeenCalled();
-    });
-
-    it("should only remove the specific listener", () => {
-      const transport = BasicTransport.empty();
-      const callback1 = vi.fn();
-      const callback2 = vi.fn();
-
-      transport.addEventListener("close", callback1);
-      transport.addEventListener("close", callback2);
-
-      transport.removeEventListener("close", callback1);
-
-      transport.close();
-
-      expect(callback1).not.toHaveBeenCalled();
-      expect(callback2).toHaveBeenCalledTimes(1);
-    });
-  });
-
-  describe("Producer functionality", () => {
-    it("should pass producer callback that triggers message events", () => {
-      let producerCallback: ((message: MessageEvent) => void) | undefined;
-      const producer = vi.fn((callback) => {
-        producerCallback = callback;
-      });
-      const transport = BasicTransport.withProducerCallback(producer);
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("message", messageCallback);
-
-      expect(producerCallback).toBeDefined();
-
-      const testMessage = new MessageEvent("message", { data: "test" });
-      producerCallback?.(testMessage);
-
-      expect(messageCallback).toHaveBeenCalledWith(testMessage);
-    });
-
-    it("should handle multiple messages from producer", () => {
-      let producerCallback: ((message: MessageEvent) => void) | undefined;
-      const producer = vi.fn((callback) => {
-        producerCallback = callback;
-      });
-      const transport = BasicTransport.withProducerCallback(producer);
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("message", messageCallback);
-
-      producerCallback?.(new MessageEvent("message", { data: "msg1" }));
-      producerCallback?.(new MessageEvent("message", { data: "msg2" }));
-      producerCallback?.(new MessageEvent("message", { data: "msg3" }));
-
-      expect(messageCallback).toHaveBeenCalledTimes(3);
-      expect(messageCallback).toHaveBeenNthCalledWith(
-        1,
-        expect.objectContaining({ data: "msg1" }),
-      );
-      expect(messageCallback).toHaveBeenNthCalledWith(
-        2,
-        expect.objectContaining({ data: "msg2" }),
-      );
-      expect(messageCallback).toHaveBeenNthCalledWith(
-        3,
-        expect.objectContaining({ data: "msg3" }),
-      );
-    });
-
-    it("should not start producer if no message listener is added", () => {
-      const producer = vi.fn();
-      const transport = BasicTransport.withProducerCallback(producer);
-
-      transport.addEventListener("open", vi.fn());
-      transport.addEventListener("close", vi.fn());
-
-      expect(producer).not.toHaveBeenCalled();
-    });
-  });
-
-  describe("Integration scenarios", () => {
-    it("should handle complete connection lifecycle", async () => {
-      const transport = BasicTransport.empty();
-      const openCallback = vi.fn();
-      const closeCallback = vi.fn();
-
-      transport.addEventListener("open", openCallback);
-      transport.addEventListener("close", closeCallback);
-
-      expect(openCallback).toHaveBeenCalledTimes(1);
-
-      transport.reconnect();
-      await new Promise((resolve) => setTimeout(resolve, 10));
-      expect(openCallback).toHaveBeenCalledTimes(2);
-      expect(closeCallback).toHaveBeenCalledTimes(1);
-
-      transport.close();
-      expect(closeCallback).toHaveBeenCalledTimes(2);
-    });
-
-    it("should handle send and receive flow", async () => {
-      let producerCallback: ((message: MessageEvent) => void) | undefined;
-      const producer = vi.fn((callback) => {
-        producerCallback = callback;
-      });
-      const transport = BasicTransport.withProducerCallback(producer);
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("message", messageCallback);
-
-      transport.send("outgoing");
-      expect(messageCallback).toHaveBeenCalledWith(
-        expect.objectContaining({ data: "outgoing" }),
-      );
-
-      messageCallback.mockClear();
-
-      producerCallback?.(new MessageEvent("message", { data: "incoming" }));
-      expect(messageCallback).toHaveBeenCalledWith(
-        expect.objectContaining({ data: "incoming" }),
-      );
-    });
-
-    it("should maintain separate event queues for different event types", async () => {
-      const transport = BasicTransport.empty();
-      const openCallback = vi.fn();
-      const closeCallback = vi.fn();
-      const messageCallback = vi.fn();
-
-      transport.addEventListener("open", openCallback);
-      transport.addEventListener("close", closeCallback);
-      transport.addEventListener("message", messageCallback);
-
-      openCallback.mockClear();
-
-      transport.reconnect();
-      await new Promise((resolve) => setTimeout(resolve, 10));
-      transport.send("test");
-      transport.close();
-
-      expect(openCallback).toHaveBeenCalledTimes(1);
-      expect(messageCallback).toHaveBeenCalledTimes(1);
-      expect(closeCallback).toHaveBeenCalledTimes(2);
-    });
-
-    it("should handle rapid reconnections", async () => {
-      const transport = BasicTransport.empty();
-      const openCallback = vi.fn();
-      const closeCallback = vi.fn();
-
-      transport.addEventListener("open", openCallback);
-      transport.addEventListener("close", closeCallback);
-
-      openCallback.mockClear();
-
-      transport.reconnect();
-      transport.reconnect();
-      transport.reconnect();
-
-      await new Promise((resolve) => setTimeout(resolve, 20));
-
-      expect(closeCallback).toHaveBeenCalledTimes(3);
-      expect(openCallback).toHaveBeenCalledTimes(3);
-    });
-  });
-});