ai-terminal-assistant/packages/server/tests/mocks/session.mock.ts

/**
 * SessionManager Mock 工厂
 *
 * 提供会话管理器的 mock 实现
 */

import { vi } from 'vitest';

export interface MockSession {
  id: string;
  name?: string;
  status: 'idle' | 'busy';
  createdAt: number;
  updatedAt: number;
  workdir?: string;
}

export interface MockMessage {
  id: string;
  role: 'user' | 'assistant';
  content: string;
  timestamp: number;
}

/**
 * 创建 Mock SessionManager
 */
export function createMockSessionManager() {
  const sessions = new Map<string, MockSession>();
  const messages = new Map<string, MockMessage[]>();
  let messageIdCounter = 1;

  const manager = {
    // 初始化
    init: vi.fn().mockResolvedValue(undefined),

    // Session CRUD
    exists: vi.fn((id: string) => sessions.has(id)),

    get: vi.fn((id: string) => sessions.get(id)),

    create: vi.fn((data?: Partial<MockSession>) => {
      const id = data?.id ?? `session-${Date.now()}`;
      const session: MockSession = {
        id,
        name: data?.name,
        status: 'idle',
        createdAt: Date.now(),
        updatedAt: Date.now(),
        workdir: data?.workdir ?? process.cwd(),
      };
      sessions.set(id, session);
      messages.set(id, []);
      return session;
    }),

    delete: vi.fn(async (id: string) => {
      const existed = sessions.has(id);
      sessions.delete(id);
      messages.delete(id);
      return existed;
    }),

    list: vi.fn(() => Array.from(sessions.values())),

    // 消息管理
    addMessage: vi.fn(async (sessionId: string, msg: { role: 'user' | 'assistant'; content: string }) => {
      const msgList = messages.get(sessionId) || [];
      const newMsg: MockMessage = {
        id: `msg-${messageIdCounter++}`,
        role: msg.role,
        content: msg.content,
        timestamp: Date.now(),
      };
      msgList.push(newMsg);
      messages.set(sessionId, msgList);
      return newMsg;
    }),

    getMessages: vi.fn((id: string) => messages.get(id) || []),

    // 状态更新
    updateStatus: vi.fn((id: string, status: 'idle' | 'busy') => {
      const session = sessions.get(id);
      if (session) {
        session.status = status;
        session.updatedAt = Date.now();
      }
    }),

    updateSessionName: vi.fn(async (id: string, name: string) => {
      const session = sessions.get(id);
      if (session) {
        session.name = name;
        session.updatedAt = Date.now();
        return session;
      }
      return null;
    }),

    // 测试辅助方法
    _addSession: (session: MockSession) => {
      sessions.set(session.id, session);
      messages.set(session.id, []);
    },

    _addMessage: (sessionId: string, message: MockMessage) => {
      const msgList = messages.get(sessionId) || [];
      msgList.push(message);
      messages.set(sessionId, msgList);
    },

    _clear: () => {
      sessions.clear();
      messages.clear();
      messageIdCounter = 1;
    },

    _getSessions: () => sessions,
    _getMessages: () => messages,
  };

  return manager;
}

/**
 * 创建测试用的 Session 数据
 */
export function createTestSession(overrides: Partial<MockSession> = {}): MockSession {
  return {
    id: 'test-session-1',
    name: 'Test Session',
    status: 'idle',
    createdAt: Date.now(),
    updatedAt: Date.now(),
    workdir: '/test/workdir',
    ...overrides,
  };
}

/**
 * 创建测试用的 Message 数据
 */
export function createTestMessage(overrides: Partial<MockMessage> = {}): MockMessage {
  return {
    id: 'test-msg-1',
    role: 'user',
    content: 'Test message',
    timestamp: Date.now(),
    ...overrides,
  };
}