test: add verification stop propagation tests for error handling

tests/verification-stop-propagation.test.js

@@ -0,0 +1,186 @@
+const assert = require('assert');
+const fs = require('fs');
+
+const source = fs.readFileSync('background.js', 'utf8');
+
+function extractFunction(name) {
+  const markers = [`async function ${name}(`, `function ${name}(`];
+  const start = markers
+    .map(marker => source.indexOf(marker))
+    .find(index => index >= 0);
+  if (start < 0) {
+    throw new Error(`missing function ${name}`);
+  }
+
+  let parenDepth = 0;
+  let signatureEnded = false;
+  let braceStart = -1;
+  for (let i = start; i < source.length; i += 1) {
+    const ch = source[i];
+    if (ch === '(') {
+      parenDepth += 1;
+    } else if (ch === ')') {
+      parenDepth -= 1;
+      if (parenDepth === 0) {
+        signatureEnded = true;
+      }
+    } else if (ch === '{' && signatureEnded) {
+      braceStart = i;
+      break;
+    }
+  }
+  if (braceStart < 0) {
+    throw new Error(`missing body for function ${name}`);
+  }
+
+  let depth = 0;
+  let end = braceStart;
+  for (; end < source.length; end += 1) {
+    const ch = source[end];
+    if (ch === '{') depth += 1;
+    if (ch === '}') {
+      depth -= 1;
+      if (depth === 0) {
+        end += 1;
+        break;
+      }
+    }
+  }
+
+  return source.slice(start, end);
+}
+
+async function testPollFreshVerificationCodeRethrowsStop() {
+  const bundle = [
+    extractFunction('isStopError'),
+    extractFunction('throwIfStopped'),
+    extractFunction('pollFreshVerificationCode'),
+  ].join('\n');
+
+  const api = new Function(`
+let stopRequested = false;
+const STOP_ERROR_MESSAGE = '流程已被用户停止。';
+const HOTMAIL_PROVIDER = 'hotmail-api';
+const VERIFICATION_POLL_MAX_ROUNDS = 5;
+const logs = [];
+let resendCalls = 0;
+
+function getHotmailVerificationPollConfig() {
+  return {};
+}
+async function pollHotmailVerificationCode() {
+  throw new Error('hotmail path should not run in this test');
+}
+function getVerificationCodeStateKey(step) {
+  return step === 4 ? 'lastSignupCode' : 'lastLoginCode';
+}
+function getVerificationPollPayload(step, state, overrides = {}) {
+  return {
+    filterAfterTimestamp: 123,
+    ...overrides,
+  };
+}
+async function sendToMailContentScriptResilient() {
+  throw new Error(STOP_ERROR_MESSAGE);
+}
+async function requestVerificationCodeResend() {
+  resendCalls += 1;
+}
+async function addLog(message, level) {
+  logs.push({ message, level });
+}
+
+${bundle}
+
+return {
+  pollFreshVerificationCode,
+  snapshot() {
+    return { logs, resendCalls };
+  },
+};
+`)();
+
+  let error = null;
+  try {
+    await api.pollFreshVerificationCode(7, {}, { provider: 'qq' }, {});
+  } catch (err) {
+    error = err;
+  }
+
+  const state = api.snapshot();
+  assert.strictEqual(error?.message, '流程已被用户停止。', 'Stop 错误应原样向上抛出');
+  assert.strictEqual(state.resendCalls, 0, 'Stop 后不应继续请求新的验证码');
+  assert.deepStrictEqual(state.logs, [], 'Stop 后不应再记录普通失败或重试日志');
+}
+
+async function testResolveVerificationStepRethrowsStopFromFreshRequest() {
+  const bundle = [
+    extractFunction('isStopError'),
+    extractFunction('resolveVerificationStep'),
+  ].join('\n');
+
+  const api = new Function(`
+const STOP_ERROR_MESSAGE = '流程已被用户停止。';
+const HOTMAIL_PROVIDER = 'hotmail-api';
+const logs = [];
+let pollCalls = 0;
+
+function getVerificationCodeStateKey(step) {
+  return step === 4 ? 'lastSignupCode' : 'lastLoginCode';
+}
+function getHotmailVerificationPollConfig() {
+  return {};
+}
+function getVerificationCodeLabel(step) {
+  return step === 4 ? '注册' : '登录';
+}
+function isStep7RestartFromStep6Error() {
+  return false;
+}
+async function requestVerificationCodeResend() {
+  throw new Error(STOP_ERROR_MESSAGE);
+}
+async function addLog(message, level) {
+  logs.push({ message, level });
+}
+async function pollFreshVerificationCode() {
+  pollCalls += 1;
+  return { code: '123456', emailTimestamp: Date.now() };
+}
+async function submitVerificationCode() {
+  throw new Error('submit should not run in this test');
+}
+async function setState() {}
+async function completeStepFromBackground() {}
+
+${bundle}
+
+return {
+  resolveVerificationStep,
+  snapshot() {
+    return { logs, pollCalls };
+  },
+};
+`)();
+
+  let error = null;
+  try {
+    await api.resolveVerificationStep(7, {}, { provider: 'qq' }, { requestFreshCodeFirst: true });
+  } catch (err) {
+    error = err;
+  }
+
+  const state = api.snapshot();
+  assert.strictEqual(error?.message, '流程已被用户停止。', '首次请求新验证码收到 Stop 后应立即终止');
+  assert.strictEqual(state.pollCalls, 0, 'Stop 后不应继续进入邮箱轮询');
+  assert.deepStrictEqual(state.logs, [], 'Stop 后不应追加降级日志');
+}
+
+(async () => {
+  await testPollFreshVerificationCodeRethrowsStop();
+  await testResolveVerificationStepRethrowsStopFromFreshRequest();
+  console.log('verification stop propagation tests passed');
+})().catch((error) => {
+  console.error(error);
+  process.exit(1);
+});