项目文件夹

文件
Alex Rudenko b4546ef86b
release-please / release-please (push) Has been cancelled
Compile and run tests / Tests on macos-latest with node 22 (push) Has been cancelled
Compile and run tests / Tests on ubuntu-latest with node 22 (push) Has been cancelled
Compile and run tests / Tests on windows-latest with node 22 (push) Has been cancelled
Compile and run tests / Tests on macos-latest with node 24 (push) Has been cancelled
Compile and run tests / Tests on ubuntu-latest with node 24 (push) Has been cancelled
Compile and run tests / Tests on windows-latest with node 24 (push) Has been cancelled
Compile and run tests / Tests on macos-latest with node 26 (push) Has been cancelled
Compile and run tests / Tests on ubuntu-latest with node 26 (push) Has been cancelled
Compile and run tests / Tests on windows-latest with node 26 (push) Has been cancelled
Check code before submitting / [Required] Check correct format (push) Has been cancelled
Check code before submitting / [Required] Check docs updated (push) Has been cancelled
Compile and run tests / [Required] Tests passed (push) Has been cancelled
test: fix potential flakiness in tests (#2396)
Assortment of various flakiness conditions found running tests in a loop
locally:


This PR introduces a comprehensive set of hermetic retry layers and
aggressive
timeout handlers across the test suite to insulate it from random
Chromium
startup hangs, CDP deadlocks, and Puppeteer lifecycle flakes. It
guarantees that
temporary browser infrastructure failures are automatically retried
without
failing the CI, while actual code assertion failures still fail fast.

### Test Harness & Retry Improvements

• tests/utils.ts: Rewrote withBrowser to include a 30-second internal
timeout and
a 3-attempt retry loop. If Chromium locks up or disconnects (Target
closed /
socket hang up), the browser is forcibly evicted (via SIGKILL if
browser.close()
hangs) and the test setup is cleanly retried.
• tests/index.test.ts: Wrapped withClient (used by E2E tests) in a
3-attempt
retry loop to handle the daemon/Chromium hanging during launch and
triggering the
60-second MCP client timeout.
• tests/browser.test.ts: Added a safeClose helper that imposes a
2-second timeout
before SIGKILLing browsers, and wrapped raw Puppeteer tests in
runWithRetry to
handle startup hangs.
• tests/shutdown.test.ts: Added a setupServerWithRetry helper to prevent
random
60s RPC timeouts when the server's Chrome instance hangs during boot.

### Flaky Operations & Navigation Fixes

• src/tools/performance.ts & tests/tools/performance.test.ts: Replaced
the
notoriously flaky waitUntil: ['networkidle0'] with 'load' when
navigating to
about:blank in performance_start_trace. This prevents random 10-second
Navigation
timeout exceeded errors. Also stubbed goto in the associated unit tests
for
better hermeticity.
• src/McpContext.ts: Wrapped browser.installExtension() with a 15-second
timeout
to prevent deadlocks when an extension fails to load.
• tests/tools/extensions.test.ts: Removed flaky headless UI navigations
to
chrome://extensions in favor of using the context.listExtensions() API.
• tests/tools/pages.test.js.snapshot: Synced test snapshots to reflect
updated
environment baselines.
2026-07-21 13:47:47 +00:00

134 行
4.1 KiB
TypeScript

/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import assert from 'node:assert';
import {describe, it} from 'node:test';
import type {McpPage} from '../../src/McpPage.js';
import {listPages, navigatePage, selectPage} from '../../src/tools/pages.js';
import {executeWebMcpTool} from '../../src/tools/webmcp.js';
import {html, withMcpContext} from '../utils.js';
describe('webmcp', () => {
describe('list_webmcp_tools', () => {
it('list webmcp tools in navigate_page response', async () => {
await withMcpContext(async (response, context) => {
await navigatePage().handler(
{
params: {url: 'data:text/html,<html></html>'},
page: context.getSelectedMcpPage(),
},
response,
context,
);
assert.ok(response.listWebMcpTools);
});
});
it('list webmcp tools in list_pages response', async () => {
await withMcpContext(async (response, context) => {
await listPages().handler({params: {}}, response, context);
assert.ok(response.listWebMcpTools);
});
});
it('list webmcp tools in select_page response', async () => {
await withMcpContext(async (response, context) => {
const pageId = context.getSelectedMcpPage().id ?? 1;
await selectPage.handler({params: {pageId}}, response, context);
assert.ok(response.listWebMcpTools);
});
});
});
describe('execute_webmcp_tool', () => {
async function setupWebMcpTool(page: McpPage) {
await page.pptrPage.setContent(
html`<form
toolname="test_tool"
tooldescription="A test tool"
toolautosubmit
></form
><script>
document.querySelector('form').onsubmit = event => {
event.preventDefault();
event.respondWith('hello');
};
</script>`,
);
}
// TODO: Remove `.skip` once Chrome 149 reaches stable channel.
it.skip('executes a tool successfully', async () => {
await withMcpContext(
async (response, context) => {
const page = context.getSelectedMcpPage();
await setupWebMcpTool(page);
await executeWebMcpTool.handler(
{params: {toolName: 'test_tool', input: JSON.stringify({})}, page},
response,
context,
);
assert.strictEqual(
response.responseLines[0],
JSON.stringify({status: 'Completed', output: 'hello'}, null, 2),
);
},
{args: ['--enable-features=WebMCP,DevToolsWebMCPSupport']},
{categoryExperimentalWebmcp: true},
);
});
it('throws if tool is not found', async () => {
await withMcpContext(
async (response, context) => {
await assert.rejects(
async () => {
await executeWebMcpTool.handler(
{
params: {toolName: 'missing-tool', input: JSON.stringify({})},
page: context.getSelectedMcpPage(),
},
response,
context,
);
},
{message: /Tool missing-tool not found/},
);
},
{args: ['--enable-features=WebMCP,DevToolsWebMCPSupport']},
{categoryExperimentalWebmcp: true},
);
});
it('throws if input is invalid', async () => {
await withMcpContext(
async (response, context) => {
await assert.rejects(
async () => {
const page = context.getSelectedMcpPage();
await setupWebMcpTool(page);
await executeWebMcpTool.handler(
{params: {toolName: 'test_tool', input: 'invalid'}, page},
response,
context,
);
},
{
message:
/Failed to parse input as JSON: Unexpected token 'i', "invalid" is not valid JSON/,
},
);
},
{args: ['--enable-features=WebMCP,DevToolsWebMCPSupport']},
{categoryExperimentalWebmcp: true},
);
});
});
});