task-20: 拆分 shop-data-crawl / query-asin 模块 API 至 types/modules/query-asin.ts，java-modules.ts re-export

2026-08-31 19:45:59 +08:00
parent be02833b11
commit 71b482df6d
4 changed files with 684 additions and 363 deletions
@@ -0,0 +1,249 @@
+import { test } from 'node:test'
+import assert from 'node:assert/strict'
+import { http } from '../src/shared/api/http.ts'
+import {
+  listShopDataCrawlCandidates,
+  addShopDataCrawlCandidate,
+  deleteShopDataCrawlCandidate,
+  getShopDataCrawlCountryPreference,
+  putShopDataCrawlCountryPreference,
+  matchShopDataCrawlShops,
+  getShopDataCrawlDashboard,
+  getShopDataCrawlHistory,
+  createShopDataCrawlTask,
+  getShopDataCrawlTaskProgressBatch,
+  getShopDataCrawlResultDownloadUrl,
+  deleteShopDataCrawlTask,
+  deleteShopDataCrawlHistory,
+  listQueryAsinCandidates,
+  addQueryAsinCandidate,
+  deleteQueryAsinCandidate,
+  matchQueryAsinShops,
+  getQueryAsinDashboard,
+  getQueryAsinHistory,
+  getQueryAsinTaskProgressBatch,
+  createQueryAsinTask,
+  submitQueryAsinTaskResult,
+  getQueryAsinResultDownloadUrl,
+  deleteQueryAsinTask,
+  deleteQueryAsinHistory,
+} from '../src/shared/api/types/modules/query-asin.ts'
+
+function setupWindow() {
+  ;(globalThis as Record<string, unknown>).window = {
+    localStorage: { getItem: () => '42' },
+    location: { origin: 'http://localhost' },
+  }
+}
+
+function mockRequest(
+  t: Parameters<typeof test>[1] extends (t: infer T) => unknown ? T : never,
+  impl: (config: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown; timeout?: number }) => Promise<unknown>,
+) {
+  t.mock.method(http, 'request', impl as never)
+}
+
+const okResponse = (data: unknown) => Promise.resolve({ data: { success: true, message: 'ok', data } })
+
+test('test_shop_data_crawl_candidates', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ id: 1, shop_name: 's1' })
+  })
+  await listShopDataCrawlCandidates()
+  await addShopDataCrawlCandidate('店铺A')
+  await deleteShopDataCrawlCandidate(3)
+  assert.equal(calls[0].url, '/newApi/api/shop-data-crawl/candidates')
+  assert.deepEqual(calls[0].params, { user_id: 42 })
+  assert.equal(calls[1].url, '/newApi/api/shop-data-crawl/candidates')
+  assert.equal(calls[1].method, 'POST')
+  assert.deepEqual(calls[1].data, { user_id: 42, shop_name: '店铺A' })
+  assert.equal(calls[2].url, '/newApi/api/shop-data-crawl/candidates/3')
+  assert.equal(calls[2].method, 'DELETE')
+})
+
+test('test_shop_data_crawl_preference_match', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ country_codes: ['US'] })
+  })
+  await getShopDataCrawlCountryPreference()
+  await putShopDataCrawlCountryPreference(['US'])
+  await matchShopDataCrawlShops(['店铺A'])
+  assert.equal(calls[0].url, '/newApi/api/shop-data-crawl/country-preference')
+  assert.deepEqual(calls[0].params, { user_id: 42 })
+  assert.equal(calls[1].url, '/newApi/api/shop-data-crawl/country-preference')
+  assert.equal(calls[1].method, 'PUT')
+  assert.deepEqual(calls[1].data, { user_id: 42, country_codes: ['US'] })
+  assert.equal(calls[2].url, '/newApi/api/shop-data-crawl/match-shops')
+  assert.deepEqual(calls[2].data, { user_id: 42, shop_names: ['店铺A'] })
+})
+
+test('test_shop_data_crawl_task_apis', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ taskId: 1, items: [] })
+  })
+  await createShopDataCrawlTask([{ shopName: '店铺A', matched: true }], ['US'])
+  await deleteShopDataCrawlTask(7)
+  await deleteShopDataCrawlHistory(9)
+  assert.equal(calls[0].url, '/newApi/api/shop-data-crawl/tasks')
+  assert.equal(calls[0].method, 'POST')
+  assert.deepEqual(calls[0].data, { user_id: 42, items: [{ shopName: '店铺A', matched: true }], country_codes: ['US'] })
+  assert.equal(calls[1].url, '/newApi/api/shop-data-crawl/tasks/7')
+  assert.equal(calls[1].method, 'DELETE')
+  assert.deepEqual(calls[1].params, { user_id: 42 })
+  assert.equal(calls[2].url, '/newApi/api/shop-data-crawl/history/9')
+  assert.equal(calls[2].method, 'DELETE')
+})
+
+test('test_shop_data_crawl_dashboard_history', async (t) => {
+  setupWindow()
+  const calls: { url?: string; params?: Record<string, unknown> }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ candidateCount: 0, processedTaskCount: 0, successTaskCount: 1, failedTaskCount: 0 })
+  })
+  await getShopDataCrawlDashboard()
+  await getShopDataCrawlHistory()
+  assert.equal(calls[0].url, '/newApi/api/shop-data-crawl/dashboard')
+  assert.deepEqual(calls[0].params, { user_id: 42 })
+  assert.equal(calls[1].url, '/newApi/api/shop-data-crawl/history')
+  assert.deepEqual(calls[1].params, { user_id: 42 })
+  assert.equal(
+    getShopDataCrawlResultDownloadUrl(7),
+    'http://localhost/newApi/api/shop-data-crawl/results/7/download?user_id=42',
+  )
+})
+
+test('test_shop_data_crawl_progress_batch_timeout', async (t) => {
+  setupWindow()
+  let captured: { url?: string; method?: string; data?: unknown; timeout?: number } = {}
+  mockRequest(t, (config) => {
+    captured = config
+    return okResponse({ items: [], missingTaskIds: [] })
+  })
+  const result = await getShopDataCrawlTaskProgressBatch([5, 3])
+  assert.equal(captured.url, '/newApi/api/shop-data-crawl/tasks/progress/batch')
+  assert.equal(captured.method, 'POST')
+  assert.deepEqual(captured.data, { taskIds: [3, 5] })
+  assert.equal(captured.timeout, 10000)
+  assert.deepEqual(result, { items: [], missingTaskIds: [] })
+})
+
+test('test_query_asin_candidates', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ id: 1, shop_name: 's1' })
+  })
+  await listQueryAsinCandidates()
+  await addQueryAsinCandidate('店铺A')
+  await deleteQueryAsinCandidate(3)
+  assert.equal(calls[0].url, '/newApi/api/query-asin/candidates')
+  assert.deepEqual(calls[0].params, { user_id: 42 })
+  assert.equal(calls[1].url, '/newApi/api/query-asin/candidates')
+  assert.equal(calls[1].method, 'POST')
+  assert.deepEqual(calls[1].data, { user_id: 42, shop_name: '店铺A' })
+  assert.equal(calls[2].url, '/newApi/api/query-asin/candidates/3')
+  assert.equal(calls[2].method, 'DELETE')
+})
+
+test('test_query_asin_match_dashboard_history', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; params?: Record<string, unknown>; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ items: [] })
+  })
+  await matchQueryAsinShops(['店铺A'])
+  await getQueryAsinDashboard()
+  await getQueryAsinHistory()
+  assert.equal(calls[0].url, '/newApi/api/query-asin/match-shops')
+  assert.deepEqual(calls[0].data, { user_id: 42, shop_names: ['店铺A'] })
+  assert.equal(calls[1].url, '/newApi/api/query-asin/dashboard')
+  assert.deepEqual(calls[1].params, { user_id: 42 })
+  assert.equal(calls[2].url, '/newApi/api/query-asin/history')
+  assert.deepEqual(calls[2].params, { user_id: 42 })
+})
+
+test('test_query_asin_task_apis', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; data?: unknown }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ taskId: 1, items: [] })
+  })
+  await createQueryAsinTask([{ queryAsins: [{ country: 'US', asins: ['B001'] }] }])
+  const payload = {
+    shops: [{ shopName: '店铺A', countryResults: [{ country: 'US', items: [{ asin: 'B001', status: 'ok' }] }], shopDone: true }],
+  }
+  await submitQueryAsinTaskResult(7, payload)
+  assert.equal(calls[0].url, '/newApi/api/query-asin/tasks')
+  assert.equal(calls[0].method, 'POST')
+  assert.deepEqual(calls[0].data, { user_id: 42, items: [{ queryAsins: [{ country: 'US', asins: ['B001'] }] }] })
+  assert.equal(calls[1].url, '/newApi/api/query-asin/tasks/7/result')
+  assert.deepEqual(calls[1].data, payload)
+})
+
+test('test_query_asin_progress_delete_download', async (t) => {
+  setupWindow()
+  const calls: { url?: string; method?: string; data?: unknown; timeout?: number }[] = []
+  mockRequest(t, (config) => {
+    calls.push(config)
+    return okResponse({ items: [], missingTaskIds: [] })
+  })
+  await getQueryAsinTaskProgressBatch([5, 3])
+  await deleteQueryAsinTask(7)
+  await deleteQueryAsinHistory(9)
+  assert.equal(calls[0].url, '/newApi/api/query-asin/tasks/progress/batch')
+  assert.deepEqual(calls[0].data, { taskIds: [3, 5] })
+  assert.equal(calls[0].timeout, 10000)
+  assert.equal(calls[1].url, '/newApi/api/query-asin/tasks/7')
+  assert.equal(calls[1].method, 'DELETE')
+  assert.equal(calls[2].url, '/newApi/api/query-asin/history/9')
+  assert.equal(calls[2].method, 'DELETE')
+  assert.equal(
+    getQueryAsinResultDownloadUrl(7),
+    'http://localhost/newApi/api/query-asin/results/7/download?user_id=42',
+  )
+})
+
+test('test_query_asin_export_compat', async (t) => {
+  setupWindow()
+  const fromJavaModules = await import('../src/shared/api/java-modules.ts')
+  const fromQueryAsin = await import('../src/shared/api/types/modules/query-asin.ts')
+  const names = [
+    'listShopDataCrawlCandidates', 'addShopDataCrawlCandidate', 'deleteShopDataCrawlCandidate',
+    'getShopDataCrawlCountryPreference', 'putShopDataCrawlCountryPreference', 'matchShopDataCrawlShops',
+    'getShopDataCrawlDashboard', 'getShopDataCrawlHistory', 'createShopDataCrawlTask',
+    'getShopDataCrawlTaskProgressBatch', 'getShopDataCrawlResultDownloadUrl',
+    'deleteShopDataCrawlTask', 'deleteShopDataCrawlHistory',
+    'listQueryAsinCandidates', 'addQueryAsinCandidate', 'deleteQueryAsinCandidate',
+    'matchQueryAsinShops', 'getQueryAsinDashboard', 'getQueryAsinHistory',
+    'getQueryAsinTaskProgressBatch', 'createQueryAsinTask', 'submitQueryAsinTaskResult',
+    'getQueryAsinResultDownloadUrl', 'deleteQueryAsinTask', 'deleteQueryAsinHistory',
+  ]
+  for (const name of names) {
+    assert.equal(fromJavaModules[name], fromQueryAsin[name], `${name} 应为同一引用`)
+  }
+  const item = { resultId: 1, taskId: 2, shopName: 's1', queryAsins: [{ country: 'US', asins: ['B001'] }] }
+  assert.equal(item.queryAsins[0].asins[0], 'B001')
+})
+
+test('test_query_asin_unwrap_and_error', async (t) => {
+  setupWindow()
+  mockRequest(t, () => okResponse({ taskId: 1, items: [] }))
+  const result = await createQueryAsinTask([])
+  assert.equal(result.taskId, 1)
+
+  mockRequest(t, () => Promise.resolve({ data: { success: false, message: '店铺不存在' } }))
+  await assert.rejects(addQueryAsinCandidate('x'), /店铺不存在/)
+})