badgr-cli 1.1.0 → 1.1.2

package/tests/rerun.test.js

@@ -1,94 +0,0 @@
-/**
- * badgr rerun — replays a past job or endpoint with its exact original spec
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { rerunCommand } from '../src/commands/rerun.js';
-
-vi.mock('../src/api.js', () => ({
-  rerunDeployment: vi.fn(),
-}));
-
-vi.mock('../src/store.js', () => ({
-  findDeployment:    vi.fn(() => null),
-  addDeployment:     vi.fn(),
-  addReceipt:        vi.fn(),
-  generateReceiptId: vi.fn(() => 'rcpt-rerun-001'),
-}));
-
-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.spyOn(process.stdout, 'write').mockImplementation(() => true);
-  vi.resetAllMocks();
-  store.findDeployment.mockReturnValue(null);
-  store.generateReceiptId.mockReturnValue('rcpt-rerun-001');
-});
-
-afterEach(() => {
-  vi.restoreAllMocks();
-  process.exitCode = undefined;
-});
-
-it('requires a deployment id', async () => {
-  await rerunCommand(config, [], chalk);
-  expect(api.rerunDeployment).not.toHaveBeenCalled();
-});
-
-it('calls rerunDeployment and prints the new deployment id, never removing the source', async () => {
-  api.rerunDeployment.mockResolvedValue({
-    deployment_id: 'dep-new-002',
-    workload_type: 'job',
-    gpu_type: 'A100',
-    gpu_count: 1,
-    status: 'provisioning',
-    rerun_of: 'dep-old-001',
-    cost_per_hour: 1.9,
-  });
-
-  const lines = [];
-  console.log.mockImplementation((...args) => lines.push(args.join(' ')));
-
-  await rerunCommand(config, ['dep-old-001'], chalk);
-
-  expect(api.rerunDeployment).toHaveBeenCalledWith(config, 'dep-old-001');
-  expect(store.addDeployment).toHaveBeenCalledWith(expect.objectContaining({ id: 'dep-new-002' }));
-  expect(store.addReceipt).toHaveBeenCalledWith(expect.objectContaining({ rerunOf: 'dep-old-001' }));
-  expect(lines.some(l => l.includes('dep-new-002'))).toBe(true);
-  expect(process.exitCode).toBeFalsy();
-});
-
-it('resolves a local name/alias to its deployment id before calling the API', async () => {
-  store.findDeployment.mockReturnValue({ id: 'dep-resolved-999' });
-  api.rerunDeployment.mockResolvedValue({ deployment_id: 'dep-new-003', workload_type: 'job', status: 'provisioning' });
-
-  await rerunCommand(config, ['my-job-alias'], chalk);
-
-  expect(api.rerunDeployment).toHaveBeenCalledWith(config, 'dep-resolved-999');
-});
-
-it('reports an error and sets exitCode on failure', async () => {
-  api.rerunDeployment.mockRejectedValue(new Error('deployment not found'));
-
-  await rerunCommand(config, ['dep-missing'], chalk);
-
-  expect(process.exitCode).toBe(1);
-});

package/tests/restart.test.js

@@ -1,88 +0,0 @@
-/**
- * badgr restart — relaunches an endpoint with the same config and API key
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { restartCommand } from '../src/commands/restart.js';
-
-vi.mock('../src/api.js', () => ({
-  restartDeployment: vi.fn(),
-}));
-
-vi.mock('../src/store.js', () => ({
-  findDeployment:    vi.fn(() => null),
-  removeDeployment:  vi.fn(),
-  addDeployment:     vi.fn(),
-  addReceipt:        vi.fn(),
-  generateReceiptId: vi.fn(() => 'rcpt-restart-001'),
-}));
-
-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.spyOn(process.stdout, 'write').mockImplementation(() => true);
-  vi.resetAllMocks();
-  store.findDeployment.mockReturnValue(null);
-  store.generateReceiptId.mockReturnValue('rcpt-restart-001');
-});
-
-afterEach(() => {
-  vi.restoreAllMocks();
-  process.exitCode = undefined;
-});
-
-it('requires a deployment id', async () => {
-  await restartCommand(config, [], chalk);
-  expect(api.restartDeployment).not.toHaveBeenCalled();
-});
-
-it('calls restartDeployment and prints the new deployment id and URL', async () => {
-  api.restartDeployment.mockResolvedValue({
-    deployment_id: 'dep-new-002',
-    workload_type: 'endpoint',
-    model: 'Qwen/Qwen2.5-7B-Instruct',
-    gpu_type: 'L40S',
-    gpu_count: 1,
-    status: 'provisioning',
-    endpoint_url: 'https://dep-new-002.aibadgr.com/v1',
-    cost_per_hour: 1.2,
-  });
-
-  const lines = [];
-  console.log.mockImplementation((...args) => lines.push(args.join(' ')));
-
-  await restartCommand(config, ['dep-old-001'], chalk);
-
-  expect(api.restartDeployment).toHaveBeenCalledWith(config, 'dep-old-001');
-  expect(store.removeDeployment).toHaveBeenCalledWith('dep-old-001');
-  expect(store.addDeployment).toHaveBeenCalledWith(expect.objectContaining({ id: 'dep-new-002' }));
-  expect(lines.some(l => l.includes('dep-new-002'))).toBe(true);
-  expect(lines.some(l => l.includes('https://dep-new-002.aibadgr.com/v1'))).toBe(true);
-  expect(process.exitCode).toBeFalsy();
-});
-
-it('reports an error and sets exitCode on failure', async () => {
-  api.restartDeployment.mockRejectedValue(new Error('deployment not found'));
-
-  await restartCommand(config, ['dep-missing'], chalk);
-
-  expect(process.exitCode).toBe(1);
-});

package/tests/router.test.js

@@ -1,98 +0,0 @@
-import { describe, it, expect } from 'vitest';
-import {
-  findById, findByCanonical, findCheapest, listAll,
-  estimateCost,
-  GPU_CATALOG,
-} from '../src/router.js';
-
-// getRoutePlan and PROVIDER_CATALOG have been removed from the CLI.
-// Provider routing is now server-side only. Tests below cover the
-// remaining local GPU catalog helpers.
-
-describe('findById', () => {
-  it('finds GPU by id', () => {
-    expect(findById('rtx-4090').name).toContain('4090');
-  });
-
-  it('returns null for unknown id', () => {
-    expect(findById('rtx-9999')).toBeNull();
-  });
-});
-
-describe('findByCanonical', () => {
-  it('finds GPU by canonical name', () => {
-    expect(findByCanonical('RTX_4090').id).toBe('rtx-4090');
-  });
-
-  it('returns null for unknown canonical', () => {
-    expect(findByCanonical('RTX_9999')).toBeNull();
-  });
-});
-
-describe('findCheapest', () => {
-  it('returns a GPU with no requirements', () => {
-    expect(findCheapest()).not.toBeNull();
-  });
-
-  it('returns the cheapest overall', () => {
-    const gpu = findCheapest();
-    const allRates = GPU_CATALOG.map(g => g.ratePerHour);
-    expect(gpu.ratePerHour).toBe(Math.min(...allRates));
-  });
-
-  it('filters by minVramGb', () => {
-    const gpu = findCheapest({ minVramGb: 40 });
-    expect(gpu.vramGb).toBeGreaterThanOrEqual(40);
-  });
-
-  it('filters by tag', () => {
-    const gpu = findCheapest({ tag: 'large-model' });
-    expect(gpu.tags).toContain('large-model');
-  });
-
-  it('returns null when no match', () => {
-    expect(findCheapest({ minVramGb: 999 })).toBeNull();
-  });
-});
-
-describe('listAll', () => {
-  it('returns all GPUs sorted by rate ascending', () => {
-    const gpus = listAll();
-    expect(gpus.length).toBe(GPU_CATALOG.length);
-    for (let i = 1; i < gpus.length; i++) {
-      expect(gpus[i].ratePerHour).toBeGreaterThanOrEqual(gpus[i - 1].ratePerHour);
-    }
-  });
-
-  it('does not mutate the original catalog', () => {
-    const before = GPU_CATALOG.map(g => g.ratePerHour);
-    listAll();
-    expect(GPU_CATALOG.map(g => g.ratePerHour)).toEqual(before);
-  });
-});
-
-describe('estimateCost', () => {
-  it('calculates cost for 60 minutes = 1 hour', () => {
-    expect(estimateCost(1.10, 60)).toBeCloseTo(1.10, 5);
-  });
-
-  it('calculates cost for 30 minutes = half rate', () => {
-    expect(estimateCost(2.00, 30)).toBeCloseTo(1.00, 5);
-  });
-
-  it('returns 0 for 0 minutes', () => {
-    expect(estimateCost(3.50, 0)).toBe(0);
-  });
-});
-
-describe('L40S', () => {
-  it('exists in GPU_CATALOG', () => {
-    expect(findById('l40s')).not.toBeNull();
-    expect(findByCanonical('L40S').vramGb).toBe(48);
-  });
-
-  it('has no PROVIDER_CATALOG (provider routing is server-side)', async () => {
-    const mod = await import('../src/router.js');
-    expect(mod.PROVIDER_CATALOG).toBeUndefined();
-  });
-});