@monotykamary/pi-tps 1.3.4 → 1.3.6

package/extensions/pi-tps/__tests__/extension-setup.test.ts

@@ -1,41 +0,0 @@
-import { describe, it, expect, beforeEach, afterEach } from 'vitest';
-import { createTestFixture, activateExtension } from './helpers';
-
-describe('pi-tps extension — setup', () => {
-  let fixture: ReturnType<typeof createTestFixture>;
-
-  beforeEach(async () => {
-    fixture = createTestFixture();
-    await activateExtension(fixture);
-  });
-
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
-  it('should register all required event handlers and commands', () => {
-    const { mockPi, registerCommandSpy } = fixture;
-
-    expect(mockPi.on).toHaveBeenCalledWith('session_start', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('session_tree', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('turn_start', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('message_start', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('message_update', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('message_end', expect.any(Function));
-    expect(mockPi.on).toHaveBeenCalledWith('turn_end', expect.any(Function));
-    expect(registerCommandSpy).toHaveBeenCalledWith(
-      'tps-export',
-      expect.objectContaining({
-        description: expect.any(String),
-        handler: expect.any(Function),
-      })
-    );
-    expect(registerCommandSpy).toHaveBeenCalledWith(
-      'session-export',
-      expect.objectContaining({
-        description: expect.any(String),
-        handler: expect.any(Function),
-      })
-    );
-  });
-});

package/extensions/pi-tps/__tests__/format-duration.test.ts

@@ -1,83 +0,0 @@
-import { describe, it, expect } from 'vitest';
-
-describe('formatDuration', () => {
-  const importFormatDuration = async () => {
-    const mod = await import('../index.js');
-    return mod.formatDuration;
-  };
-
-  it('formats sub-minute durations with 1 decimal', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(0.8)).toBe('0.8s');
-    expect(formatDuration(1.0)).toBe('1.0s');
-    expect(formatDuration(2.3)).toBe('2.3s');
-    expect(formatDuration(9.9)).toBe('9.9s');
-    expect(formatDuration(10.5)).toBe('10.5s');
-    expect(formatDuration(45.0)).toBe('45.0s');
-    expect(formatDuration(59.4)).toBe('59.4s');
-  });
-
-  it('formats minute+ durations as minutes and seconds', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(60)).toBe('1m 0s');
-    expect(formatDuration(90)).toBe('1m 30s');
-    expect(formatDuration(300)).toBe('5m 0s');
-    expect(formatDuration(323)).toBe('5m 23s');
-    expect(formatDuration(3599)).toBe('59m 59s');
-  });
-
-  it('formats hour+ durations as hours and minutes', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(3600)).toBe('1h 0m');
-    expect(formatDuration(4500)).toBe('1h 15m');
-    expect(formatDuration(7200)).toBe('2h 0m');
-    expect(formatDuration(86399)).toBe('23h 59m');
-  });
-
-  it('formats day+ durations as days and hours', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(86400)).toBe('1d 0h');
-    expect(formatDuration(129600)).toBe('1d 12h');
-    expect(formatDuration(172800)).toBe('2d 0h');
-    expect(formatDuration(302400)).toBe('3d 12h');
-    expect(formatDuration(518400)).toBe('6d 0h');
-  });
-
-  it('formats week+ durations as weeks and days', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(604800)).toBe('1w 0d');
-    expect(formatDuration(907200)).toBe('1w 3d');
-    expect(formatDuration(1209600)).toBe('2w 0d');
-    expect(formatDuration(1814400)).toBe('3w 0d');
-    expect(formatDuration(2419200)).toBe('4w 0d');
-  });
-
-  it('formats month+ durations as months and days', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(2592000)).toBe('1mo 0d');
-    expect(formatDuration(2851200)).toBe('1mo 3d');
-    expect(formatDuration(5184000)).toBe('2mo 0d');
-    expect(formatDuration(7776000)).toBe('3mo 0d');
-    expect(formatDuration(9504000)).toBe('3mo 2w');
-  });
-
-  it('formats year+ durations as years and days', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(31536000)).toBe('1y 0d');
-    expect(formatDuration(31622400)).toBe('1y 1d');
-    expect(formatDuration(34128000)).toBe('1y 1mo');
-    expect(formatDuration(63072000)).toBe('2y 0d');
-  });
-
-  it('handles large multi-year durations', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(15552000)).toBe('6mo 0d');
-    expect(formatDuration(315360000)).toBe('10y 0d');
-  });
-
-  it('rounds correctly for multi-unit durations', async () => {
-    const formatDuration = await importFormatDuration();
-    expect(formatDuration(89.9)).toBe('1m 30s');
-    expect(formatDuration(90.1)).toBe('1m 30s');
-  });
-});

package/extensions/pi-tps/__tests__/helpers.ts

@@ -1,177 +0,0 @@
-import { vi } from 'vitest';
-import type { ExtensionAPI, ExtensionContext } from '@earendil-works/pi-coding-agent';
-import type { AssistantMessage } from '@earendil-works/pi-ai';
-
-// ─── Event types (mirrors extension/index.ts — not exported from pi's public API) ────
-
-export interface TurnStartEvent {
-  type: 'turn_start';
-  turnIndex: number;
-  timestamp: number;
-}
-
-export interface TurnEndEvent {
-  type: 'turn_end';
-  turnIndex: number;
-  message: unknown;
-  toolResults: unknown[];
-}
-
-export interface MessageStartEvent {
-  type: 'message_start';
-  message: unknown;
-}
-
-export interface MessageUpdateEvent {
-  type: 'message_update';
-  message: unknown;
-}
-
-export interface MessageEndEvent {
-  type: 'message_end';
-  message: unknown;
-}
-
-// ─── Helpers ─────────────────────────────────────────────────────────────────
-
-/** Advance real time by ms (for integration-style timing tests) */
-export const tick = (ms = 10) => new Promise((resolve) => setTimeout(resolve, ms));
-
-/** Create a standard AssistantMessage with overridable fields */
-export function makeAssistantMessage(
-  overrides: { output?: number; input?: number; provider?: string; model?: string } = {}
-): AssistantMessage {
-  const { output = 20, input = 10, provider = 'openai', model = 'gpt-4' } = overrides;
-
-  return {
-    role: 'assistant',
-    content: [{ type: 'text' as const, text: 'Hello' }],
-    api: 'openai-completions',
-    provider,
-    model,
-    usage: {
-      input,
-      output,
-      cacheRead: 0,
-      cacheWrite: 0,
-      totalTokens: input + output,
-      cost: {
-        input: 0.001,
-        output: 0.002,
-        cacheRead: 0,
-        cacheWrite: 0,
-        total: 0.003,
-      },
-    },
-    stopReason: 'stop',
-    timestamp: Date.now(),
-  };
-}
-
-// ─── Mock setup ──────────────────────────────────────────────────────────────
-
-export interface TestFixture {
-  mockPi: Partial<ExtensionAPI>;
-  handlers: Record<string, (...args: unknown[]) => void>;
-  commands: Record<string, { handler: (args: string, ctx: any) => Promise<void> }>;
-  notifySpy: ReturnType<typeof vi.fn>;
-  appendEntrySpy: ReturnType<typeof vi.fn>;
-  eventsEmitSpy: ReturnType<typeof vi.fn>;
-  /** Emit an event on the mock event bus, dispatching to registered listeners (and the emit spy). */
-  emitEvent: (event: string, payload: unknown) => void;
-  registerCommandSpy: ReturnType<typeof vi.fn>;
-  mockEntries: Array<{ type?: string; role?: string; customType?: string; data?: unknown }>;
-  mockCtx: ExtensionContext;
-}
-
-/**
- * Create a fresh set of mocks for one test.
- * Call `activateExtension()` on the result to wire up the extension.
- */
-export function createTestFixture(): TestFixture {
-  const handlers: Record<string, (...args: unknown[]) => void> = {};
-  const commands: Record<string, { handler: (args: string, ctx: any) => Promise<void> }> = {};
-  const notifySpy = vi.fn();
-  const appendEntrySpy = vi.fn();
-  const registerCommandSpy = vi.fn((name: string, options: any) => {
-    commands[name] = options;
-  });
-  const mockEntries: Array<{
-    type?: string;
-    role?: string;
-    customType?: string;
-    data?: unknown;
-  }> = [];
-
-  const mockCtx = {
-    hasUI: true,
-    ui: { notify: notifySpy } as any,
-    sessionManager: {
-      getEntries: vi.fn().mockReturnValue(mockEntries),
-      getBranch: vi.fn().mockReturnValue(mockEntries),
-      getSessionId: vi.fn(),
-    },
-    modelRegistry: undefined as any,
-    model: undefined,
-    cwd: '/tmp',
-    isIdle: vi.fn(),
-    signal: undefined,
-    abort: vi.fn(),
-    hasPendingMessages: vi.fn(),
-    shutdown: vi.fn(),
-    getContextUsage: vi.fn(),
-    compact: vi.fn(),
-    getSystemPrompt: vi.fn(),
-  } as any as ExtensionContext;
-
-  const eventsEmitSpy = vi.fn();
-  // Real event bus: listeners registered via events.on are dispatched on emit.
-  // eventsEmitSpy still captures every emit for assertions (e.g. 'tps:telemetry').
-  const eventListeners = new Map<string, ((payload: unknown) => void)[]>();
-  const dispatchEvent = (event: string, payload: unknown) => {
-    eventsEmitSpy(event, payload);
-    for (const listener of eventListeners.get(event) ?? []) {
-      try {
-        listener(payload);
-      } catch {
-        // listener errors shouldn't break emit dispatch in tests
-      }
-    }
-  };
-
-  const mockPi: Partial<ExtensionAPI> = {
-    on: vi.fn((event: string, handler: any) => {
-      handlers[event] = handler;
-      return mockPi as ExtensionAPI;
-    }),
-    appendEntry: appendEntrySpy,
-    registerCommand: registerCommandSpy,
-    events: {
-      emit: dispatchEvent,
-      on: vi.fn((event: string, listener: (payload: unknown) => void) => {
-        const list = eventListeners.get(event) ?? [];
-        list.push(listener);
-        eventListeners.set(event, list);
-      }),
-    } as any,
-  };
-
-  return {
-    mockPi,
-    handlers,
-    commands,
-    notifySpy,
-    appendEntrySpy,
-    eventsEmitSpy,
-    emitEvent: dispatchEvent,
-    registerCommandSpy,
-    mockEntries,
-    mockCtx,
-  };
-}
-
-/** Import the extension module and wire it to the test fixture's mockPi */
-export async function activateExtension(fixture: TestFixture) {
-  const { default: tpsExtension } = await import('../index.js');
-  tpsExtension(fixture.mockPi as ExtensionAPI);
-}