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
2 changes: 1 addition & 1 deletion packages/playwright-core/src/tools/backend/navigate.ts
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ const reload = defineTabTool({
},

handle: async (tab, params, response) => {
await tab.page.reload(tab.navigationTimeoutOptions);
await tab.reload();
response.setIncludeSnapshot();
response.addCode(`await page.reload();`);
},
Expand Down
26 changes: 18 additions & 8 deletions packages/playwright-core/src/tools/backend/tab.ts
Original file line number Diff line number Diff line change
Expand Up @@ -348,8 +348,11 @@ export class Tab extends EventEmitter<TabEventsInterface> {
this._clearCollectedArtifacts();

const { promise: downloadEvent, abort: abortDownloadEvent } = eventWaiter<playwright.Download>(this.page, 'download', 3000);
let modalStates: ModalState[];
try {
await this.page.goto(url, { waitUntil: 'domcontentloaded', ...this.navigationTimeoutOptions });
modalStates = await this._raceAgainstModalStates(async () => {
await this.page.goto(url, { waitUntil: 'domcontentloaded', ...this.navigationTimeoutOptions });
});
abortDownloadEvent();
} catch (_e: unknown) {
const e = _e as Error;
Expand All @@ -364,11 +367,20 @@ export class Tab extends EventEmitter<TabEventsInterface> {
await new Promise(resolve => setTimeout(resolve, 500));
return;
}
if (modalStates.length)
return;

// Cap load event to 5 seconds, the page is operational at this point.
await this.waitForLoadState('load', { timeout: 5000 });
}

async reload() {
await this._initializedPromise;
await this._raceAgainstModalStates(async () => {
await this.page.reload(this.navigationTimeoutOptions);
});
}

async consoleMessageCount(): Promise<{ total: number, errors: number, warnings: number }> {
await this._initializedPromise;
const messages = await this.page.consoleMessages({ filter: 'since-navigation' });
Expand Down Expand Up @@ -504,13 +516,11 @@ export class Tab extends EventEmitter<TabEventsInterface> {
const listener = (modalState: ModalState) => promise.resolve([modalState]);
this.once(TabEvents.modalState, listener);

return await Promise.race([
action().then(() => {
this.off(TabEvents.modalState, listener);
return [];
}),
promise,
]);
try {
return await Promise.race([action().then(() => []), promise]);
} finally {
this.off(TabEvents.modalState, listener);
}
}

async waitForCompletion(callback: () => Promise<void>) {
Expand Down
11 changes: 11 additions & 0 deletions tests/mcp/cli-navigation.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,17 @@ test('go-forward', async ({ cli, server }) => {
- Page Title: Title`);
});

test('reload with alert during load', async ({ cli, server }) => {
server.setContent('/', `<title>Title</title>`, 'text/html');
await cli('open', server.PREFIX);
server.setContent('/', `<title>Title</title><script>alert('MyAlert')</script><button>Button</button>`, 'text/html');
const { output } = await cli('reload');
expect(output).toContain('["alert" dialog with message "MyAlert"]: can be handled by dialog-accept or dialog-dismiss');
await cli('dialog-accept');
const { inlineSnapshot } = await cli('snapshot');
expect(inlineSnapshot).toContain('button "Button"');
});

test('open without url opens about:blank', async ({ cli }) => {
const { output } = await cli('open');
expect(output).toContain('- Page URL: about:blank');
Expand Down
26 changes: 26 additions & 0 deletions tests/mcp/dialogs.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -293,3 +293,29 @@ test('alert dialog w/ race', async ({ client, server }) => {
- Page Title: Title`),
});
});

test('alert dialog during navigation', async ({ client, server }) => {
server.setContent('/', `<title>Title</title><script>alert('Alert')</script><button>Button</button>`, 'text/html');
expect(await client.callTool({
name: 'browser_navigate',
arguments: { url: server.PREFIX },
})).toHaveResponse({
modalState: expect.stringContaining(`- ["alert" dialog with message "Alert"]: can be handled by browser_handle_dialog`),
});

expect(await client.callTool({
name: 'browser_handle_dialog',
arguments: { accept: true },
})).toHaveResponse({
modalState: undefined,
page: expect.stringContaining(`- Page URL: ${server.PREFIX}/
- Page Title: Title`),
});

expect(await client.callTool({
name: 'browser_snapshot',
arguments: {},
})).toHaveResponse({
inlineSnapshot: expect.stringContaining(`- button "Button"`),
});
});
Loading