feat: 添加邮件刷新和收件箱链接查找功能，并增加相关测试用例

2026-04-18 12:17:23 +08:00
parent 033e76a16f
commit 7290649930
2 changed files with 276 additions and 3 deletions
@@ -0,0 +1,221 @@
+const test = require('node:test');
+const assert = require('node:assert/strict');
+const fs = require('node:fs');
+
+const source = fs.readFileSync('content/mail-2925.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);
+}
+
+test('handlePollEmail returns to inbox before initial refresh when 2925 opens on a detail page', async () => {
+  const bundle = extractFunction('handlePollEmail');
+
+  const api = new Function(`
+let detailMode = true;
+const clickOrder = [];
+const seenCodes = new Set();
+const mailItem = { text: 'OpenAI verification code 654321' };
+
+function findMailItems() {
+  return detailMode ? [] : [mailItem];
+}
+
+function getMailItemId() {
+  return 'mail-1';
+}
+
+function getCurrentMailIds(items = []) {
+  return new Set(items.map(() => 'mail-1'));
+}
+
+function normalizeMinuteTimestamp(value) {
+  return Number(value) || 0;
+}
+
+function parseMailItemTimestamp() {
+  return Date.now();
+}
+
+function matchesMailFilters() {
+  return true;
+}
+
+function getMailItemText(item) {
+  return item.text;
+}
+
+function extractVerificationCode(text) {
+  const match = String(text || '').match(/(\\d{6})/);
+  return match ? match[1] : null;
+}
+
+function extractEmails() {
+  return [];
+}
+
+function emailMatchesTarget() {
+  return true;
+}
+
+function getTargetEmailMatchState() {
+  return { matches: true, hasExplicitEmail: false };
+}
+
+async function sleep() {}
+async function sleepRandom() {}
+
+async function returnToInbox() {
+  clickOrder.push('inbox');
+  detailMode = false;
+  return true;
+}
+
+async function refreshInbox() {
+  clickOrder.push('refresh');
+}
+
+function persistSeenCodes() {}
+function log() {}
+
+${bundle}
+
+return {
+  handlePollEmail,
+  getClickOrder() {
+    return clickOrder.slice();
+  },
+};
+`)();
+
+  const result = await api.handlePollEmail(4, {
+    senderFilters: ['openai'],
+    subjectFilters: ['verification'],
+    maxAttempts: 1,
+    intervalMs: 1,
+    filterAfterTimestamp: Date.now(),
+  });
+
+  assert.equal(result.code, '654321');
+  assert.deepEqual(api.getClickOrder(), ['inbox', 'refresh']);
+});
+
+test('openMailAndGetMessageText always returns to inbox after opening a 2925 message', async () => {
+  const bundle = [
+    extractFunction('findInboxLink'),
+    extractFunction('returnToInbox'),
+    extractFunction('openMailAndGetMessageText'),
+  ].join('\n');
+
+  const api = new Function(`
+const MAIL_INBOX_SELECTORS = [
+  'a[href*="mailList"]',
+  '[class*="inbox"]',
+  '[class*="Inbox"]',
+  '[title*="鏀朵欢绠?]',
+];
+const clickOrder = [];
+const mailItem = { kind: 'mail' };
+const inboxLink = { kind: 'inbox' };
+let listVisible = true;
+let bodyText = '';
+
+const document = {
+  body: {
+    get textContent() {
+      return bodyText;
+    },
+  },
+  querySelector(selector) {
+    if (selector.includes('mailList') || selector.includes('inbox') || selector.includes('Inbox')) {
+      return inboxLink;
+    }
+    return null;
+  },
+};
+
+function findMailItems() {
+  return listVisible ? [mailItem] : [];
+}
+
+function simulateClick(node) {
+  if (node === mailItem) {
+    clickOrder.push('mail');
+    listVisible = false;
+    bodyText = 'Your ChatGPT code is 731091';
+    return;
+  }
+  if (node === inboxLink) {
+    clickOrder.push('inbox');
+    listVisible = true;
+    return;
+  }
+  throw new Error('unexpected node');
+}
+
+async function sleep() {}
+async function sleepRandom() {}
+
+${bundle}
+
+return {
+  mailItem,
+  openMailAndGetMessageText,
+  getClickOrder() {
+    return clickOrder.slice();
+  },
+  isListVisible() {
+    return listVisible;
+  },
+};
+`)();
+
+  const text = await api.openMailAndGetMessageText(api.mailItem);
+
+  assert.match(text, /731091/);
+  assert.deepEqual(api.getClickOrder(), ['mail', 'inbox']);
+  assert.equal(api.isListVisible(), true);
+});