/* Copyright 2026 Marimo. All rights reserved. */
import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";
import { PyodideRouter } from "../router";

describe("URLPyodideRouter", () => {
  const originalLocation = window.location;
  const originalReplaceState = window.history.replaceState;

  beforeEach(() => {
    window.location = { href: "https://marimo.app" } as never;
    window.history.replaceState = vi.fn();
  });

  afterEach(() => {
    window.location = originalLocation as never;
    window.history.replaceState = originalReplaceState;
  });

  test("getFilename returns correct filename from URL", () => {
    window.location.href = "https://marimo.app?filename=test.py";
    expect(PyodideRouter.getFilename()).toBe("test.py");
  });

  test("getFilename returns null when filename is not in URL", () => {
    window.location.href = "https://marimo.app";
    expect(PyodideRouter.getFilename()).toBeNull();
  });

  test("getCode returns correct code from URL", () => {
    window.location.href = 'https://marimo.app?code=print("Hello, World!")';
    expect(PyodideRouter.getCodeFromSearchParam()).toBe(
      'print("Hello, World!")',
    );
  });

  test("getCode from hash returns correct code", () => {
    window.location.href = "https://marimo.app";
    window.location.hash = "#code/print('Hello, World!')";
    expect(PyodideRouter.getCodeFromHash()).toBe("print('Hello, World!')");
  });

  test("getCode from hash with query params returns correct code", () => {
    window.location.href = "https://marimo.app?filename=test.py";
    window.location.hash = "#code/print('Hello, World!')";
    expect(PyodideRouter.getCodeFromHash()).toBe("print('Hello, World!')");
    expect(PyodideRouter.getFilename()).toBe("test.py");
  });

  test("setFilename correctly modifies URL", () => {
    PyodideRouter.setFilename("newfile.py");
    expect(window.history.replaceState).toHaveBeenCalledWith(
      {},
      "",
      "https://marimo.app/?filename=newfile.py",
    );
  });
});