hamster-wheel-cli 0.1.0 → 0.2.0-beta.2

package/tests/branch-name.test.ts

@@ -0,0 +1,28 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { parseBranchName } from '../src/ai';
+
+test('parseBranchName 规范化 JSON 分支名', () => {
+  const output = '{"branch":"feat/Add new_feature"}';
+  assert.equal(parseBranchName(output), 'feat/add-new-feature');
+});
+
+test('parseBranchName 支持类型别名', () => {
+  const output = '{"branch":"feature/new-flow"}';
+  assert.equal(parseBranchName(output), 'feat/new-flow');
+});
+
+test('parseBranchName 支持文本格式', () => {
+  const output = '分支名: fix/bug-123';
+  assert.equal(parseBranchName(output), 'fix/bug-123');
+});
+
+test('parseBranchName 拒绝非法类型', () => {
+  const output = '{"branch":"release/prepare"}';
+  assert.equal(parseBranchName(output), null);
+});
+
+test('parseBranchName 拒绝过短 slug', () => {
+  const output = '{"branch":"fix/a"}';
+  assert.equal(parseBranchName(output), null);
+});

package/tests/e2e/cli.e2e.test.ts

@@ -22,6 +22,7 @@ test('CLI 帮助信息可正常输出', async () => {
   assert.ok(stdout.includes('--skip-install'));
   assert.ok(stdout.includes('--webhook'));
   assert.ok(stdout.includes('--webhook-timeout'));
+  assert.ok(stdout.includes('--multi-task-mode'));
 });
 
 test('CLI monitor 帮助信息可正常输出', async () => {
@@ -35,6 +36,7 @@ test('CLI monitor 帮助信息可正常输出', async () => {
   });
 
   assert.ok(stdout.includes('Usage: wheel-ai monitor'));
+  assert.ok(stdout.includes('t 终止任务'));
 });
 
 test('CLI monitor 在非 TTY 下输出提示', async () => {
@@ -75,3 +77,42 @@ test('CLI logs 在非 TTY 下输出提示', async () => {
 
   assert.ok(stdout.includes('当前终端不支持交互式 logs。'));
 });
+
+test('CLI set alias 帮助信息可正常输出', async () => {
+  const execFileAsync = promisify(execFile);
+  const cliPath = path.join(process.cwd(), 'src', 'cli.ts');
+  const { stdout } = await execFileAsync('node', ['--require', 'ts-node/register', cliPath, 'set', 'alias', '--help'], {
+    env: {
+      ...process.env,
+      FORCE_COLOR: '0'
+    }
+  });
+
+  assert.ok(stdout.includes('Usage: wheel-ai set alias'));
+});
+
+test('CLI alias 帮助信息可正常输出', async () => {
+  const execFileAsync = promisify(execFile);
+  const cliPath = path.join(process.cwd(), 'src', 'cli.ts');
+  const { stdout } = await execFileAsync('node', ['--require', 'ts-node/register', cliPath, 'alias', '--help'], {
+    env: {
+      ...process.env,
+      FORCE_COLOR: '0'
+    }
+  });
+
+  assert.ok(stdout.includes('Usage: wheel-ai alias'));
+});
+
+test('CLI alias 在非 TTY 下输出提示', async () => {
+  const execFileAsync = promisify(execFile);
+  const cliPath = path.join(process.cwd(), 'src', 'cli.ts');
+  const { stdout } = await execFileAsync('node', ['--require', 'ts-node/register', cliPath, 'alias'], {
+    env: {
+      ...process.env,
+      FORCE_COLOR: '0'
+    }
+  });
+
+  assert.ok(stdout.includes('当前终端不支持交互式 alias。'));
+});

package/tests/global-config.test.ts

@@ -1,6 +1,6 @@
 import assert from 'node:assert/strict';
 import { test } from 'node:test';
-import { applyShortcutArgv, parseGlobalConfig, splitCommandArgs } from '../src/global-config';
+import { applyShortcutArgv, parseAliasEntries, parseGlobalConfig, splitCommandArgs, updateAliasContent } from '../src/global-config';
 
 test('parseGlobalConfig 读取 shortcut 配置', () => {
   const content = `
@@ -50,3 +50,54 @@ command = "run --task \\\"demo\\\" --run-tests"
     '--run-e2e'
   ]);
 });
+
+test('parseAliasEntries 读取 alias 并附带 shortcut', () => {
+  const content = `
+[alias]
+daily = "--task \\"补充文档\\""
+weekly = "run --task \\"补充测试\\" --run-tests"
+
+[shortcut]
+name = "quick"
+command = "--run-e2e"
+`;
+  const entries = parseAliasEntries(content);
+  assert.deepEqual(entries, [
+    {
+      name: 'daily',
+      command: '--task "补充文档"',
+      source: 'alias'
+    },
+    {
+      name: 'weekly',
+      command: 'run --task "补充测试" --run-tests',
+      source: 'alias'
+    },
+    {
+      name: 'quick',
+      command: '--run-e2e',
+      source: 'shortcut'
+    }
+  ]);
+});
+
+test('updateAliasContent 会在空文件中写入 alias', () => {
+  const result = updateAliasContent('', 'daily', '--task "补充文档"');
+  assert.ok(result.includes('[alias]'));
+  assert.ok(result.includes('daily = "--task \\"补充文档\\""'));
+  assert.ok(result.endsWith('\n'));
+});
+
+test('updateAliasContent 会更新已有 alias 并保留其他配置', () => {
+  const content = `
+[alias]
+daily = "--task \\"旧命令\\""
+
+[shortcut]
+name = "quick"
+command = "--run-e2e"
+`;
+  const result = updateAliasContent(content, 'daily', '--task "新命令"');
+  assert.ok(result.includes('daily = "--task \\"新命令\\""'));
+  assert.ok(result.includes('[shortcut]'));
+});

package/tests/monitor.test.ts

@@ -0,0 +1,17 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { buildConfirmDialogLines, resolveTerminationTarget } from '../src/monitor';
+
+test('resolveTerminationTarget 在不同平台输出不同目标', () => {
+  assert.equal(resolveTerminationTarget(123, 'win32'), 123);
+  assert.equal(resolveTerminationTarget(123, 'linux'), -123);
+});
+
+test('buildConfirmDialogLines 会生成有限宽度的确认框', () => {
+  const lines = buildConfirmDialogLines('alpha.log', 24);
+  assert.ok(lines.length >= 3);
+  for (const line of lines) {
+    assert.ok(line.length <= 24);
+  }
+  assert.ok(lines.some(line => line.includes('alpha.log')));
+});

package/tests/multi-task.test.ts

@@ -0,0 +1,77 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { buildTaskPlans, normalizeTaskList, parseMultiTaskMode } from '../src/multi-task';
+
+test('parseMultiTaskMode 支持默认与中文别名', () => {
+  assert.equal(parseMultiTaskMode(undefined), 'relay');
+  assert.equal(parseMultiTaskMode(''), 'relay');
+  assert.equal(parseMultiTaskMode('relay'), 'relay');
+  assert.equal(parseMultiTaskMode('串行执行'), 'serial');
+  assert.equal(parseMultiTaskMode('串行执行但是失败也继续'), 'serial-continue');
+  assert.equal(parseMultiTaskMode('并行执行'), 'parallel');
+});
+
+test('parseMultiTaskMode 对未知模式抛错', () => {
+  assert.throws(() => parseMultiTaskMode('unknown-mode'));
+});
+
+test('normalizeTaskList 过滤空白任务', () => {
+  assert.deepEqual(normalizeTaskList(undefined), []);
+  assert.deepEqual(normalizeTaskList('  '), []);
+  assert.deepEqual(normalizeTaskList('task-a'), ['task-a']);
+  assert.deepEqual(normalizeTaskList(['task-a', '  ', 'task-b ']), ['task-a', 'task-b']);
+});
+
+test('buildTaskPlans 在接力模式下使用上一任务分支作为基线', () => {
+  const plans = buildTaskPlans({
+    tasks: ['task-a', 'task-b', 'task-c'],
+    mode: 'relay',
+    useWorktree: true,
+    baseBranch: 'main',
+    branchInput: 'feat/demo',
+    worktreePath: '/tmp/wt/demo',
+    logFile: '/tmp/logs/demo.log'
+  });
+
+  assert.equal(plans.length, 3);
+  assert.equal(plans[0].branchName, 'feat/demo');
+  assert.equal(plans[1].branchName, 'feat/demo-2');
+  assert.equal(plans[2].branchName, 'feat/demo-3');
+  assert.equal(plans[0].baseBranch, 'main');
+  assert.equal(plans[1].baseBranch, 'feat/demo');
+  assert.equal(plans[2].baseBranch, 'feat/demo-2');
+  assert.equal(plans[0].worktreePath, '/tmp/wt/demo');
+  assert.equal(plans[1].worktreePath, '/tmp/wt/demo-task-2');
+  assert.equal(plans[2].worktreePath, '/tmp/wt/demo-task-3');
+  assert.equal(plans[0].logFile, '/tmp/logs/demo.log');
+  assert.equal(plans[1].logFile, '/tmp/logs/demo-task-2.log');
+  assert.equal(plans[2].logFile, '/tmp/logs/demo-task-3.log');
+});
+
+test('buildTaskPlans 在串行模式下保持基线分支不变', () => {
+  const plans = buildTaskPlans({
+    tasks: ['task-a', 'task-b'],
+    mode: 'serial',
+    useWorktree: true,
+    baseBranch: 'develop',
+    branchInput: 'feat/serial',
+    worktreePath: '/tmp/wt/serial',
+    logFile: '/tmp/logs/serial.log'
+  });
+
+  assert.equal(plans[0].baseBranch, 'develop');
+  assert.equal(plans[1].baseBranch, 'develop');
+});
+
+test('buildTaskPlans 在非 worktree 下保留传入分支名', () => {
+  const plans = buildTaskPlans({
+    tasks: ['task-a', 'task-b'],
+    mode: 'serial',
+    useWorktree: false,
+    baseBranch: 'main',
+    branchInput: 'feat/no-worktree'
+  });
+
+  assert.equal(plans[0].branchName, 'feat/no-worktree');
+  assert.equal(plans[1].branchName, 'feat/no-worktree');
+});