signup-entry-diagnostics.test.js 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. const test = require('node:test');
  2. const assert = require('node:assert/strict');
  3. const fs = require('node:fs');
  4. const source = fs.readFileSync('content/signup-page.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('signup entry diagnostics summarizes current page inputs and visible actions', () => {
  49. const api = new Function(`
  50. const SIGNUP_ENTRY_TRIGGER_PATTERN = /免费注册|立即注册|注册|sign\\s*up|register|create\\s*account|create\\s+account/i;
  51. const location = { href: 'https://chatgpt.com/' };
  52. const document = {
  53. title: 'ChatGPT',
  54. readyState: 'complete',
  55. querySelector() {
  56. return null;
  57. },
  58. querySelectorAll(selector) {
  59. if (selector === 'a, button, [role="button"], [role="link"], input[type="button"], input[type="submit"]') {
  60. return [
  61. {
  62. tagName: 'BUTTON',
  63. textContent: 'Get started',
  64. disabled: false,
  65. getBoundingClientRect() {
  66. return { width: 120, height: 40 };
  67. },
  68. getAttribute(name) {
  69. return name === 'type' ? 'button' : '';
  70. },
  71. },
  72. {
  73. tagName: 'A',
  74. textContent: 'Log in',
  75. disabled: false,
  76. getBoundingClientRect() {
  77. return { width: 96, height: 40 };
  78. },
  79. getAttribute() {
  80. return '';
  81. },
  82. },
  83. ];
  84. }
  85. return [];
  86. },
  87. };
  88. function isVisibleElement() {
  89. return true;
  90. }
  91. function getActionText(el) {
  92. return [el?.textContent, el?.value, el?.getAttribute?.('aria-label'), el?.getAttribute?.('title')]
  93. .filter(Boolean)
  94. .join(' ')
  95. .replace(/\\s+/g, ' ')
  96. .trim();
  97. }
  98. function isActionEnabled(el) {
  99. return Boolean(el) && !el.disabled && el.getAttribute('aria-disabled') !== 'true';
  100. }
  101. function getSignupEmailInput() {
  102. return null;
  103. }
  104. function getSignupPasswordInput() {
  105. return null;
  106. }
  107. function getPageTextSnapshot() {
  108. return 'Welcome to ChatGPT. Try our latest models.';
  109. }
  110. ${extractFunction('getSignupEntryDiagnostics')}
  111. return {
  112. run() {
  113. return getSignupEntryDiagnostics();
  114. },
  115. };
  116. `)();
  117. const result = api.run();
  118. assert.equal(result.url, 'https://chatgpt.com/');
  119. assert.equal(result.title, 'ChatGPT');
  120. assert.equal(result.readyState, 'complete');
  121. assert.equal(result.hasEmailInput, false);
  122. assert.equal(result.hasPasswordInput, false);
  123. assert.equal(result.bodyContainsSignupText, false);
  124. assert.deepStrictEqual(result.signupLikeActions, []);
  125. assert.deepStrictEqual(result.visibleActions, [
  126. { tag: 'button', type: 'button', text: 'Get started', enabled: true },
  127. { tag: 'a', type: '', text: 'Log in', enabled: true },
  128. ]);
  129. assert.match(result.bodyTextPreview, /Welcome to ChatGPT/);
  130. });