重构修改2925获取邮箱

2026-04-18 17:28:43 +08:00
parent 7a4b1e1e12
commit 8e3e085f3d
8 changed files with 717 additions and 205 deletions
@@ -51,37 +51,38 @@ function extractFunction(name) {
  return source.slice(start, end);
 }

-test('handlePollEmail returns to inbox before initial refresh when 2925 opens on a detail page', async () => {
+test('handlePollEmail establishes a baseline after opening from detail view and only picks mail from a later refresh', async () => {
  const bundle = extractFunction('handlePollEmail');

  const api = new Function(`
-let detailMode = true;
+let state = 'detail';
+let refreshCalls = 0;
 const clickOrder = [];
+const readAndDeleteCalls = [];
 const seenCodes = new Set();
-const mailItem = { text: 'OpenAI verification code 654321' };
+const baselineMail = { id: 'baseline', text: 'OpenAI newsletter without code' };
+const newMail = { id: 'new', text: 'OpenAI verification code 654321' };

 function findMailItems() {
-  return detailMode ? [] : [mailItem];
+  if (state === 'detail') return [];
+  if (state === 'baseline') return [baselineMail];
+  return [baselineMail, newMail];
 }

-function getMailItemId() {
-  return 'mail-1';
+function getMailItemId(item) {
+  return item.id;
 }

 function getCurrentMailIds(items = []) {
-  return new Set(items.map(() => 'mail-1'));
-}
-
-function normalizeMinuteTimestamp(value) {
-  return Number(value) || 0;
+  return new Set(items.map((item) => item.id));
 }

 function parseMailItemTimestamp() {
  return Date.now();
 }

-function matchesMailFilters() {
-  return true;
+function matchesMailFilters(text) {
+  return /openai|verification/i.test(String(text || ''));
 }

 function getMailItemText(item) {
@@ -93,29 +94,26 @@ function extractVerificationCode(text) {
  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;
+  state = 'baseline';
  return true;
 }

 async function refreshInbox() {
  clickOrder.push('refresh');
+  refreshCalls += 1;
+  if (refreshCalls >= 2) {
+    state = 'with-new';
+  }
+}
+
+async function openMailAndDeleteAfterRead(item) {
+  readAndDeleteCalls.push(item.id);
+  return item.id === 'new' ? 'Your ChatGPT code is 654321' : 'No code here';
 }

 function persistSeenCodes() {}
@@ -128,38 +126,113 @@ return {
  getClickOrder() {
    return clickOrder.slice();
  },
+  getReadAndDeleteCalls() {
+    return readAndDeleteCalls.slice();
+  },
 };
 `)();

  const result = await api.handlePollEmail(4, {
    senderFilters: ['openai'],
    subjectFilters: ['verification'],
-    maxAttempts: 1,
+    maxAttempts: 2,
    intervalMs: 1,
-    filterAfterTimestamp: Date.now(),
  });

  assert.equal(result.code, '654321');
-  assert.deepEqual(api.getClickOrder(), ['inbox', 'refresh']);
+  assert.deepEqual(api.getClickOrder(), ['inbox', 'refresh', 'inbox', 'refresh']);
+  assert.deepEqual(api.getReadAndDeleteCalls(), ['new']);
+});
+
+test('handlePollEmail ignores targetEmail and still tests any matching ChatGPT mail', async () => {
+  const bundle = extractFunction('handlePollEmail');
+
+  const api = new Function(`
+let state = 'empty';
+const seenCodes = new Set();
+const readAndDeleteCalls = [];
+const matchingMail = {
+  id: 'mail-1',
+  text: 'ChatGPT verification code 112233 for another.user@example.com',
+};
+
+function findMailItems() {
+  return state === 'ready' ? [matchingMail] : [];
+}
+
+function getMailItemId(item) {
+  return item.id;
+}
+
+function getCurrentMailIds(items = []) {
+  return new Set(items.map((item) => item.id));
+}
+
+function parseMailItemTimestamp() {
+  return Date.now();
+}
+
+function matchesMailFilters(text) {
+  return /chatgpt|openai|verification/i.test(String(text || ''));
+}
+
+function getMailItemText(item) {
+  return item.text;
+}
+
+function extractVerificationCode(text) {
+  const match = String(text || '').match(/(\\d{6})/);
+  return match ? match[1] : null;
+}
+
+async function sleep() {}
+async function sleepRandom() {}
+async function returnToInbox() {
+  return true;
+}
+async function refreshInbox() {
+  state = 'ready';
+}
+
+async function openMailAndDeleteAfterRead(item) {
+  readAndDeleteCalls.push(item.id);
+  return item.text;
+}
+
+function persistSeenCodes() {}
+function log() {}
+
+${bundle}
+
+return {
+  handlePollEmail,
+  getReadAndDeleteCalls() {
+    return readAndDeleteCalls.slice();
+  },
+};
+`)();
+
+  const result = await api.handlePollEmail(8, {
+    senderFilters: ['chatgpt'],
+    subjectFilters: ['verification'],
+    maxAttempts: 4,
+    intervalMs: 1,
+    targetEmail: 'expected@example.com',
+  });
+
+  assert.equal(result.code, '112233');
+  assert.deepEqual(api.getReadAndDeleteCalls(), ['mail-1']);
 });

 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 = '';

@@ -169,18 +242,16 @@ const document = {
      return bodyText;
    },
  },
-  querySelector(selector) {
-    if (selector.includes('mailList') || selector.includes('inbox') || selector.includes('Inbox')) {
-      return inboxLink;
-    }
-    return null;
-  },
 };

 function findMailItems() {
  return listVisible ? [mailItem] : [];
 }

+function findInboxLink() {
+  return { kind: 'inbox' };
+}
+
 function simulateClick(node) {
  if (node === mailItem) {
    clickOrder.push('mail');
@@ -188,12 +259,8 @@ function simulateClick(node) {
    bodyText = 'Your ChatGPT code is 731091';
    return;
  }
-  if (node === inboxLink) {
-    clickOrder.push('inbox');
-    listVisible = true;
-    return;
-  }
-  throw new Error('unexpected node');
+  clickOrder.push('inbox');
+  listVisible = true;
 }

 async function sleep() {}
@@ -219,3 +286,132 @@ return {
  assert.deepEqual(api.getClickOrder(), ['mail', 'inbox']);
  assert.equal(api.isListVisible(), true);
 });
+
+test('openMailAndDeleteAfterRead deletes the opened message before returning to inbox', async () => {
+  const bundle = [
+    extractFunction('deleteCurrentMailboxEmail'),
+    extractFunction('returnToInbox'),
+    extractFunction('openMailAndDeleteAfterRead'),
+  ].join('\n');
+
+  const api = new Function(`
+const calls = [];
+const mailItem = { kind: 'mail' };
+const deleteButton = { kind: 'delete' };
+let listVisible = true;
+let bodyText = '';
+
+const document = {
+  body: {
+    get textContent() {
+      return bodyText;
+    },
+  },
+};
+
+function findMailItems() {
+  return listVisible ? [mailItem] : [];
+}
+
+function findDeleteButton() {
+  return deleteButton;
+}
+
+function findInboxLink() {
+  return { kind: 'inbox' };
+}
+
+function simulateClick(node) {
+  if (node === mailItem) {
+    calls.push('mail');
+    listVisible = false;
+    bodyText = 'Your ChatGPT code is 778899';
+    return;
+  }
+  if (node === deleteButton) {
+    calls.push('delete');
+    return;
+  }
+  calls.push('inbox');
+  listVisible = true;
+}
+
+async function sleep() {}
+async function sleepRandom() {}
+const console = { warn() {} };
+const MAIL2925_PREFIX = '[MultiPage:mail-2925]';
+
+${bundle}
+
+return {
+  mailItem,
+  openMailAndDeleteAfterRead,
+  getCalls() {
+    return calls.slice();
+  },
+};
+`)();
+
+  const text = await api.openMailAndDeleteAfterRead(api.mailItem, 8);
+
+  assert.match(text, /778899/);
+  assert.deepEqual(api.getCalls(), ['mail', 'delete', 'inbox']);
+});
+
+test('deleteAllMailboxEmails selects all messages and clicks delete', async () => {
+  const bundle = extractFunction('deleteAllMailboxEmails');
+
+  const api = new Function(`
+const calls = [];
+const selectAllControl = { kind: 'select-all' };
+const deleteButton = { kind: 'delete' };
+
+async function returnToInbox() {
+  calls.push('inbox');
+  return true;
+}
+
+function findSelectAllControl() {
+  return selectAllControl;
+}
+
+function isCheckboxChecked() {
+  return false;
+}
+
+function findDeleteButton() {
+  return deleteButton;
+}
+
+function simulateClick(node) {
+  if (node === selectAllControl) {
+    calls.push('select-all');
+    return;
+  }
+  if (node === deleteButton) {
+    calls.push('delete');
+    return;
+  }
+  throw new Error('unexpected node');
+}
+
+async function sleepRandom() {}
+
+const console = { warn() {} };
+const MAIL2925_PREFIX = '[MultiPage:mail-2925]';
+
+${bundle}
+
+return {
+  deleteAllMailboxEmails,
+  getCalls() {
+    return calls.slice();
+  },
+};
+`)();
+
+  const result = await api.deleteAllMailboxEmails(4);
+
+  assert.equal(result, true);
+  assert.deepEqual(api.getCalls(), ['inbox', 'select-all', 'delete']);
+});