step9-cpa-mode.test.js 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  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. extractFunction('parseUrlSafely'),
  27. extractFunction('isLocalCpaUrl'),
  28. extractFunction('shouldBypassStep9ForLocalCpa'),
  29. ].join('\n');
  30. const api = new Function(`${bundle}; return { isLocalCpaUrl, shouldBypassStep9ForLocalCpa };`)();
  31. assert.strictEqual(api.isLocalCpaUrl('http://127.0.0.1:8317/management.html#/oauth'), true, '127.0.0.1 应视为本地 CPA');
  32. assert.strictEqual(api.isLocalCpaUrl('http://localhost:1455/management.html#/oauth'), true, 'localhost 应视为本地 CPA');
  33. assert.strictEqual(api.isLocalCpaUrl('https://example.com/management.html#/oauth'), false, '远程域名不应视为本地 CPA');
  34. assert.strictEqual(api.isLocalCpaUrl('notaurl'), false, '非法 URL 不应视为本地 CPA');
  35. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  36. vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
  37. localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
  38. }), true, '本地 CPA 且已有 callback 时应跳过远程提交流程');
  39. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  40. vpsUrl: 'https://example.com/management.html#/oauth',
  41. localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
  42. }), false, '远程 CPA 不应跳过步骤 9');
  43. assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
  44. vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
  45. localhostUrl: '',
  46. }), false, '没有 callback 时不应跳过步骤 9');
  47. console.log('step9 cpa mode tests passed');