/
jsdom.ts
66 lines (62 loc) · 1.7 KB
/
jsdom.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import { importModule } from 'local-pkg'
import type { Environment, JSDOMOptions } from '../../types'
import { getWindowKeys } from './utils'
export default <Environment>({
name: 'jsdom',
async setup(global, { jsdom = {} }) {
const {
CookieJar,
JSDOM,
ResourceLoader,
VirtualConsole,
} = await importModule('jsdom') as typeof import('jsdom')
const {
html = '<!DOCTYPE html>',
userAgent,
url = 'http://localhost:3000',
contentType = 'text/html',
pretendToBeVisual = true,
includeNodeLocations = false,
runScripts = 'dangerously',
resources,
console = false,
cookieJar = false,
...restOptions
} = jsdom as JSDOMOptions
const dom = new JSDOM(
html,
{
pretendToBeVisual,
resources: resources ?? (userAgent ? new ResourceLoader({ userAgent }) : undefined),
runScripts,
url,
virtualConsole: console && global.console ? new VirtualConsole().sendTo(global.console) : undefined,
cookieJar: cookieJar ? new CookieJar() : undefined,
includeNodeLocations,
contentType,
userAgent,
...restOptions,
},
)
const keys = getWindowKeys(global, dom.window)
const overrideObject = new Map<string, any>()
for (const key of keys) {
Object.defineProperty(global, key, {
get() {
if (overrideObject.has(key))
return overrideObject.get(key)
return dom.window[key]
},
set(v) {
overrideObject.set(key, v)
},
configurable: true,
})
}
return {
teardown(global) {
keys.forEach(key => delete global[key])
},
}
},
})