hamster-wheel-cli 0.2.0 → 0.3.1

package/tests/global-config.test.ts

@@ -1,6 +1,15 @@
 import assert from 'node:assert/strict';
 import { test } from 'node:test';
-import { applyShortcutArgv, parseAliasEntries, parseGlobalConfig, splitCommandArgs, updateAliasContent } from '../src/global-config';
+import {
+  applyShortcutArgv,
+  parseAgentEntries,
+  parseAliasEntries,
+  parseGlobalConfig,
+  removeAgentContent,
+  splitCommandArgs,
+  updateAgentContent,
+  updateAliasContent
+} from '../src/global-config';
 
 test('parseGlobalConfig 读取 shortcut 配置', () => {
   const content = `
@@ -101,3 +110,82 @@ command = "--run-e2e"
   assert.ok(result.includes('daily = "--task \\"新命令\\""'));
   assert.ok(result.includes('[shortcut]'));
 });
+
+test('parseAgentEntries 读取 agent 配置', () => {
+  const content = `
+[agent]
+claude = "claude --model sonnet"
+`;
+  const entries = parseAgentEntries(content);
+  assert.deepEqual(entries, [
+    {
+      name: 'claude',
+      command: 'claude --model sonnet'
+    }
+  ]);
+});
+
+test('parseAgentEntries 支持 agents 并忽略重复', () => {
+  const content = `
+[agents]
+daily = "cmd-a"
+
+[agent]
+daily = "cmd-b"
+weekly = "cmd-c"
+`;
+  const entries = parseAgentEntries(content);
+  assert.deepEqual(entries, [
+    {
+      name: 'daily',
+      command: 'cmd-a'
+    },
+    {
+      name: 'weekly',
+      command: 'cmd-c'
+    }
+  ]);
+});
+
+test('updateAgentContent 会在空文件中写入 agent', () => {
+  const result = updateAgentContent('', 'daily', 'claude --model sonnet');
+  assert.ok(result.includes('[agent]'));
+  assert.ok(result.includes('daily = "claude --model sonnet"'));
+  assert.ok(result.endsWith('\n'));
+});
+
+test('updateAgentContent 会更新已有 agent 并保留其他配置', () => {
+  const content = `
+[agent]
+daily = "cmd-a"
+
+[shortcut]
+name = "quick"
+command = "--run-e2e"
+`;
+  const result = updateAgentContent(content, 'daily', 'cmd-b');
+  assert.ok(result.includes('daily = "cmd-b"'));
+  assert.ok(result.includes('[shortcut]'));
+});
+
+test('removeAgentContent 会删除指定 agent', () => {
+  const content = `
+[agent]
+daily = "cmd-a"
+weekly = "cmd-b"
+`;
+  const { removed, nextContent } = removeAgentContent(content, 'daily');
+  assert.equal(removed, true);
+  assert.ok(!nextContent.includes('daily = "cmd-a"'));
+  assert.ok(nextContent.includes('weekly = "cmd-b"'));
+});
+
+test('removeAgentContent 未命中时保持原内容', () => {
+  const content = `
+[agent]
+weekly = "cmd-b"
+`;
+  const { removed, nextContent } = removeAgentContent(content, 'daily');
+  assert.equal(removed, false);
+  assert.ok(nextContent.includes('weekly = "cmd-b"'));
+});

package/tests/webhook.test.ts

@@ -8,17 +8,20 @@ test('normalizeWebhookUrls 会去除空值并裁剪空白', () => {
   assert.deepEqual(urls, ['https://example.com', 'http://a.local']);
 });
 
-test('buildWebhookPayload 会补齐时间戳', () => {
+test('buildWebhookPayload 会补齐本地时间戳', () => {
   const payload = buildWebhookPayload({
     event: 'task_start',
     task: 'demo',
     iteration: 0,
-    stage: '任务开始'
+    stage: '任务开始',
+    project: 'wheel-ai'
   });
   assert.equal(payload.event, 'task_start');
   assert.equal(payload.task, 'demo');
   assert.equal(payload.iteration, 0);
-  assert.ok(payload.timestamp.length > 0);
+  assert.match(payload.timestamp, /^\d{8}-\d{6}$/);
+  assert.equal(payload.commit, '');
+  assert.equal(payload.pr, '');
 });
 
 test('sendWebhookNotifications 在无配置时直接跳过', async () => {
@@ -34,6 +37,7 @@ test('sendWebhookNotifications 在无配置时直接跳过', async () => {
     task: 'demo',
     iteration: 0,
     stage: '任务开始',
+    project: 'wheel-ai',
     timestamp: '2024-01-01T00:00:00.000Z'
   });
 
@@ -55,10 +59,10 @@ test('sendWebhookNotifications 会向多个地址发送', async () => {
 
   const payload = buildWebhookPayload({
     event: 'iteration_start',
-    task: 'demo',
     branch: 'feat/demo',
     iteration: 2,
     stage: '开始第 2 轮迭代',
+    project: 'wheel-ai',
     timestamp: '2024-01-01T00:00:00.000Z'
   });
 
@@ -67,17 +71,17 @@ test('sendWebhookNotifications 会向多个地址发送', async () => {
   assert.equal(calls.length, 2);
   assert.equal(calls[0].contentType, 'application/json');
   const decoded = JSON.parse(calls[0].body) as typeof payload;
-  assert.equal(decoded.task, 'demo');
   assert.equal(decoded.branch, 'feat/demo');
+  assert.ok(!('task' in decoded));
 });
 
 test('sendWebhookNotifications 捕获异常并不中断', async () => {
   const logger = new Logger({ verbose: false });
   const payload = buildWebhookPayload({
     event: 'task_end',
-    task: 'demo',
     iteration: 1,
     stage: '任务结束',
+    project: 'wheel-ai',
     timestamp: '2024-01-01T00:00:00.000Z'
   });