cloudflare-temp-email-provider.test.js 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. const test = require('node:test');
  2. const assert = require('node:assert/strict');
  3. const fs = require('fs');
  4. const source = fs.readFileSync('background.js', 'utf8');
  5. function extractFunction(name) {
  6. const markers = [`async function ${name}(`, `function ${name}(`];
  7. const start = markers
  8. .map((marker) => source.indexOf(marker))
  9. .find((index) => index >= 0);
  10. if (start < 0) {
  11. throw new Error(`missing function ${name}`);
  12. }
  13. let parenDepth = 0;
  14. let signatureEnded = false;
  15. let braceStart = -1;
  16. for (let i = start; i < source.length; i += 1) {
  17. const ch = source[i];
  18. if (ch === '(') {
  19. parenDepth += 1;
  20. } else if (ch === ')') {
  21. parenDepth -= 1;
  22. if (parenDepth === 0) {
  23. signatureEnded = true;
  24. }
  25. } else if (ch === '{' && signatureEnded) {
  26. braceStart = i;
  27. break;
  28. }
  29. }
  30. if (braceStart < 0) {
  31. throw new Error(`missing body for function ${name}`);
  32. }
  33. let depth = 0;
  34. let end = braceStart;
  35. for (; end < source.length; end += 1) {
  36. const ch = source[end];
  37. if (ch === '{') depth += 1;
  38. if (ch === '}') {
  39. depth -= 1;
  40. if (depth === 0) {
  41. end += 1;
  42. break;
  43. }
  44. }
  45. }
  46. return source.slice(start, end);
  47. }
  48. test('pollCloudflareTempEmailVerificationCode returns code even if delete fails', async () => {
  49. const bundle = [
  50. extractFunction('isStopError'),
  51. extractFunction('throwIfStopped'),
  52. extractFunction('summarizeCloudflareTempEmailMessagesForLog'),
  53. extractFunction('pollCloudflareTempEmailVerificationCode'),
  54. ].join('\n');
  55. const api = new Function(`
  56. let stopRequested = false;
  57. const STOP_ERROR_MESSAGE = '流程已被用户停止。';
  58. const CLOUDFLARE_TEMP_EMAIL_DEFAULT_PAGE_SIZE = 20;
  59. const logs = [];
  60. function normalizeCloudflareTempEmailAddress(value) {
  61. return String(value || '').trim().toLowerCase();
  62. }
  63. async function addLog(message, level) {
  64. logs.push({ message, level });
  65. }
  66. async function sleepWithStop() {}
  67. async function listCloudflareTempEmailMessages() {
  68. return {
  69. config: {},
  70. messages: [{
  71. id: 'mail-1',
  72. address: 'user@example.com',
  73. receivedDateTime: '2026-04-13T09:20:00.000Z',
  74. subject: 'OpenAI verification code',
  75. from: { emailAddress: { address: 'noreply@tm.openai.com' } },
  76. bodyPreview: 'Your verification code is 123456.',
  77. }],
  78. };
  79. }
  80. function pickVerificationMessageWithTimeFallback(messages) {
  81. return {
  82. match: {
  83. code: '123456',
  84. receivedAt: Date.parse(messages[0].receivedDateTime),
  85. message: messages[0],
  86. },
  87. usedRelaxedFilters: false,
  88. usedTimeFallback: false,
  89. };
  90. }
  91. async function deleteCloudflareTempEmailMail() {
  92. throw new Error('delete failed');
  93. }
  94. ${bundle}
  95. return {
  96. pollCloudflareTempEmailVerificationCode,
  97. snapshot() {
  98. return { logs };
  99. },
  100. };
  101. `)();
  102. const result = await api.pollCloudflareTempEmailVerificationCode(4, { email: 'user@example.com' }, {
  103. targetEmail: 'user@example.com',
  104. maxAttempts: 1,
  105. intervalMs: 1,
  106. });
  107. assert.equal(result.code, '123456');
  108. const state = api.snapshot();
  109. assert.equal(state.logs.some((entry) => entry.message.includes('删除 Cloudflare Temp Email 邮件失败')), true);
  110. });
  111. test('pollCloudflareTempEmailVerificationCode requires target email', async () => {
  112. const bundle = [
  113. extractFunction('isStopError'),
  114. extractFunction('throwIfStopped'),
  115. extractFunction('pollCloudflareTempEmailVerificationCode'),
  116. ].join('\n');
  117. const api = new Function(`
  118. let stopRequested = false;
  119. const STOP_ERROR_MESSAGE = '流程已被用户停止。';
  120. const CLOUDFLARE_TEMP_EMAIL_DEFAULT_PAGE_SIZE = 20;
  121. function normalizeCloudflareTempEmailAddress(value) {
  122. return String(value || '').trim().toLowerCase();
  123. }
  124. async function addLog() {}
  125. async function sleepWithStop() {}
  126. async function listCloudflareTempEmailMessages() {
  127. throw new Error('should not reach list');
  128. }
  129. function pickVerificationMessageWithTimeFallback() {
  130. return { match: null, usedRelaxedFilters: false, usedTimeFallback: false };
  131. }
  132. async function deleteCloudflareTempEmailMail() {}
  133. function summarizeCloudflareTempEmailMessagesForLog() {
  134. return '';
  135. }
  136. ${bundle}
  137. return { pollCloudflareTempEmailVerificationCode };
  138. `)();
  139. await assert.rejects(
  140. api.pollCloudflareTempEmailVerificationCode(4, {}, {}),
  141. /缺少目标邮箱地址/
  142. );
  143. });