Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions packages/playwright-core/src/tools/backend/files.ts
Original file line number Diff line number Diff line change
Expand Up @@ -44,10 +44,16 @@ export const uploadFile = defineTabTool({
response.addCode(`await fileChooser.setFiles(${JSON.stringify(paths)})`);

tab.clearModalState(modalState);
await tab.waitForCompletion(async () => {
if (paths)
await modalState.fileChooser.setFiles(paths);
});
try {
await tab.waitForCompletion(async () => {
if (paths)
await modalState.fileChooser.setFiles(paths);
});
} catch (e) {
tab.setModalState(modalState);
response.addError(e instanceof Error ? e.message : String(e));
return;
}
},

clearsModalState: 'fileChooser',
Expand Down
16 changes: 12 additions & 4 deletions packages/playwright-core/src/tools/backend/route.ts
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,18 @@ const route = defineTool({
},

handle: async (context, params, response) => {
const addHeaders = params.headers ? Object.fromEntries(params.headers.map(h => {
const colonIndex = h.indexOf(':');
return [h.substring(0, colonIndex).trim(), h.substring(colonIndex + 1).trim()];
})) : undefined;
let addHeaders: Record<string, string> | undefined;
if (params.headers) {
addHeaders = {};
for (const h of params.headers) {
const colonIndex = h.indexOf(':');
if (colonIndex <= 0) {
response.addError(`Invalid header "${h}": expected "Name: Value"`);
return;
}
addHeaders[h.substring(0, colonIndex).trim()] = h.substring(colonIndex + 1).trim();
}
}
const removeHeaders = params.removeHeaders ? params.removeHeaders.split(',').map(h => h.trim()) : undefined;

const handler = async (route: playwright.Route) => {
Expand Down
37 changes: 37 additions & 0 deletions tests/mcp/files.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -103,6 +103,43 @@ test('browser_file_upload', async ({ client, server }, testInfo) => {
}
});

test('browser_file_upload keeps chooser when setFiles fails', async ({ client, server }, testInfo) => {
server.setContent('/', `<input type="file" />`, 'text/html');

await client.callTool({
name: 'browser_navigate',
arguments: { url: server.PREFIX },
});

await client.callTool({
name: 'browser_click',
arguments: {
element: 'Textbox',
target: 'e2',
},
});

const missing = testInfo.outputPath('missing.txt');
const failed = await client.callTool({
name: 'browser_file_upload',
arguments: { paths: [missing] },
});
expect(failed).toHaveResponse({
isError: true,
modalState: expect.stringContaining(`[File chooser]`),
});

const filePath = testInfo.outputPath('retry.txt');
await fs.writeFile(filePath, 'retry');
const retried = await client.callTool({
name: 'browser_file_upload',
arguments: { paths: [filePath] },
});
expect(retried).toHaveResponse({
modalState: undefined,
});
});

test('clicking on download link emits download', async ({ startClient, server }, testInfo) => {
const { client } = await startClient({
config: { outputDir: testInfo.outputPath('output') },
Expand Down
18 changes: 18 additions & 0 deletions tests/mcp/route.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,24 @@ test('browser_route modifies request headers', async ({ client, server }) => {
expect(receivedHeaders['x-custom-header']).toBe('test-value');
});

test('browser_route errors on header lines without a colon', async ({ client, server }) => {
await client.callTool({
name: 'browser_navigate',
arguments: { url: server.EMPTY_PAGE },
});

expect(await client.callTool({
name: 'browser_route',
arguments: {
pattern: '**/api/check',
headers: ['NotAHeader', 'X-Custom-Header: test-value'],
},
})).toHaveResponse({
isError: true,
error: expect.stringContaining('Invalid header "NotAHeader"'),
});
});

test('browser_route_list shows active routes', async ({ client, server }) => {
await client.callTool({
name: 'browser_navigate',
Expand Down
Loading