250 lines
10 KiB
TypeScript
250 lines
10 KiB
TypeScript
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'), /店铺不存在/)
|
|
})
|