QLHazyCoder
92 lines
3.3 KiB
JavaScript
92 lines
3.3 KiB
JavaScript
const test = require('node:test');
|
|
const assert = require('node:assert/strict');
|
|
const fs = require('node:fs');
|
|
|
|
const source = fs.readFileSync('flows/openai/background/steps/fetch-login-code.js', 'utf8');
|
|
const globalScope = {};
|
|
const api = new Function('self', `${source}; return self.MultiPageBackgroundStep8;`)(globalScope);
|
|
|
|
function createExecutorHarness(overrides = {}) {
|
|
let bypassCalls = 0;
|
|
let capturedMail = null;
|
|
let capturedState = null;
|
|
let capturedOptions = null;
|
|
const executor = api.createStep8Executor({
|
|
addLog: async () => {},
|
|
chrome: { tabs: { update: async () => {} } },
|
|
completeNodeFromBackground: async () => {},
|
|
confirmCustomVerificationStepBypass: async () => { bypassCalls += 1; },
|
|
ensureStep8VerificationPageReady: async () => ({
|
|
state: 'verification_page',
|
|
displayedEmail: 'target@example.com',
|
|
url: 'https://auth.openai.com/verify',
|
|
}),
|
|
getMailConfig: () => ({ provider: 'custom', label: '自定义邮箱' }),
|
|
getState: async () => ({ mailProvider: 'custom', email: 'target@example.com' }),
|
|
getTabId: async () => 1,
|
|
HOTMAIL_PROVIDER: 'hotmail-api',
|
|
isTabAlive: async () => false,
|
|
isVerificationMailPollingError: () => false,
|
|
LUCKMAIL_PROVIDER: 'luckmail-api',
|
|
CLOUDFLARE_TEMP_EMAIL_PROVIDER: 'cloudflare-temp-email',
|
|
CLOUD_MAIL_PROVIDER: 'cloudmail',
|
|
resolveVerificationStep: async (_step, state, mail, options) => {
|
|
capturedState = state;
|
|
capturedMail = mail;
|
|
capturedOptions = options;
|
|
},
|
|
rerunStep7ForStep8Recovery: async () => {},
|
|
resolveSignupEmailForFlow: async () => 'target@example.com',
|
|
reuseOrCreateTab: async () => {},
|
|
sendToContentScriptResilient: async () => ({}),
|
|
setState: async () => {},
|
|
shouldUseCustomRegistrationEmail: () => true,
|
|
shouldUseCustomMailHelper: () => false,
|
|
sleepWithStop: async () => {},
|
|
STANDARD_MAIL_VERIFICATION_RESEND_INTERVAL_MS: 25000,
|
|
STEP7_MAIL_POLLING_RECOVERY_MAX_ATTEMPTS: 1,
|
|
throwIfStopped: () => {},
|
|
...overrides,
|
|
});
|
|
|
|
return {
|
|
executor,
|
|
getBypassCalls: () => bypassCalls,
|
|
getCapturedMail: () => capturedMail,
|
|
getCapturedState: () => capturedState,
|
|
getCapturedOptions: () => capturedOptions,
|
|
};
|
|
}
|
|
|
|
test('step 8 keeps custom mail provider on manual confirmation by default', async () => {
|
|
const harness = createExecutorHarness();
|
|
|
|
await harness.executor.executeStep8({
|
|
mailProvider: 'custom',
|
|
customMailReceiveMode: 'manual',
|
|
email: 'target@example.com',
|
|
oauthUrl: 'https://auth.openai.com/oauth',
|
|
});
|
|
|
|
assert.equal(harness.getBypassCalls(), 1);
|
|
assert.equal(harness.getCapturedMail(), null);
|
|
});
|
|
|
|
test('step 8 routes custom mail provider through resolver only when helper mode is enabled', async () => {
|
|
const harness = createExecutorHarness({
|
|
shouldUseCustomMailHelper: () => true,
|
|
});
|
|
|
|
await harness.executor.executeStep8({
|
|
mailProvider: 'custom',
|
|
customMailReceiveMode: 'helper',
|
|
email: 'target@example.com',
|
|
oauthUrl: 'https://auth.openai.com/oauth',
|
|
});
|
|
|
|
assert.equal(harness.getBypassCalls(), 0);
|
|
assert.equal(harness.getCapturedMail().provider, 'custom');
|
|
assert.equal(harness.getCapturedState().step8VerificationTargetEmail, 'target@example.com');
|
|
assert.equal(harness.getCapturedOptions().targetEmail, 'target@example.com');
|
|
});
|