참고소스 수정본
This commit is contained in:
170
참고/playwright-main/tests/mcp/cdp.spec.ts
Normal file
170
참고/playwright-main/tests/mcp/cdp.spec.ts
Normal file
@@ -0,0 +1,170 @@
|
||||
/**
|
||||
* Copyright (c) Microsoft Corporation.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { spawnSync } from 'child_process';
|
||||
import { test, expect, mcpServerPath } from './fixtures';
|
||||
|
||||
test.describe.configure({
|
||||
retries: 1,
|
||||
});
|
||||
|
||||
test('cdp server', async ({ cdpServer, startClient, server }) => {
|
||||
await cdpServer.start();
|
||||
const { client } = await startClient({ args: [`--cdp-endpoint=${cdpServer.endpoint}`] });
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
snapshot: expect.stringContaining(`- generic [active] [ref=e1]: Hello, world!`),
|
||||
});
|
||||
});
|
||||
|
||||
test('cdp server reuse tab', async ({ cdpServer, startClient, server }) => {
|
||||
const browserContext = await cdpServer.start();
|
||||
const { client } = await startClient({ args: [`--cdp-endpoint=${cdpServer.endpoint}`] });
|
||||
|
||||
const [page] = browserContext.pages();
|
||||
await page.goto(server.HELLO_WORLD);
|
||||
|
||||
expect(await client.callTool({
|
||||
name: 'browser_click',
|
||||
arguments: {
|
||||
element: 'Hello, world!',
|
||||
target: 'f0',
|
||||
},
|
||||
})).toHaveResponse({
|
||||
error: `Error: "f0" does not match any elements.`,
|
||||
isError: true,
|
||||
});
|
||||
|
||||
expect(await client.callTool({
|
||||
name: 'browser_snapshot',
|
||||
})).toHaveResponse({
|
||||
page: `- Page URL: ${server.HELLO_WORLD}
|
||||
- Page Title: Title`,
|
||||
inlineSnapshot: `- generic [active] [ref=e1]: Hello, world!`,
|
||||
});
|
||||
});
|
||||
|
||||
test('should throw connection error and allow re-connecting', async ({ cdpServer, startClient, server }) => {
|
||||
const { client } = await startClient({ args: [`--cdp-endpoint=${cdpServer.endpoint}`] });
|
||||
|
||||
server.setContent('/', `
|
||||
<title>Title</title>
|
||||
<body>Hello, world!</body>
|
||||
`, 'text/html');
|
||||
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.PREFIX },
|
||||
})).toHaveResponse({
|
||||
error: expect.stringContaining(`Error: connect ECONNREFUSED`),
|
||||
isError: true,
|
||||
});
|
||||
await cdpServer.start();
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.PREFIX },
|
||||
})).toHaveResponse({
|
||||
snapshot: expect.stringContaining(`- generic [active] [ref=e1]: Hello, world!`),
|
||||
});
|
||||
});
|
||||
|
||||
test('auto-recover when remote browser disconnects mid-session', async ({ cdpServer, startClient, server }) => {
|
||||
const browserContext = await cdpServer.start();
|
||||
const { client } = await startClient({ args: [`--cdp-endpoint=${cdpServer.endpoint}`] });
|
||||
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
snapshot: expect.stringContaining(`Hello, world!`),
|
||||
});
|
||||
|
||||
// Simulate the remote browser dying mid-session (e.g. CDP endpoint session timeout).
|
||||
await browserContext.close();
|
||||
|
||||
// The next call hits the dead backend; it must error and let the MCP server discard
|
||||
// the backend so the next call can transparently establish a fresh connection.
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
isError: true,
|
||||
});
|
||||
|
||||
// Bring the CDP endpoint back. The next call should reconnect transparently —
|
||||
// no manual browser_close needed (regression test for playwright-mcp#1588).
|
||||
await cdpServer.start();
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
snapshot: expect.stringContaining(`Hello, world!`),
|
||||
});
|
||||
});
|
||||
|
||||
test('does not support --device', async () => {
|
||||
const result = spawnSync('node', [
|
||||
...mcpServerPath, '--device=Pixel 5', '--cdp-endpoint=http://localhost:1234',
|
||||
]);
|
||||
expect(result.error).toBeUndefined();
|
||||
expect(result.status).toBe(1);
|
||||
expect(result.stderr.toString()).toContain('Device emulation is not supported with cdpEndpoint.');
|
||||
});
|
||||
|
||||
test('cdp server with headers', async ({ startClient, server }) => {
|
||||
let authHeader = '';
|
||||
server.setRoute('/json/version/', (req, res) => {
|
||||
authHeader = req.headers['authorization'];
|
||||
res.end();
|
||||
});
|
||||
|
||||
const { client } = await startClient({ args: [`--cdp-endpoint=${server.PREFIX}`, '--cdp-header', 'Authorization: Bearer 1234567890'] });
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
isError: true,
|
||||
});
|
||||
expect(authHeader).toBe('Bearer 1234567890');
|
||||
});
|
||||
|
||||
test('cdp server with empty and complex headers', async ({ startClient, server }) => {
|
||||
let customHeader = '';
|
||||
let emptyHeader = '';
|
||||
server.setRoute('/json/version/', (req, res) => {
|
||||
customHeader = req.headers['x-forwarded-proto'] as string;
|
||||
emptyHeader = req.headers['x-empty'] as string;
|
||||
res.end();
|
||||
});
|
||||
|
||||
const { client } = await startClient({
|
||||
args: [
|
||||
`--cdp-endpoint=${server.PREFIX}`,
|
||||
'--cdp-header', 'X-Forwarded-Proto: value:with:colons',
|
||||
'--cdp-header', 'X-Empty'
|
||||
]
|
||||
});
|
||||
expect(await client.callTool({
|
||||
name: 'browser_navigate',
|
||||
arguments: { url: server.HELLO_WORLD },
|
||||
})).toHaveResponse({
|
||||
isError: true,
|
||||
});
|
||||
expect(customHeader).toBe('value:with:colons');
|
||||
expect(emptyHeader).toBe('');
|
||||
});
|
||||
Reference in New Issue
Block a user