@dsh-overdrive/gateway 0.1.3 → 0.1.4

package/test/asr.test.ts

@@ -0,0 +1,77 @@
+import { afterEach, describe, expect, it, vi } from 'vitest';
+import { createTranscriber, extensionForMime } from '../src/asr.js';
+
+describe('extensionForMime', () => {
+  it('maps known audio mimes', () => {
+    expect(extensionForMime('audio/ogg')).toBe('ogg');
+    expect(extensionForMime('audio/mpeg')).toBe('mp3');
+    expect(extensionForMime('audio/mp4')).toBe('m4a');
+    expect(extensionForMime('audio/wav')).toBe('wav');
+    expect(extensionForMime('audio/webm')).toBe('webm');
+  });
+  it('ignores parameters and case', () => {
+    expect(extensionForMime('audio/OGG; codecs=opus')).toBe('ogg');
+  });
+  it('falls back for unknown or missing mime', () => {
+    expect(extensionForMime('application/x-foo')).toBe('oga');
+    expect(extensionForMime(undefined)).toBe('oga');
+    expect(extensionForMime('')).toBe('oga');
+  });
+});
+
+describe('createTranscriber.transcribe', () => {
+  afterEach(() => vi.unstubAllGlobals());
+
+  it('未配置 API key 时禁用且不发起请求', async () => {
+    const fetchMock = vi.fn()
+    vi.stubGlobal('fetch', fetchMock)
+    const asr = createTranscriber({})
+    expect(asr.enabled).toBe(false)
+    expect(await asr.transcribe({ url: 'https://x/v.ogg' })).toBeNull()
+    expect(fetchMock).not.toHaveBeenCalled()
+  })
+
+  it('下载音频并调用 OpenAI 兼容转写端点，返回文本', async () => {
+    const fetchMock = vi.fn()
+      .mockResolvedValueOnce({ ok: true, blob: async () => new Blob(['fake-audio']) })
+      .mockResolvedValueOnce({ ok: true, json: async () => ({ text: '  你好世界  ' }) })
+    vi.stubGlobal('fetch', fetchMock)
+    const asr = createTranscriber({ apiKey: 'sk-test', baseUrl: 'https://api.siliconflow.cn/v1', model: 'whisper-1' })
+    expect(asr.enabled).toBe(true)
+    const text = await asr.transcribe({ url: 'https://x/v.ogg', mime: 'audio/ogg' })
+    expect(text).toBe('你好世界')
+
+    // 第一次 fetch：下载音频；第二次：转写端点
+    expect(fetchMock.mock.calls[0][0]).toBe('https://x/v.ogg')
+    const [url, init] = fetchMock.mock.calls[1]
+    expect(String(url)).toBe('https://api.siliconflow.cn/v1/audio/transcriptions')
+    expect(init.method).toBe('POST')
+    expect(init.headers.authorization).toBe('Bearer sk-test')
+    expect(init.body).toBeInstanceOf(FormData)
+    const filename = (init.body as FormData).get('file') as File
+    expect(filename.name).toBe('voice.ogg')
+  })
+
+  it('下载失败返回 null（不抛错）', async () => {
+    vi.stubGlobal('fetch', vi.fn().mockResolvedValueOnce({ ok: false, status: 403 }))
+    const asr = createTranscriber({ apiKey: 'sk-test' })
+    expect(await asr.transcribe({ url: 'https://x/v.ogg' })).toBeNull()
+  })
+
+  it('转写端点失败返回 null', async () => {
+    const fetchMock = vi.fn()
+      .mockResolvedValueOnce({ ok: true, blob: async () => new Blob(['x']) })
+      .mockResolvedValueOnce({ ok: false, status: 500, text: async () => 'server error' })
+    vi.stubGlobal('fetch', fetchMock)
+    const asr = createTranscriber({ apiKey: 'sk-test' })
+    expect(await asr.transcribe({ url: 'https://x/v.ogg' })).toBeNull()
+  })
+
+  it('无 url 返回 null', async () => {
+    const fetchMock = vi.fn()
+    vi.stubGlobal('fetch', fetchMock)
+    const asr = createTranscriber({ apiKey: 'sk-test' })
+    expect(await asr.transcribe({ mime: 'audio/ogg' })).toBeNull()
+    expect(fetchMock).not.toHaveBeenCalled()
+  })
+})

package/test/commands.test.ts

@@ -10,9 +10,15 @@ describe('parseCommand', () => {
     expect(parseCommand('/task 调研竞品')).toEqual({ kind: 'task', prompt: '调研竞品' });
     expect(parseCommand('/cron 0 8 * * * 每日汇报')).toEqual({ kind: 'cron', schedule: '0 8 * * *', prompt: '每日汇报' });
   });
+  it('识别 /crons 与 /cronrm', () => {
+    expect(parseCommand('/crons')).toEqual({ kind: 'crons' });
+    expect(parseCommand('/cronrm cron-1')).toEqual({ kind: 'cronrm', taskId: 'cron-1' });
+    expect(parseCommand('/cronrm cron-123-abc')).toEqual({ kind: 'cronrm', taskId: 'cron-123-abc' });
+  });
   it('非命令返回 null', () => {
     expect(parseCommand('hello')).toBeNull();
     expect(parseCommand('/unknown')).toBeNull();
     expect(parseCommand('/task')).toBeNull(); // 缺参数
+    expect(parseCommand('/cronrm')).toBeNull(); // 缺任务 id
   });
 });

package/test/pending-buttons.test.ts

@@ -0,0 +1,61 @@
+import { describe, expect, it, vi } from 'vitest';
+import { PendingButtons, PENDING_BUTTONS_TTL_MS } from '../src/pending-buttons.js';
+
+describe('PendingButtons', () => {
+  it('数字回复命中并消费', () => {
+    const pb = new PendingButtons();
+    pb.set('chat-1', [
+      { id: 'approve:1', label: '同意' },
+      { id: 'reject:1', label: '拒绝' },
+    ]);
+    expect(pb.match('chat-1', '2')).toEqual({ id: 'reject:1', label: '拒绝' });
+    expect(pb.match('chat-1', '1')).toBeUndefined(); // 已消费
+  });
+
+  it('非数字 / 越界不消费，按钮仍有效', () => {
+    const pb = new PendingButtons();
+    pb.set('chat-1', [{ id: 'a', label: '同意' }]);
+    expect(pb.match('chat-1', 'hello')).toBeUndefined();
+    expect(pb.match('chat-1', '5')).toBeUndefined();
+    expect(pb.match('chat-1', '1')).toEqual({ id: 'a', label: '同意' });
+  });
+
+  it('过期后自动失效（不吞后续数字消息）', () => {
+    vi.useFakeTimers();
+    try {
+      const pb = new PendingButtons(1000);
+      pb.set('chat-1', [{ id: 'a', label: '同意' }]);
+      vi.advanceTimersByTime(1001);
+      expect(pb.match('chat-1', '1')).toBeUndefined();
+    } finally {
+      vi.useRealTimers();
+    }
+  });
+
+  it('TTL 内不失效', () => {
+    vi.useFakeTimers();
+    try {
+      const pb = new PendingButtons(PENDING_BUTTONS_TTL_MS);
+      pb.set('chat-1', [{ id: 'a', label: '同意' }]);
+      vi.advanceTimersByTime(PENDING_BUTTONS_TTL_MS - 1);
+      expect(pb.match('chat-1', '1')).toEqual({ id: 'a', label: '同意' });
+    } finally {
+      vi.useRealTimers();
+    }
+  });
+
+  it('consume 删除 pending（原生按钮点击）', () => {
+    const pb = new PendingButtons();
+    pb.set('chat-1', [{ id: 'a', label: '同意' }]);
+    pb.consume('chat-1');
+    expect(pb.match('chat-1', '1')).toBeUndefined();
+  });
+
+  it('不同 chat 互不影响', () => {
+    const pb = new PendingButtons();
+    pb.set('chat-1', [{ id: 'a', label: '同意' }]);
+    pb.set('chat-2', [{ id: 'b', label: '拒绝' }]);
+    expect(pb.match('chat-2', '1')).toEqual({ id: 'b', label: '拒绝' });
+    expect(pb.match('chat-1', '1')).toEqual({ id: 'a', label: '同意' });
+  });
+});