crawler-plugin/frontend-vue/tests/modules-query-asin.test.ts

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], userId: 42 })
  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], userId: 42 })
  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'), /店铺不存在/)
})