103 lines
5.0 KiB
JavaScript
103 lines
5.0 KiB
JavaScript
// 本地网络中继（Worker 侧）：node test/relay.test.mjs
|
||
// 重点不是「中继能跑」（那要真扩展），而是**出错时面板能不能看出卡在哪里**：
|
||
// 吾爱破解线上日志曾只有一句「本地网络失败：中继执行超时（45秒）」，用户在面板上
|
||
// 完全不知道卡的是哪个请求。所以失败/超时都必须把请求行写进 err.detail。
|
||
import assert from 'node:assert/strict';
|
||
import { waitRelayResult } from '../src/lib/relay.js';
|
||
|
||
let n = 0;
|
||
const t = async (name, fn) => { await fn(); n++; console.log('ok -', name); };
|
||
|
||
// 极简 D1 fake：一条 relay_jobs 记录
|
||
function fakeDb(job) {
|
||
return {
|
||
prepare() {
|
||
const stmt = {
|
||
bind() { return stmt; },
|
||
async first() { return job; },
|
||
async run() { return {}; },
|
||
};
|
||
return stmt;
|
||
},
|
||
};
|
||
}
|
||
|
||
await t('中继：扩展回传失败时，错误里带上卡住的请求（method + url）', async () => {
|
||
const db = fakeDb({ status: 'failed', error: '中继执行超时（45秒），已放弃该请求', method: 'post', url: 'https://www.52pojie.cn/home.php?mod=task&do=apply&id=2' });
|
||
const err = await waitRelayResult(db, 'r1', 200, 20).catch((e) => e);
|
||
assert.ok(err instanceof Error);
|
||
assert.match(err.message, /本地网络失败/);
|
||
assert.match(err.detail, /POST https:\/\/www\.52pojie\.cn\/home\.php\?mod=task/);
|
||
});
|
||
|
||
await t('中继：Worker 端等超时，也要说清是哪个请求没回来', async () => {
|
||
const db = fakeDb({ status: 'pending', method: 'GET', url: 'https://example.com/api/sign' });
|
||
const err = await waitRelayResult(db, 'r2', 150, 30).catch((e) => e);
|
||
assert.ok(err instanceof Error);
|
||
assert.match(err.message, /等待本地网络响应超时/);
|
||
assert.match(err.detail, /GET https:\/\/example\.com\/api\/sign/);
|
||
});
|
||
|
||
await t('中继：url 过长时截断，不把整串查询参数灌进日志', async () => {
|
||
const long = 'https://example.com/api/sign?' + 'a=1&'.repeat(200);
|
||
const db = fakeDb({ status: 'failed', error: 'boom', method: 'GET', url: long });
|
||
const err = await waitRelayResult(db, 'r3', 200, 20).catch((e) => e);
|
||
assert.ok(err.detail.length < 200, 'detail 应被截断，实际长度 ' + err.detail.length);
|
||
});
|
||
|
||
// ---- runner：中继没等到回包时不能报「失败」 ----
|
||
// 线上现象（糊涂鳄）：POST admin-ajax 被吊到超时，面板显示「签到失败」，
|
||
// 但服务端很可能已经处理了这次请求（网站那天确实已签）—— 报失败就是冤枉网站。
|
||
// 现在记「结果未知」（skip，不写 last_signin_date），并允许定时器稍后自动复核。
|
||
function fakeDbRelayTimeout() {
|
||
const inserted = [];
|
||
const now = Date.now();
|
||
let lastUrl = '';
|
||
let lastMethod = 'GET';
|
||
const home = Buffer.from('<html>首页</html>', 'utf8').toString('base64');
|
||
return {
|
||
inserted,
|
||
prepare(sql) {
|
||
const stmt = {
|
||
_args: [],
|
||
bind(...args) { stmt._args = args; return stmt; },
|
||
async run() {
|
||
if (/INSERT INTO runs/i.test(sql)) inserted.push({ sql, args: stmt._args });
|
||
// 记下队列里的任务，好让后续查询能区分「主页 GET」与「签到 POST」
|
||
if (/INSERT INTO relay_jobs/i.test(sql)) { lastUrl = String(stmt._args[1]); lastMethod = String(stmt._args[2]).toUpperCase(); }
|
||
return {};
|
||
},
|
||
async all() { return { results: [] }; },
|
||
async first() {
|
||
if (/FROM settings/i.test(sql)) return { value: String(now) }; // 扩展在线
|
||
if (/FROM relay_jobs/i.test(sql)) {
|
||
// 主页能读到，签到 POST 被吊到超时（线上糊涂鳄的真实情形）
|
||
if (lastMethod === 'POST') return { status: 'failed', error: '中继执行超时：请求已发出但没收到回包', method: lastMethod, url: lastUrl };
|
||
return { status: 'done', resp_status: 200, resp_headers: '{}', resp_body: home };
|
||
}
|
||
return null;
|
||
},
|
||
};
|
||
return stmt;
|
||
},
|
||
async batch() { return []; },
|
||
};
|
||
}
|
||
|
||
await t('runner：中继超时 → 记「结果未知」（skip）而不是「失败」，并允许自动补跑', async () => {
|
||
const { encryptJSON } = await import('../src/crypto.js');
|
||
const { runAccount } = await import('../src/runner.js');
|
||
const db = fakeDbRelayTimeout();
|
||
const env = { DB: db, ENCRYPT_KEY: 'El771KvGwTGzl6K9C2dqmMOsOBYgF3LR9pIm/FvTEbs=' };
|
||
const credsEnc = await encryptJSON(env, db, { site_url: 'https://dj.hutue.cn', cookie: 'c=x' });
|
||
const account = { id: 21, site: 'hutue', name: '糊涂鳄', creds: credsEnc, meta: '{"execution":"relay"}', enabled: 1 };
|
||
const r = await runAccount(env, account);
|
||
assert.equal(r.status, 'skip', '不能记成 fail，实际：' + r.status + ' / ' + r.message);
|
||
assert.equal(r.retryable, true, '稍后应能自动补跑复核');
|
||
assert.match(r.message, /结果未知/);
|
||
// 写入 runs 的 status 也必须是 skip（面板才不会显示 ❌ 失败）
|
||
assert.equal(db.inserted[0].args[3], 'skip');
|
||
});
|
||
|
||
console.log(`\n${n} 组通过`);
|