OpenHands/frontend/__tests__/services/session.test.ts

import { afterEach, describe, expect, it, vi } from "vitest";
import ActionType from "#/types/ActionType";
import { Settings, saveSettings } from "../../src/services/settings";
import Session from "../../src/services/session";

const sendSpy = vi.spyOn(Session, "send");
// @ts-expect-error - spying on private function
const setupSpy = vi.spyOn(Session, "_setupSocket").mockImplementation(() => {
  // @ts-expect-error - calling a private function
  Session._initializeAgent();
});

describe("startNewSession", () => {
  afterEach(() => {
    sendSpy.mockClear();
    setupSpy.mockClear();
  });

  it("Should start a new session with the current settings", () => {
    const settings: Settings = {
      LLM_MODEL: "llm_value",
      LLM_BASE_URL: "base_url",
      AGENT: "agent_value",
      LANGUAGE: "language_value",
      LLM_API_KEY: "sk-...",
      CONFIRMATION_MODE: true,
      SECURITY_ANALYZER: "analyzer",
    };

    const event = {
      action: ActionType.INIT,
      args: settings,
    };

    saveSettings(settings);
    Session.startNewSession();

    expect(setupSpy).toHaveBeenCalledTimes(1);
    expect(sendSpy).toHaveBeenCalledWith(JSON.stringify(event));
  });
});