mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
fix(page): fix page.off method for request event (#7624)
This patch fixes page.off for request event Closes: #7572
This commit is contained in:
parent
30ac9d168d
commit
d0cb9436a3
@ -16,7 +16,7 @@
|
||||
|
||||
import type { Readable } from 'stream';
|
||||
|
||||
import { EventEmitter } from './EventEmitter.js';
|
||||
import { EventEmitter, Handler } from './EventEmitter.js';
|
||||
import {
|
||||
Connection,
|
||||
CDPSession,
|
||||
@ -459,6 +459,7 @@ export class Page extends EventEmitter {
|
||||
|
||||
private _disconnectPromise?: Promise<Error>;
|
||||
private _userDragInterceptionEnabled = false;
|
||||
private _handlerMap = new WeakMap<Handler, Handler>();
|
||||
|
||||
/**
|
||||
* @internal
|
||||
@ -623,11 +624,15 @@ export class Page extends EventEmitter {
|
||||
handler: (event: PageEventObject[K]) => void
|
||||
): EventEmitter {
|
||||
if (eventName === 'request') {
|
||||
return super.on(eventName, (event: HTTPRequest) => {
|
||||
const wrap = (event: HTTPRequest) => {
|
||||
event.enqueueInterceptAction(() =>
|
||||
handler(event as PageEventObject[K])
|
||||
);
|
||||
});
|
||||
};
|
||||
|
||||
this._handlerMap.set(handler, wrap);
|
||||
|
||||
return super.on(eventName, wrap);
|
||||
}
|
||||
return super.on(eventName, handler);
|
||||
}
|
||||
@ -641,6 +646,17 @@ export class Page extends EventEmitter {
|
||||
return super.once(eventName, handler);
|
||||
}
|
||||
|
||||
off<K extends keyof PageEventObject>(
|
||||
eventName: K,
|
||||
handler: (event: PageEventObject[K]) => void
|
||||
): EventEmitter {
|
||||
if (eventName === 'request') {
|
||||
handler = this._handlerMap.get(handler) || handler;
|
||||
}
|
||||
|
||||
return super.off(eventName, handler);
|
||||
}
|
||||
|
||||
/**
|
||||
* This method is typically coupled with an action that triggers file
|
||||
* choosing. The following example clicks a button that issues a file chooser
|
||||
|
@ -139,6 +139,23 @@ describe('Page', function () {
|
||||
// Two now because we added the handler back.
|
||||
expect(handler.callCount).toBe(2);
|
||||
});
|
||||
|
||||
it('should correctly added and removed request events', async () => {
|
||||
const { page, server } = getTestState();
|
||||
|
||||
const handler = sinon.spy();
|
||||
page.on('request', handler);
|
||||
await page.goto(server.EMPTY_PAGE);
|
||||
expect(handler.callCount).toBe(1);
|
||||
page.off('request', handler);
|
||||
await page.goto(server.EMPTY_PAGE);
|
||||
// Still one because we removed the handler.
|
||||
expect(handler.callCount).toBe(1);
|
||||
page.on('request', handler);
|
||||
await page.goto(server.EMPTY_PAGE);
|
||||
// Two now because we added the handler back.
|
||||
expect(handler.callCount).toBe(2);
|
||||
});
|
||||
});
|
||||
|
||||
describeFailsFirefox('Page.Events.error', function () {
|
||||
|
Loading…
Reference in New Issue
Block a user