@monotykamary/pi-tps 1.0.0

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

@@ -0,0 +1,83 @@
+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

@@ -0,0 +1,154 @@
+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>;
+  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(),
+      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();
+
+  const mockPi: Partial<ExtensionAPI> = {
+    on: vi.fn((event: string, handler: (...args: unknown[]) => void) => {
+      handlers[event] = handler;
+      return mockPi as ExtensionAPI;
+    }),
+    appendEntry: appendEntrySpy,
+    registerCommand: registerCommandSpy,
+    events: { emit: eventsEmitSpy, on: vi.fn() },
+  };
+
+  return {
+    mockPi,
+    handlers,
+    commands,
+    notifySpy,
+    appendEntrySpy,
+    eventsEmitSpy,
+    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);
+}