step9-cpa-mode.test.js 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. const assert = require('assert');
  2. const fs = require('fs');
  3. const source = fs.readFileSync('background.js', 'utf8');
  4. function extractFunction(name) {
  5. const start = source.indexOf(`function ${name}(`);
  6. if (start < 0) {
  7. throw new Error(`missing function ${name}`);
  8. }
  9. const braceStart = source.indexOf('{', start);
  10. let depth = 0;
  11. let end = braceStart;
  12. for (; end < source.length; end++) {
  13. const ch = source[end];
  14. if (ch === '{') depth += 1;
  15. if (ch === '}') {
  16. depth -= 1;
  17. if (depth === 0) {
  18. end += 1;
  19. break;
  20. }
  21. }
  22. }
  23. return source.slice(start, end);
  24. }
  25. const bundle = [
  26. 'const DEFAULT_LOCAL_CPA_STEP9_MODE = "submit";',
  27. extractFunction('parseUrlSafely'),
  28. extractFunction('isLocalCpaUrl'),
  29. extractFunction('normalizeLocalCpaStep9Mode'),
  30. extractFunction('shouldBypassStep9ForLocalCpa'),
  31. ].join('\n');
  32. const api = new Function(`${bundle}; return { isLocalCpaUrl, normalizeLocalCpaStep9Mode, shouldBypassStep9ForLocalCpa };`)();
  33. assert.strictEqual(api.isLocalCpaUrl('http://127.0.0.1:8317/management.html#/oauth'), true, '127.0.0.1 应视为本地 CPA');
  34. assert.strictEqual(api.isLocalCpaUrl('http://localhost:1455/management.html#/oauth'), true, 'localhost 应视为本地 CPA');
  35. assert.strictEqual(api.isLocalCpaUrl('https://example.com/management.html#/oauth'), false, '远程域名不应视为本地 CPA');
  36. assert.strictEqual(api.isLocalCpaUrl('notaurl'), false, '非法 URL 不应视为本地 CPA');
  37. assert.strictEqual(api.normalizeLocalCpaStep9Mode('submit'), 'submit', 'submit 应保持为 submit');
  38. assert.strictEqual(api.normalizeLocalCpaStep9Mode('bypass'), 'bypass', 'bypass 应保持为 bypass');
  39. assert.strictEqual(api.normalizeLocalCpaStep9Mode('other'), 'submit', '未知模式应回退为 submit');
  40. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  41. vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
  42. localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
  43. }), false, '默认模式下,本地 CPA 也应执行步骤 9');
  44. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  45. localCpaStep9Mode: 'bypass',
  46. vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
  47. localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
  48. }), true, '切换为 bypass 后,本地 CPA 且已有 callback 时应跳过步骤 9');
  49. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  50. localCpaStep9Mode: 'bypass',
  51. vpsUrl: 'https://example.com/management.html#/oauth',
  52. localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
  53. }), false, '远程 CPA 不应跳过步骤 9');
  54. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  55. localCpaStep9Mode: 'bypass',
  56. vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
  57. localhostUrl: '',
  58. }), false, '没有 callback 时不应跳过步骤 9');
  59. console.log('step9 cpa mode tests passed');