badgr-cli 1.0.36 → 1.0.37

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "badgr-cli",
-  "version": "1.0.36",
+  "version": "1.0.37",
   "description": "Badgr, run or serve GPU workloads from one command",
   "type": "module",
   "bin": {

package/src/commands/down.js

@@ -2,6 +2,15 @@ import { requireApiKey } from '../config.js';
 import { findDeployment, removeDeployment, addReceipt, generateReceiptId } from '../store.js';
 import { terminateDeployment, listDeployments } from '../api.js';
 
+function formatRuntime(minutes) {
+  if (minutes < 60) return `${minutes}m`;
+  const days = Math.floor(minutes / 1440);
+  const hours = Math.floor((minutes % 1440) / 60);
+  const mins = minutes % 60;
+  if (days > 0) return `${days}d ${hours}h ${mins}m`;
+  return `${hours}h ${mins}m`;
+}
+
 export async function downCommand(config, args, chalk) {
   const hasAll = args.includes('--all');
   const hasYes = args.includes('--yes') || args.includes('-y');
@@ -60,7 +69,7 @@ export async function downCommand(config, args, chalk) {
 
   console.log(chalk.green('\n✓ Stopped'));
   console.log(chalk.green('  Billing ended\n'));
-  console.log(`  ${chalk.bold('Runtime:')}    ${runtimeMin}m`);
+  console.log(`  ${chalk.bold('Runtime:')}    ${formatRuntime(runtimeMin)}`);
   if (finalCost > 0) console.log(`  ${chalk.bold('Final cost:')} $${finalCost.toFixed(4)}`);
   console.log(`  ${chalk.bold('Receipt:')}    ${chalk.dim(rcptId)}\n`);
 }

package/tests/down.test.js

@@ -0,0 +1,128 @@
+/**
+ * badgr down — receipt display and runtime formatting tests
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { downCommand } from '../src/commands/down.js';
+
+vi.mock('../src/api.js', () => ({
+  terminateDeployment: vi.fn(),
+  listDeployments:     vi.fn(),
+}));
+
+vi.mock('../src/store.js', () => ({
+  findDeployment:    vi.fn(() => null),
+  removeDeployment:  vi.fn(),
+  addReceipt:        vi.fn(),
+  generateReceiptId: vi.fn(() => 'rcpt-test-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,
+};
+
+function makeStoppedDep(runtimeSeconds, costPerHour = 1.00) {
+  const now = Date.now() / 1000;
+  return {
+    deployment_id: 'dep-test-001',
+    gpu_type:      'L40S',
+    cost_per_hour: costPerHour,
+    stopped_at:    now,
+    started_at:    now - runtimeSeconds,
+  };
+}
+
+beforeEach(() => {
+  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-test-001');
+});
+
+afterEach(() => {
+  vi.restoreAllMocks();
+});
+
+describe('runtime display formatting', () => {
+  it('shows minutes only for short runs (< 1h)', async () => {
+    api.terminateDeployment.mockResolvedValue(makeStoppedDep(27 * 60, 0.34));
+    const lines = [];
+    console.log.mockImplementation((...args) => lines.push(args.join(' ')));
+
+    await downCommand(config, ['dep-test-001'], chalk);
+
+    const runtimeLine = lines.find(l => l.includes('Runtime:'));
+    expect(runtimeLine).toMatch(/27m/);
+    expect(runtimeLine).not.toMatch(/\dh/);
+  });
+
+  it('shows hours and minutes for runs >= 1h', async () => {
+    api.terminateDeployment.mockResolvedValue(makeStoppedDep(4091 * 60, 1.00));
+    const lines = [];
+    console.log.mockImplementation((...args) => lines.push(args.join(' ')));
+
+    await downCommand(config, ['dep-test-001'], chalk);
+
+    const runtimeLine = lines.find(l => l.includes('Runtime:'));
+    expect(runtimeLine).toMatch(/2d 20h 11m/);
+    expect(runtimeLine).not.toMatch(/4091m/);
+  });
+
+  it('shows days, hours, minutes for multi-day runs', async () => {
+    api.terminateDeployment.mockResolvedValue(makeStoppedDep(5722 * 60, 0.46));
+    const lines = [];
+    console.log.mockImplementation((...args) => lines.push(args.join(' ')));
+
+    await downCommand(config, ['dep-test-001'], chalk);
+
+    const runtimeLine = lines.find(l => l.includes('Runtime:'));
+    expect(runtimeLine).toMatch(/3d 23h 22m/);
+    expect(runtimeLine).not.toMatch(/5722m/);
+  });
+});
+
+describe('receipt is recorded', () => {
+  it('adds a receipt with correct fields on stop', async () => {
+    api.terminateDeployment.mockResolvedValue(makeStoppedDep(27 * 60, 0.34));
+
+    await downCommand(config, ['dep-test-001'], chalk);
+
+    expect(store.addReceipt).toHaveBeenCalledWith(expect.objectContaining({
+      receiptId:    'rcpt-test-001',
+      deploymentId: 'dep-test-001',
+      gpu:          'L40S',
+      status:       'terminated',
+    }));
+    const call = store.addReceipt.mock.calls[0][0];
+    expect(call.runtimeSeconds).toBeCloseTo(27 * 60, -1);
+    expect(call.finalCost).toBeGreaterThan(0);
+  });
+});
+
+describe('error handling', () => {
+  it('prints error and returns when terminateDeployment throws', async () => {
+    api.terminateDeployment.mockRejectedValue(new Error('network error'));
+    const errLines = [];
+    console.error.mockImplementation(msg => errLines.push(msg));
+
+    await downCommand(config, ['dep-test-001'], chalk);
+
+    expect(errLines.join('\n')).toMatch(/network error|Could not stop/i);
+    expect(store.addReceipt).not.toHaveBeenCalled();
+  });
+});