badgr-cli 1.1.0 → 1.1.2

package/tests/heartbeat.test.js

@@ -1,70 +0,0 @@
-/**
- * badgr heartbeat — resets an endpoint's idle-timeout clock
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { heartbeatCommand } from '../src/commands/heartbeat.js';
-
-vi.mock('../src/api.js', () => ({
-  heartbeatDeployment: vi.fn(),
-}));
-
-vi.mock('../src/store.js', () => ({
-  findDeployment: vi.fn(() => null),
-}));
-
-vi.mock('../src/config.js', () => ({
-  requireApiKey: vi.fn(),
-}));
-
-import * as api   from '../src/api.js';
-import * as store from '../src/store.js';
-
-const config = { apiKey: 'sk-test', baseUrl: 'https://api.test/v1' };
-
-const chalk = {
-  bold:   s => s,
-  dim:    s => s,
-  red:    s => s,
-  yellow: s => s,
-  green:  s => s,
-  cyan:   s => s,
-};
-
-beforeEach(() => {
-  process.exitCode = undefined;
-  vi.spyOn(console, 'log').mockImplementation(() => {});
-  vi.spyOn(console, 'error').mockImplementation(() => {});
-  vi.resetAllMocks();
-  store.findDeployment.mockReturnValue(null);
-});
-
-afterEach(() => {
-  vi.restoreAllMocks();
-  process.exitCode = undefined;
-});
-
-it('requires a deployment id', async () => {
-  await heartbeatCommand(config, [], chalk);
-  expect(api.heartbeatDeployment).not.toHaveBeenCalled();
-});
-
-it('calls heartbeatDeployment and prints confirmation', async () => {
-  api.heartbeatDeployment.mockResolvedValue({ deployment_id: 'dep-abc123', last_activity_at: 1700000000 });
-
-  const lines = [];
-  console.log.mockImplementation((...args) => lines.push(args.join(' ')));
-
-  await heartbeatCommand(config, ['dep-abc123'], chalk);
-
-  expect(api.heartbeatDeployment).toHaveBeenCalledWith(config, 'dep-abc123');
-  expect(lines.some(l => l.includes('Heartbeat recorded for dep-abc123'))).toBe(true);
-  expect(process.exitCode).toBeFalsy();
-});
-
-it('reports an error and sets exitCode on failure', async () => {
-  api.heartbeatDeployment.mockRejectedValue(new Error('deployment not found'));
-
-  await heartbeatCommand(config, ['dep-missing'], chalk);
-
-  expect(process.exitCode).toBe(1);
-});

package/tests/job-progress-poll.test.js

@@ -1,136 +0,0 @@
-import { describe, it, expect, vi } from 'vitest';
-import { pollJobUntilTerminal, renderJobClosingBlock } from '../src/progress.js';
-
-const chalk = new Proxy({}, { get: () => (s) => s });
-const config = { apiKey: 'k', baseUrl: 'https://api.test' };
-
-function silenceStdout() {
-  return vi.spyOn(process.stdout, 'write').mockImplementation(() => true);
-}
-
-describe('pollJobUntilTerminal', () => {
-  it('returns immediately once the job reaches a terminal status', async () => {
-    const writeSpy = silenceStdout();
-    const detail = { job_id: 'job_1', status: 'completed', stage_label: 'Complete', health: 'completed' };
-    const callApi = vi.fn().mockResolvedValue(detail);
-
-    const result = await pollJobUntilTerminal(callApi, config, 'job_1', { chalk, maxMs: 60_000, pollMs: 1 });
-
-    expect(result.outcome).toBe('completed');
-    expect(result.detail).toBe(detail);
-    writeSpy.mockRestore();
-  });
-
-  it('surfaces progress fields (stage/current/total) while polling', async () => {
-    const writeSpy = silenceStdout();
-    const callApi = vi.fn()
-      .mockResolvedValueOnce({ job_id: 'job_2', status: 'running', stage_label: 'Generating images', progress_current: 3, progress_total: 10, progress_unit: 'images', health: 'progressing' })
-      .mockResolvedValueOnce({ job_id: 'job_2', status: 'completed', stage_label: 'Complete', health: 'completed' });
-
-    const result = await pollJobUntilTerminal(callApi, config, 'job_2', { chalk, maxMs: 60_000, pollMs: 1 });
-
-    expect(result.outcome).toBe('completed');
-    const output = writeSpy.mock.calls.map(c => c[0]).join('');
-    expect(output).toContain('3/10 images');
-    writeSpy.mockRestore();
-  });
-
-  it('does not swallow repeated poll failures silently — gives up and reports polling_failed', async () => {
-    const writeSpy = silenceStdout();
-    const callApi = vi.fn().mockRejectedValue(new Error('network down'));
-
-    const result = await pollJobUntilTerminal(callApi, config, 'job_3', { chalk, maxMs: 60_000, pollMs: 1 });
-
-    expect(result.outcome).toBe('polling_failed');
-    // Must have warned the user at least once instead of looping silently.
-    const output = writeSpy.mock.calls.map(c => c[0]).join('');
-    expect(output).toMatch(/Lost contact/i);
-    writeSpy.mockRestore();
-  });
-
-  it('announces a retry-on-a-different-route once, with teardown/billing status', async () => {
-    const writeSpy = silenceStdout();
-    const callApi = vi.fn()
-      .mockResolvedValueOnce({
-        job_id: 'job_retry', status: 'running', stage: 'retrying_different_route',
-        stage_label: 'Retrying on a different route', health: 'progressing',
-        progress_message: 'Retrying on a different route... attempt 2/2',
-        teardown_status: 'ok',
-      })
-      .mockResolvedValueOnce({
-        job_id: 'job_retry', status: 'completed', stage_label: 'Complete', health: 'completed',
-      });
-
-    const result = await pollJobUntilTerminal(callApi, config, 'job_retry', { chalk, maxMs: 60_000, pollMs: 1 });
-
-    expect(result.outcome).toBe('completed');
-    const output = writeSpy.mock.calls.map(c => c[0]).join('');
-    expect(output).toContain('Retrying on a different route... attempt 2/2');
-    expect(output).toContain('Previous attempt teardown: succeeded');
-    expect(output).toContain('Billing: stopped');
-    writeSpy.mockRestore();
-  });
-
-  it('recovers from transient poll failures once the API responds again', async () => {
-    const writeSpy = silenceStdout();
-    const callApi = vi.fn()
-      .mockRejectedValueOnce(new Error('timeout'))
-      .mockRejectedValueOnce(new Error('timeout'))
-      .mockResolvedValueOnce({ job_id: 'job_4', status: 'completed', stage_label: 'Complete', health: 'completed' });
-
-    const result = await pollJobUntilTerminal(callApi, config, 'job_4', { chalk, maxMs: 60_000, pollMs: 1 });
-
-    expect(result.outcome).toBe('completed');
-    writeSpy.mockRestore();
-  });
-});
-
-describe('renderJobClosingBlock', () => {
-  it('shows teardown/billing/logs/receipt for a completed job', () => {
-    const block = renderJobClosingBlock(chalk, {
-      job_id: 'job_5', status: 'completed', elapsed_seconds: 252,
-      charged_usd: 0.08, teardown_status: 'ok', billing_status: 'stopped',
-    }, 'rcpt-abc');
-
-    expect(block).toContain('Complete');
-    expect(block).toContain('Teardown: succeeded');
-    expect(block).toContain('Billing: stopped');
-    expect(block).toContain('badgr logs job_5');
-    expect(block).toContain('badgr receipts rcpt-abc');
-  });
-
-  it('shows failure class, stage, and next action for a failed job', () => {
-    const block = renderJobClosingBlock(chalk, {
-      job_id: 'job_6', status: 'failed', elapsed_seconds: 1112,
-      error: { code: 'comfy_health_failed', message: 'ComfyUI did not start within 10 minutes.' },
-      failure_class: 'health_check_failed', stage: 'comfy_waiting_for_health',
-      teardown_status: 'ok', billing_status: 'stopped',
-      next_action: 'Run `badgr logs <job_id>` or retry with a longer max-runtime.',
-    });
-
-    expect(block).toContain('Failed');
-    expect(block).toContain('Class: health_check_failed');
-    expect(block).toContain('Stage: comfy_waiting_for_health');
-    expect(block).toContain('Next:');
-  });
-
-  it('labels a runtime/spend cap failure as "Stopped — limit reached", not a crash', () => {
-    const block = renderJobClosingBlock(chalk, {
-      job_id: 'job_7', status: 'failed', elapsed_seconds: 545,
-      error: { code: 'max_runtime_reached', message: 'Job reached max runtime and was stopped. Billing ended.' },
-      failure_class: 'runtime_cap_reached',
-      teardown_status: 'ok', billing_status: 'stopped',
-    });
-
-    expect(block).toContain('Stopped — limit reached');
-  });
-
-  it('shows cancelled block distinctly from failed', () => {
-    const block = renderJobClosingBlock(chalk, {
-      job_id: 'job_8', status: 'canceled', elapsed_seconds: 164,
-      teardown_status: 'ok', billing_status: 'stopped',
-    });
-
-    expect(block).toContain('Cancelled by user');
-  });
-});

package/tests/launch-command-argv.test.js

@@ -1,93 +0,0 @@
-/**
- * badgr launch — the task must reach the agent as a real argument array
- * (opts.cmdArgv), never a shell string. Covers the exact tricky-task list
- * from the "fix multi-word tasks permanently" spec: spaces, apostrophes,
- * embedded quotes, `--`, Unicode, and shell metacharacters must all survive
- * as exactly one array element, since there's no shell in this path at all.
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-vi.mock('../src/credentials.js', () => ({
-  getCredential: vi.fn(() => 'stored-credential'),
-  setCredential: vi.fn(),
-  PROVIDER_ENV_KEYS: { anthropic: 'ANTHROPIC_API_KEY', openai: 'OPENAI_API_KEY' },
-}));
-
-const runCommandMock = vi.fn();
-vi.mock('../src/commands/run.js', async (importOriginal) => {
-  const actual = await importOriginal();
-  return { ...actual, runCommand: (...args) => runCommandMock(...args) };
-});
-
-const { launchCommand } = await import('../src/commands/launch.js');
-
-const chalk = { red: s => s, dim: s => s, green: s => s, bold: s => s, yellow: s => s, cyan: s => s };
-const config = { apiKey: 'test-key', baseUrl: 'https://example.test/v1' };
-
-const TRICKY_TASKS = [
-  'Fix the checkout bug',
-  "Fix John's checkout bug",
-  'Fix "checkout" validation',
-  'Fix checkout -- do not modify billing',
-  'Fix café checkout',
-  'Fix $(echo dangerous)',
-  'Fix checkout; rm -rf something',
-];
-
-beforeEach(() => {
-  process.exitCode = undefined;
-  vi.spyOn(console, 'log').mockImplementation(() => {});
-  vi.spyOn(console, 'error').mockImplementation(() => {});
-  runCommandMock.mockClear();
-});
-
-afterEach(() => {
-  vi.restoreAllMocks();
-  process.exitCode = undefined;
-});
-
-describe('badgr launch <agent> "<task>" — task transport', () => {
-  for (const task of TRICKY_TASKS) {
-    it(`claude: passes ${JSON.stringify(task)} through as one argv element, not a shell string`, async () => {
-      await launchCommand(config, ['claude', task], chalk);
-
-      expect(runCommandMock).toHaveBeenCalledTimes(1);
-      const [, , , opts] = runCommandMock.mock.calls[0];
-      expect(opts.cmdArgv).toEqual(['claude', '-p', task]);
-    });
-
-    it(`codex: passes ${JSON.stringify(task)} through as one argv element, not a shell string`, async () => {
-      await launchCommand(config, ['codex', task], chalk);
-
-      const [, , , opts] = runCommandMock.mock.calls[0];
-      expect(opts.cmdArgv).toEqual(['badgr-codex-run', task]);
-    });
-
-    it(`cline: passes ${JSON.stringify(task)} through as one argv element, not a shell string`, async () => {
-      await launchCommand(config, ['cline', task], chalk);
-
-      const [, , , opts] = runCommandMock.mock.calls[0];
-      expect(opts.cmdArgv).toEqual(['badgr-cline-run', task]);
-    });
-  }
-
-  it('single-word tasks continue working', async () => {
-    await launchCommand(config, ['claude', 'hello'], chalk);
-    const [, , , opts] = runCommandMock.mock.calls[0];
-    expect(opts.cmdArgv).toEqual(['claude', '-p', 'hello']);
-  });
-
-  it('explicit `-- <command>` form passes the real array through, not a joined string', async () => {
-    await launchCommand(config, ['.', '--', 'codex', 'exec', 'Fix the checkout bug'], chalk);
-
-    expect(runCommandMock).toHaveBeenCalledTimes(1);
-    const [, , , opts] = runCommandMock.mock.calls[0];
-    expect(opts.cmdArgv).toEqual(['codex', 'exec', 'Fix the checkout bug']);
-  });
-
-  it('playwright (no natural-language task) still launches via an argv array', async () => {
-    await launchCommand(config, ['playwright'], chalk);
-    const [, , , opts] = runCommandMock.mock.calls[0];
-    expect(opts.cmdArgv).toEqual(['npx', 'playwright', 'test', '--reporter=line,html', '--trace=retain-on-failure']);
-  });
-});