|
1 | | -import { describe, expect, it } from 'vitest'; |
| 1 | +/** |
| 2 | + * @vitest-environment jsdom |
| 3 | + */ |
| 4 | +import { afterEach, describe, expect, it } from 'vitest'; |
2 | 5 | import { instrumentDOM } from '../../src/instrumentation/dom'; |
3 | 6 | import { WINDOW } from '../../src/types'; |
4 | 7 |
|
5 | 8 | // @ts-expect-error - idk |
6 | 9 | WINDOW.XMLHttpRequest = undefined; |
7 | 10 |
|
8 | | -describe('instrumentXHR', () => { |
9 | | - it('it does not throw if XMLHttpRequest is a key on window but not defined', () => { |
| 11 | +describe('instrumentDOM', () => { |
| 12 | + afterEach(() => { |
| 13 | + // @ts-expect-error - idk |
| 14 | + WINDOW.XMLHttpRequest = undefined; |
| 15 | + }); |
| 16 | + |
| 17 | + it('does not throw if XMLHttpRequest is a key on window but not defined', () => { |
10 | 18 | expect(instrumentDOM).not.toThrow(); |
11 | 19 | }); |
| 20 | + |
| 21 | + it('does not leak document click listeners when removeEventListener uses mismatched capture options', () => { |
| 22 | + instrumentDOM(); |
| 23 | + |
| 24 | + const live = { |
| 25 | + capture: new Set<EventListenerOrEventListenerObject>(), |
| 26 | + bubble: new Set<EventListenerOrEventListenerObject>(), |
| 27 | + }; |
| 28 | + const patchedAdd = EventTarget.prototype.addEventListener; |
| 29 | + const patchedRemove = EventTarget.prototype.removeEventListener; |
| 30 | + |
| 31 | + const captureFlag = (options?: boolean | AddEventListenerOptions | EventListenerOptions): boolean => |
| 32 | + typeof options === 'boolean' ? options : !!options?.capture; |
| 33 | + |
| 34 | + EventTarget.prototype.addEventListener = function ( |
| 35 | + type: string, |
| 36 | + listener: EventListenerOrEventListenerObject, |
| 37 | + options?: boolean | AddEventListenerOptions, |
| 38 | + ) { |
| 39 | + if (this === document && type === 'click') { |
| 40 | + (captureFlag(options) ? live.capture : live.bubble).add(listener); |
| 41 | + } |
| 42 | + return patchedAdd.call(this, type, listener, options); |
| 43 | + }; |
| 44 | + |
| 45 | + EventTarget.prototype.removeEventListener = function ( |
| 46 | + type: string, |
| 47 | + listener: EventListenerOrEventListenerObject, |
| 48 | + options?: boolean | EventListenerOptions, |
| 49 | + ) { |
| 50 | + if (this === document && type === 'click') { |
| 51 | + (captureFlag(options) ? live.capture : live.bubble).delete(listener); |
| 52 | + } |
| 53 | + return patchedRemove.call(this, type, listener, options); |
| 54 | + }; |
| 55 | + |
| 56 | + const baseline = live.capture.size + live.bubble.size; |
| 57 | + |
| 58 | + const never = () => {}; |
| 59 | + const onCapture = () => {}; |
| 60 | + const onBubble = () => {}; |
| 61 | + |
| 62 | + for (let i = 0; i < 20; i++) { |
| 63 | + document.addEventListener('click', onCapture, true); |
| 64 | + document.addEventListener('click', onBubble); |
| 65 | + document.removeEventListener('click', never); |
| 66 | + document.removeEventListener('click', never); |
| 67 | + document.removeEventListener('click', onCapture, true); |
| 68 | + document.removeEventListener('click', onBubble); |
| 69 | + } |
| 70 | + |
| 71 | + expect(live.capture.size + live.bubble.size - baseline).toBe(0); |
| 72 | + }); |
12 | 73 | }); |
0 commit comments