Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix(webkit): do not poll readyState if target is paused before first navigation (#721)

Browse source
This commit is contained in:
Yury Semikhatsky 2020-01-28 14:00:36 -08:00 committed by Dmitry Gozman
parent 9d34f28a40
commit 460527d8cb
2 changed files with 37 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/webkit/wkPage.ts
View file

@ -84,11 +84,11 @@ export class WKPage implements PageDelegate {
this._workers.setSession(session);
}
async initialize(session: WKSession) {
async initialize(session: WKSession, pagePausedOnStart: boolean) {
this._setSession(session);
await Promise.all([
this._initializePageProxySession(),
this._initializeSession(this._session, ({frameTree}) => this._handleFrameTree(frameTree)),
this._initializeSession(this._session, ({frameTree}) => this._handleFrameTree(frameTree, pagePausedOnStart)),
]);
}
@ -234,24 +234,26 @@ export class WKPage implements PageDelegate {
this._page._frameManager.frameLifecycleEvent(frameId, event);
}
private _handleFrameTree(frameTree: Protocol.Page.FrameResourceTree) {
private _handleFrameTree(frameTree: Protocol.Page.FrameResourceTree, pagePausedOnStart: boolean) {
const frame = this._onFrameAttached(frameTree.frame.id, frameTree.frame.parentId || null);
this._onFrameNavigated(frameTree.frame, true);
frame._utilityContext().then(async context => {
const readyState = await context.evaluate(() => document.readyState).catch(e => 'loading');
if (frame.isDetached())
return;
if (readyState === 'interactive' || readyState === 'complete')
this._page._frameManager.frameLifecycleEvent(frame._id, 'domcontentloaded');
if (readyState === 'complete')
this._page._frameManager.frameLifecycleEvent(frame._id, 'load');
});
if (!pagePausedOnStart) {
frame._utilityContext().then(async context => {
const readyState = await context.evaluate(() => document.readyState).catch(e => 'loading');
if (frame.isDetached())
return;
if (readyState === 'interactive' || readyState === 'complete')
this._page._frameManager.frameLifecycleEvent(frame._id, 'domcontentloaded');
if (readyState === 'complete')
this._page._frameManager.frameLifecycleEvent(frame._id, 'load');
});
}
if (!frameTree.childFrames)
return;
for (const child of frameTree.childFrames)
this._handleFrameTree(child);
this._handleFrameTree(child, pagePausedOnStart);
}
_onFrameAttached(frameId: string, parentFrameId: string | null): frames.Frame {

32
src/webkit/wkPageProxy.ts
View file

@ -32,6 +32,7 @@ export class WKPageProxy {
private _wkPage: WKPage | null = null;
private readonly _firstTargetPromise: Promise<void>;
private _firstTargetCallback?: () => void;
private _pagePausedOnStart: boolean = false;
private readonly _sessions = new Map<string, WKSession>();
private readonly _eventListeners: RegisteredListener[];
@ -121,7 +122,11 @@ export class WKPageProxy {
}
assert(session, 'One non-provisional target session must exist');
this._wkPage = new WKPage(this._browserContext, this._pageProxySession);
await this._wkPage.initialize(session!);
await this._wkPage.initialize(session!, this._pagePausedOnStart);
if (this._pagePausedOnStart) {
this._resumeTarget(session!.sessionId);
this._pagePausedOnStart = false;
}
return this._wkPage._page;
}
@ -140,19 +145,26 @@ export class WKPageProxy {
this._firstTargetCallback();
this._firstTargetCallback = undefined;
}
if (targetInfo.isProvisional)
if (targetInfo.isProvisional) {
(session as any)[isPovisionalSymbol] = true;
if (targetInfo.isProvisional && this._wkPage)
this._wkPage.onProvisionalLoadStarted(session);
if (targetInfo.isPaused) {
const resume = () => this._pageProxySession.send('Target.resume', { targetId: targetInfo.targetId }).catch(debugError);
if (targetInfo.isProvisional || !this._pagePromise)
resume();
else
this._pagePromise.then(resume);
if (this._wkPage)
this._wkPage.onProvisionalLoadStarted(session);
if (targetInfo.isPaused)
this._resumeTarget(targetInfo.targetId);
} else if (this._pagePromise) {
assert(!this._pagePausedOnStart);
// This is the first time page target is created, will resume
// after finishing intialization.
this._pagePausedOnStart = !!targetInfo.isPaused;
} else if (targetInfo.isPaused) {
this._resumeTarget(targetInfo.targetId);
}
}
private _resumeTarget(targetId: string) {
this._pageProxySession.send('Target.resume', { targetId }).catch(debugError);
}
private _onTargetDestroyed(event: Protocol.Target.targetDestroyedPayload) {
const { targetId, crashed } = event;
const session = this._sessions.get(targetId);