Files
FlowPilot/tests/background-icloud.test.js
2026-05-24 19:36:08 +08:00

524 lines
16 KiB
JavaScript

const test = require('node:test');
const assert = require('node:assert/strict');
const fs = require('fs');
const {
normalizeIcloudForwardMailProvider,
normalizeIcloudTargetMailboxType,
} = require('../mail-provider-utils.js');
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);
}
const bundle = [
extractFunction('normalizeEmailGenerator'),
extractFunction('getEmailGeneratorLabel'),
extractFunction('normalizeVerificationResendCount'),
extractFunction('normalizePersistentSettingValue'),
extractFunction('finalizeIcloudAliasAfterSuccessfulFlow'),
].join('\n');
function createApi(overrides = {}) {
return new Function('overrides', `
const HOTMAIL_PROVIDER = 'hotmail-api';
const HOTMAIL_SERVICE_MODE_LOCAL = 'local';
const CLOUDFLARE_TEMP_EMAIL_GENERATOR = 'cloudflare-temp-email';
const DEFAULT_LOCAL_CPA_STEP9_MODE = 'submit';
const DEFAULT_HOTMAIL_REMOTE_BASE_URL = '';
const DEFAULT_HOTMAIL_LOCAL_BASE_URL = 'http://127.0.0.1:17373';
const DEFAULT_VERIFICATION_RESEND_COUNT = 4;
const VERIFICATION_RESEND_COUNT_MIN = 0;
const VERIFICATION_RESEND_COUNT_MAX = 20;
const PERSISTED_SETTING_DEFAULTS = {
mailProvider: '163',
autoStepDelaySeconds: null,
};
const calls = {
setUsed: [],
logs: [],
deletes: [],
listCalls: 0,
};
function normalizeIcloudHost(value) {
const normalized = String(value || '').trim().toLowerCase();
return ['icloud.com', 'icloud.com.cn'].includes(normalized) ? normalized : '';
}
function normalizePanelMode(value = '') {
return String(value || '').trim().toLowerCase() === 'sub2api' ? 'sub2api' : 'cpa';
}
function normalizeMailProvider(value = '') {
return String(value || '').trim().toLowerCase() || '163';
}
function normalizeAutoRunFallbackThreadIntervalMinutes(value) {
return Math.max(0, Math.floor(Number(value) || 0));
}
function normalizeAutoStepDelaySeconds(value, fallback = null) {
const numeric = Number(value);
return Number.isFinite(numeric) ? Math.max(0, Math.floor(numeric)) : fallback;
}
function normalizeHotmailServiceMode() {
return HOTMAIL_SERVICE_MODE_LOCAL;
}
function normalizeHotmailRemoteBaseUrl(value = '') {
return String(value || '').trim() || DEFAULT_HOTMAIL_REMOTE_BASE_URL;
}
function normalizeHotmailLocalBaseUrl(value = '') {
return String(value || '').trim() || DEFAULT_HOTMAIL_LOCAL_BASE_URL;
}
function normalizeCloudflareDomain(value = '') {
return String(value || '').trim().toLowerCase();
}
function normalizeCloudflareDomains(values = []) {
return Array.isArray(values) ? values : [];
}
function normalizeCloudflareTempEmailAddress(value = '') {
return String(value || '').trim().toLowerCase();
}
function normalizeCloudflareTempEmailReceiveMailbox(value = '') {
const normalized = normalizeCloudflareTempEmailAddress(value);
return /^[^\\s@]+@[^\\s@]+\\.[^\\s@]+$/.test(normalized) ? normalized : '';
}
function normalizeHotmailAccounts(values = []) {
return Array.isArray(values) ? values : [];
}
function getManualAliasUsageMap(state) {
return { ...(state?.manualAliasUsage || {}) };
}
function getPreservedAliasMap(state) {
return { ...(state?.preservedAliases || {}) };
}
function isAliasPreserved(state, email) {
return Boolean(getPreservedAliasMap(state)[String(email || '').trim().toLowerCase()]);
}
async function setIcloudAliasUsedState(payload, options = {}) {
calls.setUsed.push({ payload, options });
}
async function addLog(message, level = 'info') {
calls.logs.push({ message, level });
}
async function deleteIcloudAlias(alias) {
calls.deletes.push(alias);
}
async function listIcloudAliases() {
calls.listCalls += 1;
return overrides.listIcloudAliases ? overrides.listIcloudAliases() : [];
}
function findIcloudAliasByEmail(aliases, email) {
return (aliases || []).find((alias) => String(alias.email || '').toLowerCase() === String(email || '').toLowerCase()) || null;
}
function getErrorMessage(error) {
return String(typeof error === 'string' ? error : error?.message || '');
}
const normalizeIcloudTargetMailboxType = overrides.normalizeIcloudTargetMailboxType;
const normalizeIcloudForwardMailProvider = overrides.normalizeIcloudForwardMailProvider;
${bundle}
return {
calls,
normalizeEmailGenerator,
getEmailGeneratorLabel,
normalizePersistentSettingValue,
finalizeIcloudAliasAfterSuccessfulFlow,
};
`)(overrides);
}
test('normalizeEmailGenerator and label support icloud', () => {
const api = createApi();
assert.equal(api.normalizeEmailGenerator('icloud'), 'icloud');
assert.equal(api.getEmailGeneratorLabel('icloud'), 'iCloud 隐私邮箱');
});
test('normalizePersistentSettingValue handles icloud settings', () => {
const api = createApi({
normalizeIcloudTargetMailboxType,
normalizeIcloudForwardMailProvider,
});
assert.equal(api.normalizePersistentSettingValue('icloudHostPreference', 'icloud.com'), 'icloud.com');
assert.equal(api.normalizePersistentSettingValue('icloudHostPreference', 'bad-host'), 'auto');
assert.equal(api.normalizePersistentSettingValue('icloudTargetMailboxType', 'forward-mailbox'), 'forward-mailbox');
assert.equal(api.normalizePersistentSettingValue('icloudTargetMailboxType', 'wrong'), 'icloud-inbox');
assert.equal(api.normalizePersistentSettingValue('icloudForwardMailProvider', 'GMAIL'), 'gmail');
assert.equal(api.normalizePersistentSettingValue('icloudForwardMailProvider', 'unknown'), 'qq');
assert.equal(api.normalizePersistentSettingValue('autoDeleteUsedIcloudAlias', 1), true);
assert.equal(api.normalizePersistentSettingValue('verificationResendCount', '6'), 6);
assert.equal(api.normalizePersistentSettingValue('verificationResendCount', 99), 20);
assert.equal(api.normalizePersistentSettingValue('cloudflareTempEmailReceiveMailbox', ' Forward@Example.com '), 'forward@example.com');
});
test('finalizeIcloudAliasAfterSuccessfulFlow marks icloud aliases as used without deleting when auto-delete is off', async () => {
const api = createApi();
const result = await api.finalizeIcloudAliasAfterSuccessfulFlow({
email: 'alias@icloud.com',
emailGenerator: 'icloud',
autoDeleteUsedIcloudAlias: false,
manualAliasUsage: {},
preservedAliases: {},
});
assert.deepEqual(result, { handled: true, deleted: false });
assert.equal(api.calls.setUsed.length, 1);
assert.equal(api.calls.listCalls, 0);
assert.equal(api.calls.deletes.length, 0);
});
test('finalizeIcloudAliasAfterSuccessfulFlow skips deleting preserved aliases', async () => {
const api = createApi();
const result = await api.finalizeIcloudAliasAfterSuccessfulFlow({
email: 'alias@icloud.com',
emailGenerator: 'icloud',
autoDeleteUsedIcloudAlias: true,
manualAliasUsage: {},
preservedAliases: { 'alias@icloud.com': true },
});
assert.deepEqual(result, { handled: true, deleted: false });
assert.equal(api.calls.setUsed.length, 1);
assert.equal(api.calls.listCalls, 0);
assert.equal(api.calls.deletes.length, 0);
});
test('finalizeIcloudAliasAfterSuccessfulFlow skips deleting aliases that are preserved in the latest alias list', async () => {
const api = createApi({
listIcloudAliases() {
return [
{ email: 'alias@icloud.com', anonymousId: 'anon-1', preserved: true },
];
},
});
const result = await api.finalizeIcloudAliasAfterSuccessfulFlow({
email: 'alias@icloud.com',
emailGenerator: 'icloud',
autoDeleteUsedIcloudAlias: true,
manualAliasUsage: {},
preservedAliases: {},
});
assert.deepEqual(result, { handled: true, deleted: false });
assert.equal(api.calls.setUsed.length, 1);
assert.equal(api.calls.listCalls, 1);
assert.equal(api.calls.deletes.length, 0);
});
test('finalizeIcloudAliasAfterSuccessfulFlow deletes alias when auto-delete is enabled and alias exists', async () => {
const api = createApi({
listIcloudAliases() {
return [
{ email: 'alias@icloud.com', anonymousId: 'anon-1', preserved: false },
];
},
});
const result = await api.finalizeIcloudAliasAfterSuccessfulFlow({
email: 'alias@icloud.com',
emailGenerator: 'icloud',
autoDeleteUsedIcloudAlias: true,
manualAliasUsage: {},
preservedAliases: {},
});
assert.deepEqual(result, { handled: true, deleted: true });
assert.equal(api.calls.setUsed.length, 1);
assert.equal(api.calls.listCalls, 1);
assert.deepEqual(api.calls.deletes, [
{ email: 'alias@icloud.com', anonymousId: 'anon-1', preserved: false },
]);
});
test('finalizeIcloudAliasAfterSuccessfulFlow ignores non-icloud flows', async () => {
const api = createApi();
const result = await api.finalizeIcloudAliasAfterSuccessfulFlow({
email: 'plain@example.com',
emailGenerator: 'duck',
autoDeleteUsedIcloudAlias: true,
manualAliasUsage: {},
preservedAliases: {},
});
assert.deepEqual(result, { handled: false, deleted: false });
assert.equal(api.calls.setUsed.length, 0);
});
test('icloudRequest retries retryable network failures and then succeeds', async () => {
const bundle = [
extractFunction('getIcloudRequestTargetLabel'),
extractFunction('getIcloudRetryDelay'),
extractFunction('isIcloudRetryableStatus'),
extractFunction('isIcloudRetryableError'),
extractFunction('icloudRequest'),
].join('\n');
const api = new Function(`
const ICLOUD_REQUEST_TIMEOUT_MS = 15000;
const ICLOUD_RETRY_DELAYS_MS = [1, 1, 1];
const activeIcloudRequestControllers = new Set();
let stopRequested = false;
const logs = [];
let fetchCalls = 0;
function throwIfStopped() {
if (stopRequested) {
throw new Error('流程已被用户停止。');
}
}
async function sleepWithStop() {}
async function addLog(message, level) {
logs.push({ message, level });
}
function getErrorMessage(error) {
return String(typeof error === 'string' ? error : error?.message || '');
}
function normalizeText(value) {
return String(value || '').replace(/\\s+/g, ' ').trim();
}
function appendIcloudClientQueryParams(url) {
return String(url || '');
}
function isIcloudMaildomainwsHost() {
return false;
}
function shouldTryIcloudRequestPageContextFallback() {
return false;
}
async function icloudRequestViaPageContext() {
throw new Error('not expected');
}
async function fetch() {
fetchCalls += 1;
if (fetchCalls === 1) {
throw new Error('Failed to fetch');
}
return {
ok: true,
text: async () => JSON.stringify({ success: true }),
};
}
${bundle}
return {
icloudRequest,
readFetchCalls: () => fetchCalls,
readLogs: () => logs,
};
`)();
const result = await api.icloudRequest('GET', 'https://p67-maildomainws.icloud.com/v2/hme/list', {
maxAttempts: 2,
logRetries: true,
retryLabel: '加载 iCloud 别名列表',
});
assert.deepEqual(result, { success: true });
assert.equal(api.readFetchCalls(), 2);
assert.equal(api.readLogs().length, 1);
});
test('icloudRequest does not retry non-retryable 403 responses', async () => {
const bundle = [
extractFunction('getIcloudRequestTargetLabel'),
extractFunction('getIcloudRetryDelay'),
extractFunction('isIcloudRetryableStatus'),
extractFunction('isIcloudRetryableError'),
extractFunction('icloudRequest'),
].join('\n');
const api = new Function(`
const ICLOUD_REQUEST_TIMEOUT_MS = 15000;
const ICLOUD_RETRY_DELAYS_MS = [1, 1, 1];
const activeIcloudRequestControllers = new Set();
let stopRequested = false;
let fetchCalls = 0;
function throwIfStopped() {
if (stopRequested) {
throw new Error('流程已被用户停止。');
}
}
async function sleepWithStop() {}
async function addLog() {}
function getErrorMessage(error) {
return String(typeof error === 'string' ? error : error?.message || '');
}
function normalizeText(value) {
return String(value || '').replace(/\\s+/g, ' ').trim();
}
function appendIcloudClientQueryParams(url) {
return String(url || '');
}
function isIcloudMaildomainwsHost() {
return false;
}
function shouldTryIcloudRequestPageContextFallback() {
return false;
}
async function icloudRequestViaPageContext() {
throw new Error('not expected');
}
async function fetch() {
fetchCalls += 1;
return {
ok: false,
status: 403,
text: async () => 'forbidden',
};
}
${bundle}
return {
icloudRequest,
readFetchCalls: () => fetchCalls,
};
`)();
await assert.rejects(
api.icloudRequest('GET', 'https://p67-maildomainws.icloud.com/v2/hme/list', {
maxAttempts: 3,
logRetries: true,
}),
/status 403/i
);
assert.equal(api.readFetchCalls(), 1);
});
test('fetchIcloudHideMyEmail treats reserve network failure as success when alias appears in the refreshed list', async () => {
const bundle = [
extractFunction('getIcloudRequestTargetLabel'),
extractFunction('getIcloudRetryDelay'),
extractFunction('isIcloudRetryableStatus'),
extractFunction('isIcloudRetryableError'),
extractFunction('fetchIcloudHideMyEmail'),
].join('\n');
const api = new Function(`
const logs = [];
const selectedEmails = [];
const broadcasts = [];
let listCallCount = 0;
const ICLOUD_REQUEST_TIMEOUT_MS = 15000;
const ICLOUD_WRITE_MAX_ATTEMPTS = 2;
function withIcloudLoginHelp(_label, action) {
return action();
}
function throwIfStopped() {}
async function addLog(message, level = 'info') {
logs.push({ message, level });
}
async function resolveIcloudPremiumMailService() {
return {
serviceUrl: 'https://p67-maildomainws.icloud.com',
setupUrl: 'https://setup.icloud.com/setup/ws/1',
};
}
function getErrorMessage(error) {
return String(typeof error === 'string' ? error : error?.message || '');
}
async function loadNormalizedIcloudAliases() {
listCallCount += 1;
if (listCallCount === 1) {
return { aliases: [], serviceUrl: 'https://p67-maildomainws.icloud.com' };
}
return {
aliases: [
{ email: 'fresh@icloud.com', active: true, used: false, preserved: false },
],
serviceUrl: 'https://p67-maildomainws.icloud.com',
};
}
async function listIcloudAliases() {
const response = await loadNormalizedIcloudAliases();
return response.aliases || [];
}
function pickReusableIcloudAlias() {
return null;
}
async function icloudRequest(method, url) {
if (method === 'POST' && /\\/v1\\/hme\\/generate$/.test(url)) {
return { success: true, result: { hme: 'fresh@icloud.com' } };
}
if (method === 'POST' && /\\/v1\\/hme\\/reserve$/.test(url)) {
const error = new Error('Failed to fetch');
error.networkFailure = true;
throw error;
}
throw new Error('unexpected request');
}
function getIcloudAliasLabel() {
return 'MultiPage 2026-04-26';
}
async function setEmailState(email) {
selectedEmails.push(email);
}
function broadcastIcloudAliasesChanged(payload) {
broadcasts.push(payload);
}
function findIcloudAliasByEmail(aliases, email) {
return (aliases || []).find((alias) => String(alias.email || '').toLowerCase() === String(email || '').toLowerCase()) || null;
}
function shouldStopIcloudAutoFetchRetries() {
return true;
}
${bundle}
return {
fetchIcloudHideMyEmail,
readLogs: () => logs,
readSelectedEmails: () => selectedEmails,
readBroadcasts: () => broadcasts,
};
`)();
const email = await api.fetchIcloudHideMyEmail({ generateNew: true });
assert.equal(email, 'fresh@icloud.com');
assert.deepEqual(api.readSelectedEmails(), ['fresh@icloud.com']);
assert.deepEqual(api.readBroadcasts(), [{ reason: 'created', email: 'fresh@icloud.com' }]);
assert.match(api.readLogs().map((entry) => entry.message).join('\n'), /已在列表确认别名/);
});