@marimo-team/frontend 0.25.1-dev27 → 0.25.1-dev28

package/src/core/dom/uiregistry.ts

@@ -45,12 +45,12 @@ interface UIElementEntry {
  */
 export class UIElementRegistry {
   // maps UIElement objectIds to entries.
-  entries: Map<UIElementId, UIElementEntry>;
+  public entries: Map<UIElementId, UIElementEntry>;
 
   /**
    * Shared instance of UIElementRegistry since this must be a singleton.
    */
-  static get INSTANCE(): UIElementRegistry {
+  public static get INSTANCE(): UIElementRegistry {
     const KEY = "_marimo_private_UIElementRegistry";
     if (!window[KEY]) {
       window[KEY] = new UIElementRegistry();
@@ -62,11 +62,11 @@ export class UIElementRegistry {
     this.entries = new Map();
   }
 
-  has(objectId: UIElementId): boolean {
+  public has(objectId: UIElementId): boolean {
     return this.entries.has(objectId);
   }
 
-  set(objectId: UIElementId, value: ValueType): void {
+  public set(objectId: UIElementId, value: ValueType): void {
     if (this.entries.has(objectId)) {
       Logger.debug(
         "UIElementRegistry overwriting entry for objectId.",
@@ -86,7 +86,7 @@ export class UIElementRegistry {
    * @param objectId - id of the UIElement
    * @param instance - the HTMLElement that the UIElement wraps
    */
-  registerInstance(objectId: UIElementId, instance: HTMLElement) {
+  public registerInstance(objectId: UIElementId, instance: HTMLElement) {
     const entry = this.entries.get(objectId);
     if (entry === undefined) {
       this.entries.set(objectId, {
@@ -109,7 +109,7 @@ export class UIElementRegistry {
    * @param instance - the HTMLElement to remove
    *
    */
-  removeInstance(objectId: UIElementId, instance: HTMLElement) {
+  public removeInstance(objectId: UIElementId, instance: HTMLElement) {
     const entry = this.entries.get(objectId);
     // The UIElement can be removed from the registry before all
     // instances are removed: UIElement removal is triggered
@@ -129,7 +129,7 @@ export class UIElementRegistry {
    *
    * @param cellId - stringified cellId
    */
-  removeElementsByCell(cellId: CellId) {
+  public removeElementsByCell(cellId: CellId) {
     const objectIds = [...this.entries.keys()].filter((objectId) =>
       objectId.startsWith(`${cellId}-`),
     );
@@ -145,12 +145,12 @@ export class UIElementRegistry {
    * @param objectId - id of the UIElement
    * @returns the value for `objectId`, or `undefined` if the object was not found.
    */
-  lookupValue(objectId: UIElementId): ValueType {
+  public lookupValue(objectId: UIElementId): ValueType {
     const entry = this.entries.get(objectId);
     return entry === undefined ? undefined : entry.value;
   }
 
-  broadcastMessage(
+  public broadcastMessage(
     objectId: UIElementId,
     message: unknown,
     buffers: readonly DataView[],
@@ -202,7 +202,7 @@ export class UIElementRegistry {
    * @param objectId - id of the UIElement
    * @param value - value to broadcast
    */
-  broadcastValueUpdate(
+  public broadcastValueUpdate(
     initiator: HTMLElement,
     objectId: UIElementId,
     value: ValueType,

package/src/core/hotkeys/hotkeys.ts

@@ -510,13 +510,13 @@ export class HotkeyProvider implements IHotkeyProvider {
   /**
    * @param platform - See {@link HotkeyProviderOptions.platform}.
    */
-  static create(platform?: Platform): HotkeyProvider {
+  public static create(platform?: Platform): HotkeyProvider {
     return new HotkeyProvider(DEFAULT_HOT_KEY, { platform });
   }
 
   private hotkeys: Record<HotkeyAction, Hotkey>;
 
-  constructor(
+  public constructor(
     hotkeys: Record<HotkeyAction, Hotkey>,
     options: HotkeyProviderOptions = {},
   ) {
@@ -525,11 +525,11 @@ export class HotkeyProvider implements IHotkeyProvider {
     this.mod = this.platform === "mac" ? "Cmd" : "Ctrl";
   }
 
-  iterate(): HotkeyAction[] {
+  public iterate(): HotkeyAction[] {
     return Objects.keys(this.hotkeys);
   }
 
-  getHotkey(action: HotkeyAction): ResolvedHotkey {
+  public getHotkey(action: HotkeyAction): ResolvedHotkey {
     const { name, key, additionalKeywords } = this.hotkeys[action];
     if (typeof key === "string") {
       return {
@@ -553,15 +553,15 @@ export class HotkeyProvider implements IHotkeyProvider {
     };
   }
 
-  getHotkeyDisplay(action: HotkeyAction): string {
+  public getHotkeyDisplay(action: HotkeyAction): string {
     return this.hotkeys[action].name;
   }
 
-  isEditable(action: HotkeyAction): boolean {
+  public isEditable(action: HotkeyAction): boolean {
     return this.hotkeys[action].editable !== false;
   }
 
-  getHotkeyGroups(): Record<HotkeyGroup, HotkeyAction[]> {
+  public getHotkeyGroups(): Record<HotkeyGroup, HotkeyAction[]> {
     return Objects.groupBy(
       Objects.entries(this.hotkeys),
       ([, hotkey]) => hotkey.group,
@@ -573,7 +573,7 @@ export class HotkeyProvider implements IHotkeyProvider {
 export class OverridingHotkeyProvider extends HotkeyProvider {
   private readonly overrides: Partial<Record<HotkeyAction, string | undefined>>;
 
-  constructor(
+  public constructor(
     overrides: Partial<Record<HotkeyAction, string | undefined>>,
     options: HotkeyProviderOptions = {},
   ) {
@@ -581,7 +581,7 @@ export class OverridingHotkeyProvider extends HotkeyProvider {
     this.overrides = overrides;
   }
 
-  override getHotkey(action: HotkeyAction): ResolvedHotkey {
+  public override getHotkey(action: HotkeyAction): ResolvedHotkey {
     const base = super.getHotkey(action);
     const override = this.overrides[action];
     return {

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

@@ -38,12 +38,12 @@ vi.stubGlobal(
 
 // Create a mock URL class that works as a constructor
 class MockURL {
-  href: string;
-  constructor(url: string, base?: string | URL) {
+  public href: string;
+  public constructor(url: string, base?: string | URL) {
     this.href = base ? `${base}/${url}` : url;
   }
-  static createObjectURL = vi.fn(() => "blob:mock-url");
-  static revokeObjectURL = vi.fn();
+  public static createObjectURL = vi.fn(() => "blob:mock-url");
+  public static revokeObjectURL = vi.fn();
 }
 vi.stubGlobal("URL", MockURL);
 

package/src/core/islands/__tests__/test-utils.tsx

@@ -236,11 +236,11 @@ export class MockWorkerFactory implements WorkerFactory {
   public workers: Worker[] = [];
   private readonly mockWorker?: Worker;
 
-  constructor(mockWorker?: Worker) {
+  public constructor(mockWorker?: Worker) {
     this.mockWorker = mockWorker;
   }
 
-  create(): Worker {
+  public create(): Worker {
     const worker = this.mockWorker || this.createMockWorker();
     this.workers.push(worker);
     return worker;
@@ -259,11 +259,11 @@ export class MockWorkerFactory implements WorkerFactory {
     } as unknown as Worker;
   }
 
-  getCreatedWorkers(): Worker[] {
+  public getCreatedWorkers(): Worker[] {
     return this.workers;
   }
 
-  terminateAll(): void {
+  public terminateAll(): void {
     for (const worker of this.workers) {
       worker.terminate();
     }

package/src/core/islands/bridge.ts

@@ -72,7 +72,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
 
   public initialized = new Deferred<void>();
 
-  constructor(config: IslandsBridgeConfig = {}) {
+  public constructor(config: IslandsBridgeConfig = {}) {
     this.store = config.store || defaultStore;
     this.root = config.root || document;
 
@@ -120,7 +120,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     });
   }
 
-  async initializeApps(): Promise<void> {
+  public async initializeApps(): Promise<void> {
     await this.enqueueAppTransition(async () => {
       await this.workerReady.promise;
       await this.startApps();
@@ -211,7 +211,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     }
   }
 
-  async stopSession(appId?: string): Promise<void> {
+  public async stopSession(appId?: string): Promise<void> {
     await this.enqueueAppTransition(async () => {
       const session = this.session;
       if (session?.code === undefined || (appId && session.appId !== appId)) {
@@ -238,7 +238,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Starts a new Python session for an app
    */
-  async startSession(opts: {
+  public async startSession(opts: {
     code: string;
     appId: string;
     sessionGeneration: number;
@@ -249,7 +249,9 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Sets up a consumer for kernel messages
    */
-  consumeMessages(consumer: (message: IslandsKernelMessage) => void): void {
+  public consumeMessages(
+    consumer: (message: IslandsKernelMessage) => void,
+  ): void {
     this.messageConsumer = consumer;
     this.rpc.proxy.send.consumerReady({});
   }
@@ -258,7 +260,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // RunRequests Implementation
   // ============================================================================
 
-  sendComponentValues: RunRequests["sendComponentValues"] = async (
+  public sendComponentValues: RunRequests["sendComponentValues"] = async (
     request,
   ): Promise<null> => {
     await this.putControlRequest({
@@ -269,11 +271,12 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     return null;
   };
 
-  sendInstantiate: RunRequests["sendInstantiate"] = async (): Promise<null> => {
-    return null;
-  };
+  public sendInstantiate: RunRequests["sendInstantiate"] =
+    async (): Promise<null> => {
+      return null;
+    };
 
-  sendFunctionRequest: RunRequests["sendFunctionRequest"] = async (
+  public sendFunctionRequest: RunRequests["sendFunctionRequest"] = async (
     request,
   ): Promise<null> => {
     await this.putControlRequest({
@@ -283,7 +286,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     return null;
   };
 
-  sendModelValue: RunRequests["sendModelValue"] = async (request) => {
+  public sendModelValue: RunRequests["sendModelValue"] = async (request) => {
     await this.putControlRequest({
       type: "model",
       ...request,
@@ -295,7 +298,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // EditRequests Implementation
   // ============================================================================
 
-  sendRun: EditRequests["sendRun"] = async (request): Promise<null> => {
+  public sendRun: EditRequests["sendRun"] = async (request): Promise<null> => {
     const session = await this.getActiveSession();
     await this.rpc.proxy.request.loadPackages({
       appId: session.appId,
@@ -316,75 +319,75 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // Not Implemented (Read-Only Mode)
   // ============================================================================
 
-  getUsageStats = throwNotImplemented;
-  getEnvironmentInfo = throwNotImplemented;
-  sendRename = throwNotImplemented;
-  sendSave = throwNotImplemented;
-  sendCopy = throwNotImplemented;
-  sendRunScratchpad = throwNotImplemented;
-  sendStdin = throwNotImplemented;
-  sendInterrupt = throwNotImplemented;
-  sendShutdown = throwNotImplemented;
-  sendFormat = throwNotImplemented;
-  sendDeleteCell = throwNotImplemented;
-  sendInstallMissingPackages = throwNotImplemented;
-  sendCodeCompletionRequest = throwNotImplemented;
-  saveUserConfig = throwNotImplemented;
-  saveAppConfig = throwNotImplemented;
-  saveCellConfig = throwNotImplemented;
-  sendRestart = throwNotImplemented;
-  sendDocumentTransaction = throwNotImplemented;
-  readCode = throwNotImplemented;
-  readSnippets = throwNotImplemented;
-  previewDatasetColumn = throwNotImplemented;
-  previewSQLTable = throwNotImplemented;
-  previewSQLTableList = throwNotImplemented;
-  previewSQLSchemaList = throwNotImplemented;
-  previewDataSourceConnection = throwNotImplemented;
-  discoverDataSources = throwNotImplemented;
-  validateSQL = throwNotImplemented;
-  openFile = throwNotImplemented;
-  getFileRoots = throwNotImplemented;
-  sendListFiles = throwNotImplemented;
-  sendSearchFiles = throwNotImplemented;
-  sendPdb = throwNotImplemented;
-  sendSetBreakpoints = throwNotImplemented;
-  sendCreateFileOrFolder = throwNotImplemented;
-  sendDeleteFileOrFolder = throwNotImplemented;
-  sendCopyFileOrFolder = throwNotImplemented;
-  sendRenameFileOrFolder = throwNotImplemented;
-  sendUpdateFile = throwNotImplemented;
-  sendFileDetails = throwNotImplemented;
-  openTutorial = throwNotImplemented;
-  getExportAvailability = throwNotImplemented;
-  installExportRequirements = throwNotImplemented;
-  exportAsHTML = throwNotImplemented;
-  exportAsIPYNB = throwNotImplemented;
-  exportAsMarkdown = throwNotImplemented;
-  exportAsScript = throwNotImplemented;
-  exportAsPDF = throwNotImplemented;
-  autoExportAsHTML = throwNotImplemented;
-  autoExportAsMarkdown = throwNotImplemented;
-  autoExportAsIPYNB = throwNotImplemented;
-  updateCellOutputs = throwNotImplemented;
-  addPackage = throwNotImplemented;
-  removePackage = throwNotImplemented;
-  getPackageList = throwNotImplemented;
-  getSandbox = throwNotImplemented;
-  updateManifest = throwNotImplemented;
-  syncSandbox = throwNotImplemented;
-  getDependencyTree = throwNotImplemented;
-  getRecentFiles = throwNotImplemented;
-  getWorkspaceFiles = throwNotImplemented;
-  getRunningNotebooks = throwNotImplemented;
-  shutdownSession = throwNotImplemented;
-  listSecretKeys = throwNotImplemented;
-  writeSecret = throwNotImplemented;
-  invokeAiTool = throwNotImplemented;
-  clearCache = throwNotImplemented;
-  getCacheInfo = throwNotImplemented;
-  listStorageEntries = throwNotImplemented;
-  downloadStorage = throwNotImplemented;
+  public getUsageStats = throwNotImplemented;
+  public getEnvironmentInfo = throwNotImplemented;
+  public sendRename = throwNotImplemented;
+  public sendSave = throwNotImplemented;
+  public sendCopy = throwNotImplemented;
+  public sendRunScratchpad = throwNotImplemented;
+  public sendStdin = throwNotImplemented;
+  public sendInterrupt = throwNotImplemented;
+  public sendShutdown = throwNotImplemented;
+  public sendFormat = throwNotImplemented;
+  public sendDeleteCell = throwNotImplemented;
+  public sendInstallMissingPackages = throwNotImplemented;
+  public sendCodeCompletionRequest = throwNotImplemented;
+  public saveUserConfig = throwNotImplemented;
+  public saveAppConfig = throwNotImplemented;
+  public saveCellConfig = throwNotImplemented;
+  public sendRestart = throwNotImplemented;
+  public sendDocumentTransaction = throwNotImplemented;
+  public readCode = throwNotImplemented;
+  public readSnippets = throwNotImplemented;
+  public previewDatasetColumn = throwNotImplemented;
+  public previewSQLTable = throwNotImplemented;
+  public previewSQLTableList = throwNotImplemented;
+  public previewSQLSchemaList = throwNotImplemented;
+  public previewDataSourceConnection = throwNotImplemented;
+  public discoverDataSources = throwNotImplemented;
+  public validateSQL = throwNotImplemented;
+  public openFile = throwNotImplemented;
+  public getFileRoots = throwNotImplemented;
+  public sendListFiles = throwNotImplemented;
+  public sendSearchFiles = throwNotImplemented;
+  public sendPdb = throwNotImplemented;
+  public sendSetBreakpoints = throwNotImplemented;
+  public sendCreateFileOrFolder = throwNotImplemented;
+  public sendDeleteFileOrFolder = throwNotImplemented;
+  public sendCopyFileOrFolder = throwNotImplemented;
+  public sendRenameFileOrFolder = throwNotImplemented;
+  public sendUpdateFile = throwNotImplemented;
+  public sendFileDetails = throwNotImplemented;
+  public openTutorial = throwNotImplemented;
+  public getExportAvailability = throwNotImplemented;
+  public installExportRequirements = throwNotImplemented;
+  public exportAsHTML = throwNotImplemented;
+  public exportAsIPYNB = throwNotImplemented;
+  public exportAsMarkdown = throwNotImplemented;
+  public exportAsScript = throwNotImplemented;
+  public exportAsPDF = throwNotImplemented;
+  public autoExportAsHTML = throwNotImplemented;
+  public autoExportAsMarkdown = throwNotImplemented;
+  public autoExportAsIPYNB = throwNotImplemented;
+  public updateCellOutputs = throwNotImplemented;
+  public addPackage = throwNotImplemented;
+  public removePackage = throwNotImplemented;
+  public getPackageList = throwNotImplemented;
+  public getSandbox = throwNotImplemented;
+  public updateManifest = throwNotImplemented;
+  public syncSandbox = throwNotImplemented;
+  public getDependencyTree = throwNotImplemented;
+  public getRecentFiles = throwNotImplemented;
+  public getWorkspaceFiles = throwNotImplemented;
+  public getRunningNotebooks = throwNotImplemented;
+  public shutdownSession = throwNotImplemented;
+  public listSecretKeys = throwNotImplemented;
+  public writeSecret = throwNotImplemented;
+  public invokeAiTool = throwNotImplemented;
+  public clearCache = throwNotImplemented;
+  public getCacheInfo = throwNotImplemented;
+  public listStorageEntries = throwNotImplemented;
+  public downloadStorage = throwNotImplemented;
 
   // The kernel uses msgspec to parse control requests, which requires a 'type'
   // field for discriminated union deserialization.
@@ -412,7 +415,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Cleans up resources (for testing)
    */
-  destroy(): void {
+  public destroy(): void {
     // Future: terminate worker if we own it
   }
 }

package/src/core/islands/components/web-components.tsx

@@ -51,7 +51,7 @@ export class MarimoIslandElement extends HTMLElement {
   public static readonly editorTagName = ISLAND_TAG_NAMES.CODE_EDITOR;
   public static readonly styleNamespace = ISLAND_CSS_CLASSES.NAMESPACE;
 
-  constructor() {
+  public constructor() {
     super();
     this.classList.add(MarimoIslandElement.styleNamespace);
   }
@@ -59,7 +59,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Gets the app ID from the element's data attribute
    */
-  get appId(): string {
+  public get appId(): string {
     const appId = this.getAttribute(ISLAND_DATA_ATTRIBUTES.APP_ID);
     invariant(appId, "Missing data-app-id attribute");
     return appId;
@@ -68,7 +68,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Whether this island is reactive (has code sent to Python for execution)
    */
-  get isReactive(): boolean {
+  public get isReactive(): boolean {
     return this.getAttribute(ISLAND_DATA_ATTRIBUTES.REACTIVE) === "true";
   }
 
@@ -76,7 +76,7 @@ export class MarimoIslandElement extends HTMLElement {
    * Gets the cell ID by looking up the cell index in the notebook state.
    * Returns undefined for non-reactive islands (they have no corresponding cell).
    */
-  get cellId(): CellId | undefined {
+  public get cellId(): CellId | undefined {
     if (!this.isReactive) {
       return undefined;
     }
@@ -95,7 +95,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Gets the code for this island cell
    */
-  get code(): string {
+  public get code(): string {
     return extractIslandCodeFromEmbed(this);
   }
 
@@ -115,7 +115,7 @@ export class MarimoIslandElement extends HTMLElement {
    * which can happen inside a React render cycle. Rendering
    * synchronously from there causes "unmount during render" warnings.
    */
-  connectedCallback(): void {
+  public connectedCallback(): void {
     const connectionGeneration = ++this.connectionGeneration;
     this.addEventListener(
       ISLAND_SOURCE_CHANGED_EVENT,
@@ -296,7 +296,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Cleanup when element is removed from DOM
    */
-  disconnectedCallback(): void {
+  public disconnectedCallback(): void {
     this.removeEventListener(
       ISLAND_SOURCE_CHANGED_EVENT,
       this.handleSourceChanged,

package/src/core/islands/worker/__tests__/controller.test.ts

@@ -5,7 +5,7 @@ import { describe, expect, it, vi } from "vitest";
 import { ReadonlyWasmController } from "../controller";
 
 class TestController extends ReadonlyWasmController {
-  setPyodide(pyodide: PyodideInterface) {
+  public setPyodide(pyodide: PyodideInterface) {
     this.pyodide = pyodide;
   }
 }

package/src/core/islands/worker/controller.ts

@@ -9,7 +9,7 @@ import type { JsonString } from "@/utils/json/base64";
 import { Logger } from "../../../utils/Logger";
 
 export class ReadonlyWasmController extends DefaultWasmController {
-  override async bootstrap(opts: {
+  public override async bootstrap(opts: {
     version: string;
     pyodideVersion: string;
   }): Promise<PyodideInterface> {
@@ -17,7 +17,10 @@ export class ReadonlyWasmController extends DefaultWasmController {
     return pyodide;
   }
 
-  override async mountFilesystem(opts: { code: string; filename: string }) {
+  public override async mountFilesystem(opts: {
+    code: string;
+    filename: string;
+  }) {
     const { code, filename } = opts;
     // Write file
     try {
@@ -34,7 +37,7 @@ export class ReadonlyWasmController extends DefaultWasmController {
     return { code, filename };
   }
 
-  override async startSession(opts: {
+  public override async startSession(opts: {
     code: string;
     filename: string | null;
     onMessage: (message: JsonString<NotificationPayload>) => void;

package/src/core/islands/worker-factory.ts

@@ -39,7 +39,7 @@ export class DefaultWorkerFactory implements WorkerFactory {
   private readonly url: string;
   private readonly name: string;
 
-  constructor(config: DefaultWorkerFactoryConfig = {}) {
+  public constructor(config: DefaultWorkerFactoryConfig = {}) {
     this.url = config.workerUrl || this.getDefaultWorkerUrl();
     this.name = config.workerName || getMarimoVersion();
   }
@@ -47,7 +47,7 @@ export class DefaultWorkerFactory implements WorkerFactory {
   /**
    * Creates a new Pyodide worker
    */
-  create(): Worker {
+  public create(): Worker {
     return createModuleWorker(new URL(this.url, import.meta.url), {
       name: this.name,
     });

package/src/core/kernel/RuntimeState.ts

@@ -17,7 +17,7 @@ export class RuntimeState {
   /**
    * Shared instance of RuntimeState since this must be a singleton.
    */
-  static get INSTANCE(): RuntimeState {
+  public static get INSTANCE(): RuntimeState {
     const KEY = "_marimo_private_RuntimeState";
     if (!window[KEY]) {
       window[KEY] = new RuntimeState(UI_ELEMENT_REGISTRY);
@@ -31,7 +31,7 @@ export class RuntimeState {
   private _sendComponentValues: RunRequests["sendComponentValues"] | undefined;
   private uiElementRegistry: UIElementRegistry;
 
-  constructor(uiElementRegistry: UIElementRegistry) {
+  public constructor(uiElementRegistry: UIElementRegistry) {
     this.uiElementRegistry = uiElementRegistry;
   }
 
@@ -45,7 +45,7 @@ export class RuntimeState {
   /**
    * Start listening for events from UIElements
    */
-  start(sendComponentValues: RunRequests["sendComponentValues"]) {
+  public start(sendComponentValues: RunRequests["sendComponentValues"]) {
     if (this.hasStarted) {
       Logger.warn("RuntimeState already started");
       return;
@@ -61,7 +61,7 @@ export class RuntimeState {
   /**
    * Stop listening for events from UIElements
    */
-  stop() {
+  public stop() {
     if (!this.hasStarted) {
       Logger.warn("RuntimeState already stopped");
       return;

package/src/core/lsp/__tests__/transport.test.ts

@@ -10,28 +10,28 @@ vi.mock("@/utils/Logger", () => ({
 }));
 
 class FakeWebSocket {
-  static readonly CONNECTING = 0;
-  static readonly OPEN = 1;
-  static readonly CLOSING = 2;
-  static readonly CLOSED = 3;
-  static readonly instances: FakeWebSocket[] = [];
-
-  readonly sent: string[] = [];
-  readonly url: string;
-  readonly protocols: string | string[] | undefined;
-  readyState = FakeWebSocket.CONNECTING;
+  public static readonly CONNECTING = 0;
+  public static readonly OPEN = 1;
+  public static readonly CLOSING = 2;
+  public static readonly CLOSED = 3;
+  public static readonly instances: FakeWebSocket[] = [];
+
+  public readonly sent: string[] = [];
+  public readonly url: string;
+  public readonly protocols: string | string[] | undefined;
+  public readyState = FakeWebSocket.CONNECTING;
   private readonly listeners = new Map<
     string,
     Set<(event: { data?: unknown }) => void>
   >();
 
-  constructor(url: string, protocols?: string | string[]) {
+  public constructor(url: string, protocols?: string | string[]) {
     this.url = url;
     this.protocols = protocols;
     FakeWebSocket.instances.push(this);
   }
 
-  addEventListener(
+  public addEventListener(
     type: string,
     listener: (event: { data?: unknown }) => void,
   ) {
@@ -40,34 +40,34 @@ class FakeWebSocket {
     this.listeners.set(type, listeners);
   }
 
-  send(frame: string) {
+  public send(frame: string) {
     this.sent.push(frame);
   }
 
-  close() {
+  public close() {
     this.readyState = FakeWebSocket.CLOSED;
     this.emit("close", {});
   }
 
-  open() {
+  public open() {
     this.readyState = FakeWebSocket.OPEN;
     this.emit("open", {});
   }
 
-  fail() {
+  public fail() {
     this.emit("error", {});
   }
 
-  disconnect() {
+  public disconnect() {
     this.readyState = FakeWebSocket.CLOSED;
     this.emit("close", {});
   }
 
-  receive(message: JSONRPCMessage) {
+  public receive(message: JSONRPCMessage) {
     this.emit("message", { data: JSON.stringify(message) });
   }
 
-  messages(): JSONRPCMessage[] {
+  public messages(): JSONRPCMessage[] {
     return this.sent.map((frame) => {
       const message: unknown = JSON.parse(frame);
       if (!isJSONRPCMessage(message)) {