chore: establish learning module baseline

OpenMAIC/tests/web-search/baidu.test.ts

@@ -0,0 +1,116 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+const proxyFetchMock = vi.hoisted(() => vi.fn());
+
+vi.mock('@/lib/server/proxy-fetch', () => ({
+  proxyFetch: proxyFetchMock,
+}));
+
+vi.mock('@/lib/logger', () => ({
+  createLogger: () => ({
+    warn: vi.fn(),
+    info: vi.fn(),
+    error: vi.fn(),
+    debug: vi.fn(),
+  }),
+}));
+
+import { searchWithBaidu } from '@/lib/web-search/baidu';
+
+describe('searchWithBaidu', () => {
+  beforeEach(() => {
+    proxyFetchMock.mockReset();
+  });
+
+  it('queries only enabled Baidu sub-sources and merges their results', async () => {
+    proxyFetchMock.mockResolvedValueOnce(
+      new Response(
+        JSON.stringify({
+          errno: 0,
+          result: {
+            lemma_title: 'OpenMAIC',
+            lemma_url: 'https://baike.baidu.com/item/OpenMAIC',
+            abstract_text: 'Baidu Baike abstract',
+          },
+        }),
+        { status: 200, headers: { 'content-type': 'application/json' } },
+      ),
+    );
+    proxyFetchMock.mockResolvedValueOnce(
+      new Response(
+        JSON.stringify({
+          code: '0',
+          data: [
+            {
+              title: 'OpenMAIC paper',
+              abstract: 'Scholar abstract',
+              aiAbstract: 'Scholar AI abstract',
+              url: 'https://xueshu.baidu.com/usercenter/paper/show?paperid=1',
+              publishYear: 2026,
+              keyword: 'AI education',
+            },
+          ],
+        }),
+        { status: 200, headers: { 'content-type': 'application/json' } },
+      ),
+    );
+
+    const result = await searchWithBaidu({
+      query: 'OpenMAIC',
+      apiKey: 'baidu-key',
+      subSources: {
+        webSearch: false,
+        baike: true,
+        scholar: true,
+      },
+    });
+
+    expect(proxyFetchMock).toHaveBeenCalledTimes(2);
+    expect(proxyFetchMock.mock.calls.map((call) => String(call[0]))).toEqual([
+      'https://appbuilder.baidu.com/v2/baike/lemma/get_content?search_type=lemmaTitle&search_key=OpenMAIC',
+      'https://qianfan.baidubce.com/v2/tools/baidu_scholar/search?wd=OpenMAIC&pageNum=0&enable_ai_abstract=true',
+    ]);
+    expect(result.sources.map((source) => source.title)).toEqual([
+      'OpenMAIC - Baidu Baike',
+      'OpenMAIC paper',
+    ]);
+    expect(result.sources[1]?.content).toBe(
+      'Scholar abstract Scholar AI abstract (2026) AI education',
+    );
+  });
+
+  it('defaults Baidu sub-sources to all enabled when omitted', async () => {
+    proxyFetchMock.mockImplementation((url: string) => {
+      if (url.includes('/ai_search/web_search')) {
+        return Promise.resolve(
+          new Response(JSON.stringify({ code: 0, references: [] }), {
+            status: 200,
+            headers: { 'content-type': 'application/json' },
+          }),
+        );
+      }
+      if (url.includes('/baike/lemma/get_content')) {
+        return Promise.resolve(
+          new Response(JSON.stringify({ errno: 0 }), {
+            status: 200,
+            headers: { 'content-type': 'application/json' },
+          }),
+        );
+      }
+      return Promise.resolve(
+        new Response(JSON.stringify({ code: '0', data: [] }), {
+          status: 200,
+          headers: { 'content-type': 'application/json' },
+        }),
+      );
+    });
+
+    await searchWithBaidu({ query: 'OpenMAIC', apiKey: 'baidu-key' });
+
+    expect(proxyFetchMock.mock.calls.map((call) => String(call[0]))).toEqual([
+      'https://qianfan.baidubce.com/v2/ai_search/web_search',
+      'https://appbuilder.baidu.com/v2/baike/lemma/get_content?search_type=lemmaTitle&search_key=OpenMAIC',
+      'https://qianfan.baidubce.com/v2/tools/baidu_scholar/search?wd=OpenMAIC&pageNum=0&enable_ai_abstract=true',
+    ]);
+  });
+});