jest-environment-webgpu-node 0.1.1 → 0.1.2

package/CHANGELOG.md

@@ -1,5 +1,5 @@
-## [jest-environment-webgpu-node-v0.1.1](https://github.com/bhouston/jest-gpu/compare/jest-environment-webgpu-node-v0.1.0...jest-environment-webgpu-node-v0.1.1) (2026-09-21)
+## [jest-environment-webgpu-node-v0.1.2](https://github.com/bhouston/jest-gpu/compare/jest-environment-webgpu-node-v0.1.1...jest-environment-webgpu-node-v0.1.2) (2026-09-21)
 
 ### Bug Fixes
 
-* **webgpu:** correct sandbox shims and pixel reads ([92c4118](https://github.com/bhouston/jest-gpu/commit/92c4118be3b36f30bb7a8b397de4e09d02035ab6)), closes [#19](https://github.com/bhouston/jest-gpu/issues/19)
+* **webgpu:** give the sandbox its own navigator and keep dawnOptions off the global ([34ff66c](https://github.com/bhouston/jest-gpu/commit/34ff66c360b602740198da938ee32d34bbd7b783)), closes [#21](https://github.com/bhouston/jest-gpu/issues/21)

package/dist/index.cjs

@@ -175,12 +175,20 @@ const domShims = (global, animationFrames) => {
 */
 var WebgpuEnvironment = class extends jest_environment_node.TestEnvironment {
 	#added = [];
-	#navigator = {};
+	#hostNavigator;
 	#nativeAdded;
 	#dawnOptions;
 	#animationFrames = /* @__PURE__ */ new Map();
 	constructor(config, context) {
-		super(config, context);
+		const { dawnOptions, ...testEnvironmentOptions } = config.projectConfig.testEnvironmentOptions;
+		super({
+			...config,
+			projectConfig: {
+				...config.projectConfig,
+				testEnvironmentOptions
+			}
+		}, context);
+		this.#dawnOptions = dawnOptions;
 		this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes("_native");
 		if (this.#nativeAdded) Object.defineProperty(this.global, "_native", {
 			value: void 0,
@@ -188,7 +196,6 @@ var WebgpuEnvironment = class extends jest_environment_node.TestEnvironment {
 			configurable: true,
 			enumerable: false
 		});
-		this.#dawnOptions = config.projectConfig.testEnvironmentOptions.dawnOptions;
 	}
 	async setup() {
 		await super.setup();
@@ -202,17 +209,25 @@ var WebgpuEnvironment = class extends jest_environment_node.TestEnvironment {
 		this.#added = Object.keys(shims).filter((key) => !(key in global));
 		for (const key of this.#added) global[key] = shims[key];
 		if (this.#nativeAdded) this.#added.push("_native");
-		this.#navigator = global.navigator ??= {};
-		Object.defineProperty(this.#navigator, "gpu", {
-			value: create(this.#dawnOptions ?? []),
-			configurable: true
+		const gpu = create(this.#dawnOptions ?? []);
+		const host = global.navigator ?? {};
+		this.#hostNavigator = Object.getOwnPropertyDescriptor(global, "navigator");
+		Object.defineProperty(global, "navigator", {
+			value: new Proxy(host, {
+				get: (target, key) => key === "gpu" ? gpu : Reflect.get(target, key, target),
+				has: (target, key) => key === "gpu" || Reflect.has(target, key)
+			}),
+			configurable: true,
+			enumerable: true,
+			writable: true
 		});
 	}
 	async teardown() {
 		const global = this.global;
 		for (const [handle, clear] of this.#animationFrames) clear(handle);
 		this.#animationFrames.clear();
-		delete this.#navigator.gpu;
+		if (this.#hostNavigator) Object.defineProperty(global, "navigator", this.#hostNavigator);
+		else delete global.navigator;
 		for (const key of this.#added) delete global[key];
 		await super.teardown();
 	}

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["TestEnvironment"],"sources":["../src/canvas.ts","../src/index.ts"],"sourcesContent":["/** RGBA8 pixels, top row first. Feed it to a screenshot matcher such as jest-image-snapshot after encoding to PNG. */\nexport type RgbaImage = { width: number; height: number; data: Uint8Array };\n\nconst COPY_SRC = 0x01;\nconst RENDER_ATTACHMENT = 0x10;\nconst MAP_READ = 0x01;\nconst COPY_DST = 0x08;\nconst READABLE_FORMATS = new Set<GPUTextureFormat>(['rgba8unorm', 'rgba8unorm-srgb', 'bgra8unorm', 'bgra8unorm-srgb']);\n\n/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */\nexport class HeadlessCanvasContext {\n  #config: GPUCanvasConfiguration | undefined;\n  #texture: GPUTexture | undefined;\n\n  constructor(readonly canvas: HeadlessCanvas) {}\n\n  configure(config: GPUCanvasConfiguration): void {\n    this.#config = config;\n    this.#drop();\n  }\n\n  unconfigure(): void {\n    this.#config = undefined;\n    this.#drop();\n  }\n\n  getConfiguration(): GPUCanvasConfiguration | null {\n    return this.#config ?? null;\n  }\n\n  getCurrentTexture(): GPUTexture {\n    const config = this.#config;\n    if (!config) throw new Error('getCurrentTexture() called before configure()');\n    const { width, height } = this.canvas;\n    if (this.#texture && (this.#texture.width !== width || this.#texture.height !== height)) this.#drop();\n    this.#texture ??= config.device.createTexture({\n      size: [width, height],\n      format: config.format,\n      usage: (config.usage ?? RENDER_ATTACHMENT) | RENDER_ATTACHMENT | COPY_SRC,\n      viewFormats: config.viewFormats,\n    });\n    return this.#texture;\n  }\n\n  /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */\n  async readPixels(): Promise<RgbaImage> {\n    const config = this.#config;\n    if (!config) throw new Error('readPixels() called before configure()');\n    const { device, format } = config;\n    if (!READABLE_FORMATS.has(format)) {\n      throw new Error(`readPixels() supports only 8-bit RGBA/BGRA formats; received ${format}`);\n    }\n    const texture = this.getCurrentTexture();\n    const { width, height } = texture;\n    const bytesPerRow = Math.ceil((width * 4) / 256) * 256;\n    const buffer = device.createBuffer({ size: bytesPerRow * height, usage: MAP_READ | COPY_DST });\n    try {\n      const encoder = device.createCommandEncoder();\n      encoder.copyTextureToBuffer({ texture }, { buffer, bytesPerRow }, [width, height]);\n      device.queue.submit([encoder.finish()]);\n      await buffer.mapAsync(MAP_READ);\n      const padded = new Uint8Array(buffer.getMappedRange());\n      const data = new Uint8Array(width * height * 4);\n      for (let y = 0; y < height; y++)\n        data.set(padded.subarray(y * bytesPerRow, y * bytesPerRow + width * 4), y * width * 4);\n      if (format.startsWith('bgra')) {\n        for (let i = 0; i < data.length; i += 4) [data[i], data[i + 2]] = [data[i + 2]!, data[i]!];\n      }\n      return { width, height, data };\n    } finally {\n      if (buffer.mapState === 'mapped') buffer.unmap();\n      buffer.destroy();\n    }\n  }\n\n  #drop(): void {\n    this.#texture?.destroy();\n    this.#texture = undefined;\n  }\n}\n\n/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */\nexport class HeadlessCanvas {\n  readonly #context = new HeadlessCanvasContext(this);\n  readonly style: Record<string, string> = {};\n  readonly #attributes = new Map<string, string>();\n\n  constructor(\n    public width = 300,\n    public height = 150,\n  ) {}\n\n  get clientWidth(): number {\n    return this.width;\n  }\n\n  get clientHeight(): number {\n    return this.height;\n  }\n\n  getContext(id: 'webgpu'): HeadlessCanvasContext;\n  getContext(id: string): HeadlessCanvasContext | null;\n  getContext(id: string): HeadlessCanvasContext | null {\n    return id === 'webgpu' ? this.#context : null;\n  }\n\n  readPixels(): Promise<RgbaImage> {\n    return this.#context.readPixels();\n  }\n\n  /** This same object, typed as an element for libraries whose signatures demand one. */\n  asElement(): HTMLCanvasElement {\n    return this as unknown as HTMLCanvasElement;\n  }\n\n  getBoundingClientRect(): { x: number; y: number; width: number; height: number; top: number; left: number } {\n    return { x: 0, y: 0, top: 0, left: 0, width: this.width, height: this.height };\n  }\n\n  setAttribute(name: string, value: string): void {\n    this.#attributes.set(name, value);\n  }\n\n  getAttribute(name: string): string | null {\n    return this.#attributes.get(name) ?? null;\n  }\n\n  addEventListener(): void {}\n  removeEventListener(): void {}\n  dispatchEvent(): boolean {\n    return true;\n  }\n}\n\nexport const createCanvas = (width: number, height: number): HeadlessCanvas => new HeadlessCanvas(width, height);\n","import { TestEnvironment } from 'jest-environment-node';\nimport { createCanvas, HeadlessCanvas } from './canvas.js';\n\ntype JestEnvironmentConfig = ConstructorParameters<typeof TestEnvironment>[0];\ntype EnvironmentContext = ConstructorParameters<typeof TestEnvironment>[1];\n\nexport { createCanvas, HeadlessCanvas, HeadlessCanvasContext, type RgbaImage } from './canvas.js';\n\n/** Set under `testEnvironmentOptions` in your Jest config. */\nexport type WebgpuNodeOptions = {\n  /** Dawn options, e.g. `['backend=vulkan', 'enable-dawn-features=allow_unsafe_apis']`. */\n  dawnOptions?: string[];\n};\n\n/** Just enough DOM for libraries that make their own canvas and drive a frame loop. Only installed when missing. */\ntype TimerHandle = ReturnType<typeof setTimeout>;\ntype SandboxGlobal = Record<string, unknown> & {\n  clearTimeout: typeof clearTimeout;\n  performance: Performance;\n  setTimeout: typeof setTimeout;\n};\n\nconst domShims = (\n  global: SandboxGlobal,\n  animationFrames: Map<TimerHandle, typeof clearTimeout>,\n): Record<string, unknown> => {\n  const requestAnimationFrame = (callback: (time: number) => void): TimerHandle => {\n    const handle = global.setTimeout(() => {\n      animationFrames.delete(handle);\n      callback(global.performance.now());\n    }, 16);\n    animationFrames.set(handle, global.clearTimeout);\n    (handle as NodeJS.Timeout).unref?.();\n    return handle;\n  };\n  const cancelAnimationFrame = (handle: TimerHandle): void => {\n    const clear = animationFrames.get(handle) ?? global.clearTimeout;\n    animationFrames.delete(handle);\n    clear(handle);\n  };\n  const document = {\n    createElement: (tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    createElementNS: (_ns: string, tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    addEventListener() {},\n    removeEventListener() {},\n  };\n  return {\n    document,\n    devicePixelRatio: 1,\n    requestAnimationFrame,\n    cancelAnimationFrame,\n    addEventListener() {},\n    removeEventListener() {},\n    HTMLCanvasElement: HeadlessCanvas,\n    window: global,\n    self: global,\n  };\n};\n\n/**\n * Jest environment that puts real headless WebGPU (Dawn via the `webgpu` package) on the sandbox\n * global: `navigator.gpu`, every `GPU*` class and constant, and a `HeadlessCanvas` shim. CommonJS\n * tests can rely on `createCanvas` being a global; ESM tests can also `import` it from this package.\n */\nexport default class WebgpuEnvironment extends TestEnvironment {\n  #added: string[] = [];\n  #navigator: { gpu?: GPU } = {};\n  #nativeAdded: boolean;\n  #dawnOptions: string[] | undefined;\n  #animationFrames = new Map<TimerHandle, typeof clearTimeout>();\n\n  constructor(config: JestEnvironmentConfig, context: EnvironmentContext) {\n    super(config, context);\n    // @babylonjs/core defines a `_native` accessor on `self` at module load; on Jest's proxied\n    // sandbox global that defineProperty violates a proxy invariant, so pre-seed the property to\n    // satisfy Babylon's hasOwnProperty guard.\n    // `Object.hasOwn`/`in` report false negatives here: jest-environment-node's sandbox global is\n    // a Proxy over a vm context, and reads of freshly-added properties from outside the sandbox\n    // (as this constructor runs) don't go through the fast path those use. getOwnPropertyNames is\n    // unaffected, so use that instead.\n    this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes('_native');\n    if (this.#nativeAdded) {\n      Object.defineProperty(this.global, '_native', {\n        value: undefined,\n        writable: true,\n        configurable: true,\n        enumerable: false,\n      });\n    }\n    this.#dawnOptions = config.projectConfig.testEnvironmentOptions.dawnOptions as string[] | undefined;\n  }\n\n  // `webgpu` is ESM-only. Loading it via a dynamic import here (rather than a static top-level\n  // import) keeps it out of this module's own top-level module graph, so a CommonJS test file's\n  // `require('jest-environment-webgpu-node')` - which only needs `createCanvas` - never forces\n  // Jest's sandboxed CJS loader to load an ESM-only dependency (jestjs/jest#15716). This method\n  // itself runs outside that sandbox, through Jest's own environment setup, where Node's native\n  // `import()` always works.\n  override async setup(): Promise<void> {\n    await super.setup();\n    const { create, globals } = await import('webgpu');\n    const global = this.global as unknown as SandboxGlobal;\n    const shims = { ...globals, ...domShims(global, this.#animationFrames), createCanvas } as Record<string, unknown>;\n    this.#added = Object.keys(shims).filter((key) => !(key in global));\n    for (const key of this.#added) global[key] = shims[key];\n    if (this.#nativeAdded) this.#added.push('_native');\n    this.#navigator = (global.navigator ??= {}) as { gpu?: GPU };\n    Object.defineProperty(this.#navigator, 'gpu', { value: create(this.#dawnOptions ?? []), configurable: true });\n  }\n\n  override async teardown(): Promise<void> {\n    const global = this.global as unknown as Record<string, unknown>;\n    for (const [handle, clear] of this.#animationFrames) clear(handle);\n    this.#animationFrames.clear();\n    delete this.#navigator.gpu;\n    for (const key of this.#added) delete global[key];\n    await super.teardown();\n  }\n}\n"],"mappings":";;;;;;AAIA,MAAM,oBAAoB;AAC1B,MAAM,WAAW;AAEjB,MAAM,mCAAmB,IAAI,IAAsB;CAAC;CAAc;CAAmB;CAAc;AAAiB,CAAC;;AAGrH,IAAa,wBAAb,MAAmC;CAIZ;CAHrB;CACA;CAEA,YAAY,QAAiC;EAAxB,KAAA,SAAA;CAAyB;CAE9C,UAAU,QAAsC;EAC9C,KAAK,UAAU;EACf,KAAK,MAAM;CACb;CAEA,cAAoB;EAClB,KAAK,UAAU,KAAA;EACf,KAAK,MAAM;CACb;CAEA,mBAAkD;EAChD,OAAO,KAAK,WAAW;CACzB;CAEA,oBAAgC;EAC9B,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,+CAA+C;EAC5E,MAAM,EAAE,OAAO,WAAW,KAAK;EAC/B,IAAI,KAAK,aAAa,KAAK,SAAS,UAAU,SAAS,KAAK,SAAS,WAAW,SAAS,KAAK,MAAM;EACpG,KAAK,aAAa,OAAO,OAAO,cAAc;GAC5C,MAAM,CAAC,OAAO,MAAM;GACpB,QAAQ,OAAO;GACf,QAAQ,OAAO,SAAS,qBAAqB;GAC7C,aAAa,OAAO;EACtB,CAAC;EACD,OAAO,KAAK;CACd;;CAGA,MAAM,aAAiC;EACrC,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,wCAAwC;EACrE,MAAM,EAAE,QAAQ,WAAW;EAC3B,IAAI,CAAC,iBAAiB,IAAI,MAAM,GAC9B,MAAM,IAAI,MAAM,gEAAgE,QAAQ;EAE1F,MAAM,UAAU,KAAK,kBAAkB;EACvC,MAAM,EAAE,OAAO,WAAW;EAC1B,MAAM,cAAc,KAAK,KAAM,QAAQ,IAAK,GAAG,IAAI;EACnD,MAAM,SAAS,OAAO,aAAa;GAAE,MAAM,cAAc;GAAQ,OAAO;EAAoB,CAAC;EAC7F,IAAI;GACF,MAAM,UAAU,OAAO,qBAAqB;GAC5C,QAAQ,oBAAoB,EAAE,QAAQ,GAAG;IAAE;IAAQ;GAAY,GAAG,CAAC,OAAO,MAAM,CAAC;GACjF,OAAO,MAAM,OAAO,CAAC,QAAQ,OAAO,CAAC,CAAC;GACtC,MAAM,OAAO,SAAS,QAAQ;GAC9B,MAAM,SAAS,IAAI,WAAW,OAAO,eAAe,CAAC;GACrD,MAAM,OAAO,IAAI,WAAW,QAAQ,SAAS,CAAC;GAC9C,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAC1B,KAAK,IAAI,OAAO,SAAS,IAAI,aAAa,IAAI,cAAc,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC;GACvF,IAAI,OAAO,WAAW,MAAM,GAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK,GAAG,CAAC,KAAK,IAAI,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,IAAK,KAAK,EAAG;GAE3F,OAAO;IAAE;IAAO;IAAQ;GAAK;EAC/B,UAAU;GACR,IAAI,OAAO,aAAa,UAAU,OAAO,MAAM;GAC/C,OAAO,QAAQ;EACjB;CACF;CAEA,QAAc;EACZ,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,KAAA;CAClB;AACF;;AAGA,IAAa,iBAAb,MAA4B;CAMjB;CACA;CANT,WAAoB,IAAI,sBAAsB,IAAI;CAClD,QAAyC,CAAC;CAC1C,8BAAuB,IAAI,IAAoB;CAE/C,YACE,QAAe,KACf,SAAgB,KAChB;EAFO,KAAA,QAAA;EACA,KAAA,SAAA;CACN;CAEH,IAAI,cAAsB;EACxB,OAAO,KAAK;CACd;CAEA,IAAI,eAAuB;EACzB,OAAO,KAAK;CACd;CAIA,WAAW,IAA0C;EACnD,OAAO,OAAO,WAAW,KAAK,WAAW;CAC3C;CAEA,aAAiC;EAC/B,OAAO,KAAK,SAAS,WAAW;CAClC;;CAGA,YAA+B;EAC7B,OAAO;CACT;CAEA,wBAA4G;EAC1G,OAAO;GAAE,GAAG;GAAG,GAAG;GAAG,KAAK;GAAG,MAAM;GAAG,OAAO,KAAK;GAAO,QAAQ,KAAK;EAAO;CAC/E;CAEA,aAAa,MAAc,OAAqB;EAC9C,KAAK,YAAY,IAAI,MAAM,KAAK;CAClC;CAEA,aAAa,MAA6B;EACxC,OAAO,KAAK,YAAY,IAAI,IAAI,KAAK;CACvC;CAEA,mBAAyB,CAAC;CAC1B,sBAA4B,CAAC;CAC7B,gBAAyB;EACvB,OAAO;CACT;AACF;AAEA,MAAa,gBAAgB,OAAe,WAAmC,IAAI,eAAe,OAAO,MAAM;;;AChH/G,MAAM,YACJ,QACA,oBAC4B;CAC5B,MAAM,yBAAyB,aAAkD;EAC/E,MAAM,SAAS,OAAO,iBAAiB;GACrC,gBAAgB,OAAO,MAAM;GAC7B,SAAS,OAAO,YAAY,IAAI,CAAC;EACnC,GAAG,EAAE;EACL,gBAAgB,IAAI,QAAQ,OAAO,YAAY;EAC/C,OAA2B,QAAQ;EACnC,OAAO;CACT;CACA,MAAM,wBAAwB,WAA8B;EAC1D,MAAM,QAAQ,gBAAgB,IAAI,MAAM,KAAK,OAAO;EACpD,gBAAgB,OAAO,MAAM;EAC7B,MAAM,MAAM;CACd;CAOA,OAAO;EACL,UAAA;GANA,gBAAgB,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC5E,kBAAkB,KAAa,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC3F,mBAAmB,CAAC;GACpB,sBAAsB,CAAC;EAGhB;EACP,kBAAkB;EAClB;EACA;EACA,mBAAmB,CAAC;EACpB,sBAAsB,CAAC;EACvB,mBAAmB;EACnB,QAAQ;EACR,MAAM;CACR;AACF;;;;;;AAOA,IAAqB,oBAArB,cAA+CA,sBAAAA,gBAAgB;CAC7D,SAAmB,CAAC;CACpB,aAA4B,CAAC;CAC7B;CACA;CACA,mCAAmB,IAAI,IAAsC;CAE7D,YAAY,QAA+B,SAA6B;EACtE,MAAM,QAAQ,OAAO;EAQrB,KAAK,eAAe,CAAC,OAAO,oBAAoB,KAAK,MAAM,CAAC,CAAC,SAAS,SAAS;EAC/E,IAAI,KAAK,cACP,OAAO,eAAe,KAAK,QAAQ,WAAW;GAC5C,OAAO,KAAA;GACP,UAAU;GACV,cAAc;GACd,YAAY;EACd,CAAC;EAEH,KAAK,eAAe,OAAO,cAAc,uBAAuB;CAClE;CAQA,MAAe,QAAuB;EACpC,MAAM,MAAM,MAAM;EAClB,MAAM,EAAE,QAAQ,YAAY,MAAM,OAAO;EACzC,MAAM,SAAS,KAAK;EACpB,MAAM,QAAQ;GAAE,GAAG;GAAS,GAAG,SAAS,QAAQ,KAAK,gBAAgB;GAAG;EAAa;EACrF,KAAK,SAAS,OAAO,KAAK,KAAK,CAAC,CAAC,QAAQ,QAAQ,EAAE,OAAO,OAAO;EACjE,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO,MAAM;EACnD,IAAI,KAAK,cAAc,KAAK,OAAO,KAAK,SAAS;EACjD,KAAK,aAAc,OAAO,cAAc,CAAC;EACzC,OAAO,eAAe,KAAK,YAAY,OAAO;GAAE,OAAO,OAAO,KAAK,gBAAgB,CAAC,CAAC;GAAG,cAAc;EAAK,CAAC;CAC9G;CAEA,MAAe,WAA0B;EACvC,MAAM,SAAS,KAAK;EACpB,KAAK,MAAM,CAAC,QAAQ,UAAU,KAAK,kBAAkB,MAAM,MAAM;EACjE,KAAK,iBAAiB,MAAM;EAC5B,OAAO,KAAK,WAAW;EACvB,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO;EAC7C,MAAM,MAAM,SAAS;CACvB;AACF"}
+{"version":3,"file":"index.cjs","names":["TestEnvironment"],"sources":["../src/canvas.ts","../src/index.ts"],"sourcesContent":["/** RGBA8 pixels, top row first. Feed it to a screenshot matcher such as jest-image-snapshot after encoding to PNG. */\nexport type RgbaImage = { width: number; height: number; data: Uint8Array };\n\nconst COPY_SRC = 0x01;\nconst RENDER_ATTACHMENT = 0x10;\nconst MAP_READ = 0x01;\nconst COPY_DST = 0x08;\nconst READABLE_FORMATS = new Set<GPUTextureFormat>(['rgba8unorm', 'rgba8unorm-srgb', 'bgra8unorm', 'bgra8unorm-srgb']);\n\n/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */\nexport class HeadlessCanvasContext {\n  #config: GPUCanvasConfiguration | undefined;\n  #texture: GPUTexture | undefined;\n\n  constructor(readonly canvas: HeadlessCanvas) {}\n\n  configure(config: GPUCanvasConfiguration): void {\n    this.#config = config;\n    this.#drop();\n  }\n\n  unconfigure(): void {\n    this.#config = undefined;\n    this.#drop();\n  }\n\n  getConfiguration(): GPUCanvasConfiguration | null {\n    return this.#config ?? null;\n  }\n\n  getCurrentTexture(): GPUTexture {\n    const config = this.#config;\n    if (!config) throw new Error('getCurrentTexture() called before configure()');\n    const { width, height } = this.canvas;\n    if (this.#texture && (this.#texture.width !== width || this.#texture.height !== height)) this.#drop();\n    this.#texture ??= config.device.createTexture({\n      size: [width, height],\n      format: config.format,\n      usage: (config.usage ?? RENDER_ATTACHMENT) | RENDER_ATTACHMENT | COPY_SRC,\n      viewFormats: config.viewFormats,\n    });\n    return this.#texture;\n  }\n\n  /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */\n  async readPixels(): Promise<RgbaImage> {\n    const config = this.#config;\n    if (!config) throw new Error('readPixels() called before configure()');\n    const { device, format } = config;\n    if (!READABLE_FORMATS.has(format)) {\n      throw new Error(`readPixels() supports only 8-bit RGBA/BGRA formats; received ${format}`);\n    }\n    const texture = this.getCurrentTexture();\n    const { width, height } = texture;\n    const bytesPerRow = Math.ceil((width * 4) / 256) * 256;\n    const buffer = device.createBuffer({ size: bytesPerRow * height, usage: MAP_READ | COPY_DST });\n    try {\n      const encoder = device.createCommandEncoder();\n      encoder.copyTextureToBuffer({ texture }, { buffer, bytesPerRow }, [width, height]);\n      device.queue.submit([encoder.finish()]);\n      await buffer.mapAsync(MAP_READ);\n      const padded = new Uint8Array(buffer.getMappedRange());\n      const data = new Uint8Array(width * height * 4);\n      for (let y = 0; y < height; y++)\n        data.set(padded.subarray(y * bytesPerRow, y * bytesPerRow + width * 4), y * width * 4);\n      if (format.startsWith('bgra')) {\n        for (let i = 0; i < data.length; i += 4) [data[i], data[i + 2]] = [data[i + 2]!, data[i]!];\n      }\n      return { width, height, data };\n    } finally {\n      if (buffer.mapState === 'mapped') buffer.unmap();\n      buffer.destroy();\n    }\n  }\n\n  #drop(): void {\n    this.#texture?.destroy();\n    this.#texture = undefined;\n  }\n}\n\n/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */\nexport class HeadlessCanvas {\n  readonly #context = new HeadlessCanvasContext(this);\n  readonly style: Record<string, string> = {};\n  readonly #attributes = new Map<string, string>();\n\n  constructor(\n    public width = 300,\n    public height = 150,\n  ) {}\n\n  get clientWidth(): number {\n    return this.width;\n  }\n\n  get clientHeight(): number {\n    return this.height;\n  }\n\n  getContext(id: 'webgpu'): HeadlessCanvasContext;\n  getContext(id: string): HeadlessCanvasContext | null;\n  getContext(id: string): HeadlessCanvasContext | null {\n    return id === 'webgpu' ? this.#context : null;\n  }\n\n  readPixels(): Promise<RgbaImage> {\n    return this.#context.readPixels();\n  }\n\n  /** This same object, typed as an element for libraries whose signatures demand one. */\n  asElement(): HTMLCanvasElement {\n    return this as unknown as HTMLCanvasElement;\n  }\n\n  getBoundingClientRect(): { x: number; y: number; width: number; height: number; top: number; left: number } {\n    return { x: 0, y: 0, top: 0, left: 0, width: this.width, height: this.height };\n  }\n\n  setAttribute(name: string, value: string): void {\n    this.#attributes.set(name, value);\n  }\n\n  getAttribute(name: string): string | null {\n    return this.#attributes.get(name) ?? null;\n  }\n\n  addEventListener(): void {}\n  removeEventListener(): void {}\n  dispatchEvent(): boolean {\n    return true;\n  }\n}\n\nexport const createCanvas = (width: number, height: number): HeadlessCanvas => new HeadlessCanvas(width, height);\n","import { TestEnvironment } from 'jest-environment-node';\nimport { createCanvas, HeadlessCanvas } from './canvas.js';\n\ntype JestEnvironmentConfig = ConstructorParameters<typeof TestEnvironment>[0];\ntype EnvironmentContext = ConstructorParameters<typeof TestEnvironment>[1];\n\nexport { createCanvas, HeadlessCanvas, HeadlessCanvasContext, type RgbaImage } from './canvas.js';\n\n/** Set under `testEnvironmentOptions` in your Jest config. */\nexport type WebgpuNodeOptions = {\n  /** Dawn options, e.g. `['backend=vulkan', 'enable-dawn-features=allow_unsafe_apis']`. */\n  dawnOptions?: string[];\n};\n\n/** Just enough DOM for libraries that make their own canvas and drive a frame loop. Only installed when missing. */\ntype TimerHandle = ReturnType<typeof setTimeout>;\ntype SandboxGlobal = Record<string, unknown> & {\n  clearTimeout: typeof clearTimeout;\n  performance: Performance;\n  setTimeout: typeof setTimeout;\n};\n\nconst domShims = (\n  global: SandboxGlobal,\n  animationFrames: Map<TimerHandle, typeof clearTimeout>,\n): Record<string, unknown> => {\n  const requestAnimationFrame = (callback: (time: number) => void): TimerHandle => {\n    const handle = global.setTimeout(() => {\n      animationFrames.delete(handle);\n      callback(global.performance.now());\n    }, 16);\n    animationFrames.set(handle, global.clearTimeout);\n    (handle as NodeJS.Timeout).unref?.();\n    return handle;\n  };\n  const cancelAnimationFrame = (handle: TimerHandle): void => {\n    const clear = animationFrames.get(handle) ?? global.clearTimeout;\n    animationFrames.delete(handle);\n    clear(handle);\n  };\n  const document = {\n    createElement: (tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    createElementNS: (_ns: string, tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    addEventListener() {},\n    removeEventListener() {},\n  };\n  return {\n    document,\n    devicePixelRatio: 1,\n    requestAnimationFrame,\n    cancelAnimationFrame,\n    addEventListener() {},\n    removeEventListener() {},\n    HTMLCanvasElement: HeadlessCanvas,\n    window: global,\n    self: global,\n  };\n};\n\n/**\n * Jest environment that puts real headless WebGPU (Dawn via the `webgpu` package) on the sandbox\n * global: `navigator.gpu`, every `GPU*` class and constant, and a `HeadlessCanvas` shim. CommonJS\n * tests can rely on `createCanvas` being a global; ESM tests can also `import` it from this package.\n */\nexport default class WebgpuEnvironment extends TestEnvironment {\n  #added: string[] = [];\n  #hostNavigator: PropertyDescriptor | undefined;\n  #nativeAdded: boolean;\n  #dawnOptions: string[] | undefined;\n  #animationFrames = new Map<TimerHandle, typeof clearTimeout>();\n\n  constructor(config: JestEnvironmentConfig, context: EnvironmentContext) {\n    // jest-environment-node assigns every `testEnvironmentOptions` entry onto the sandbox global, so\n    // strip this environment's own option first to keep `dawnOptions` off `globalThis`.\n    const { dawnOptions, ...testEnvironmentOptions } = config.projectConfig.testEnvironmentOptions;\n    super({ ...config, projectConfig: { ...config.projectConfig, testEnvironmentOptions } }, context);\n    this.#dawnOptions = dawnOptions as string[] | undefined;\n    // @babylonjs/core defines a `_native` accessor on `self` at module load; on Jest's proxied\n    // sandbox global that defineProperty violates a proxy invariant, so pre-seed the property to\n    // satisfy Babylon's hasOwnProperty guard.\n    // `Object.hasOwn`/`in` report false negatives here: jest-environment-node's sandbox global is\n    // a Proxy over a vm context, and reads of freshly-added properties from outside the sandbox\n    // (as this constructor runs) don't go through the fast path those use. getOwnPropertyNames is\n    // unaffected, so use that instead.\n    this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes('_native');\n    if (this.#nativeAdded) {\n      Object.defineProperty(this.global, '_native', {\n        value: undefined,\n        writable: true,\n        configurable: true,\n        enumerable: false,\n      });\n    }\n  }\n\n  // `webgpu` is ESM-only. Loading it via a dynamic import here (rather than a static top-level\n  // import) keeps it out of this module's own top-level module graph, so a CommonJS test file's\n  // `require('jest-environment-webgpu-node')` - which only needs `createCanvas` - never forces\n  // Jest's sandboxed CJS loader to load an ESM-only dependency (jestjs/jest#15716). This method\n  // itself runs outside that sandbox, through Jest's own environment setup, where Node's native\n  // `import()` always works.\n  override async setup(): Promise<void> {\n    await super.setup();\n    const { create, globals } = await import('webgpu');\n    const global = this.global as unknown as SandboxGlobal;\n    const shims = { ...globals, ...domShims(global, this.#animationFrames), createCanvas } as Record<string, unknown>;\n    this.#added = Object.keys(shims).filter((key) => !(key in global));\n    for (const key of this.#added) global[key] = shims[key];\n    if (this.#nativeAdded) this.#added.push('_native');\n    // jest-environment-node hands the sandbox Node's own `navigator` by reference, so adding `gpu` to it\n    // would leak onto the host process and be shared by every environment in this worker. Give the\n    // sandbox its own proxy instead; `Reflect.get(target, key, target)` keeps Node's private-field\n    // getters (hardwareConcurrency, userAgent, ...) working through it.\n    const gpu = create(this.#dawnOptions ?? []);\n    const host = (global.navigator ?? {}) as object;\n    this.#hostNavigator = Object.getOwnPropertyDescriptor(global, 'navigator');\n    Object.defineProperty(global, 'navigator', {\n      value: new Proxy(host, {\n        get: (target, key) => (key === 'gpu' ? gpu : Reflect.get(target, key, target)),\n        has: (target, key) => key === 'gpu' || Reflect.has(target, key),\n      }),\n      configurable: true,\n      enumerable: true,\n      writable: true,\n    });\n  }\n\n  override async teardown(): Promise<void> {\n    const global = this.global as unknown as Record<string, unknown>;\n    for (const [handle, clear] of this.#animationFrames) clear(handle);\n    this.#animationFrames.clear();\n    if (this.#hostNavigator) Object.defineProperty(global, 'navigator', this.#hostNavigator);\n    else delete global.navigator;\n    for (const key of this.#added) delete global[key];\n    await super.teardown();\n  }\n}\n"],"mappings":";;;;;;AAIA,MAAM,oBAAoB;AAC1B,MAAM,WAAW;AAEjB,MAAM,mCAAmB,IAAI,IAAsB;CAAC;CAAc;CAAmB;CAAc;AAAiB,CAAC;;AAGrH,IAAa,wBAAb,MAAmC;CAIZ;CAHrB;CACA;CAEA,YAAY,QAAiC;EAAxB,KAAA,SAAA;CAAyB;CAE9C,UAAU,QAAsC;EAC9C,KAAK,UAAU;EACf,KAAK,MAAM;CACb;CAEA,cAAoB;EAClB,KAAK,UAAU,KAAA;EACf,KAAK,MAAM;CACb;CAEA,mBAAkD;EAChD,OAAO,KAAK,WAAW;CACzB;CAEA,oBAAgC;EAC9B,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,+CAA+C;EAC5E,MAAM,EAAE,OAAO,WAAW,KAAK;EAC/B,IAAI,KAAK,aAAa,KAAK,SAAS,UAAU,SAAS,KAAK,SAAS,WAAW,SAAS,KAAK,MAAM;EACpG,KAAK,aAAa,OAAO,OAAO,cAAc;GAC5C,MAAM,CAAC,OAAO,MAAM;GACpB,QAAQ,OAAO;GACf,QAAQ,OAAO,SAAS,qBAAqB;GAC7C,aAAa,OAAO;EACtB,CAAC;EACD,OAAO,KAAK;CACd;;CAGA,MAAM,aAAiC;EACrC,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,wCAAwC;EACrE,MAAM,EAAE,QAAQ,WAAW;EAC3B,IAAI,CAAC,iBAAiB,IAAI,MAAM,GAC9B,MAAM,IAAI,MAAM,gEAAgE,QAAQ;EAE1F,MAAM,UAAU,KAAK,kBAAkB;EACvC,MAAM,EAAE,OAAO,WAAW;EAC1B,MAAM,cAAc,KAAK,KAAM,QAAQ,IAAK,GAAG,IAAI;EACnD,MAAM,SAAS,OAAO,aAAa;GAAE,MAAM,cAAc;GAAQ,OAAO;EAAoB,CAAC;EAC7F,IAAI;GACF,MAAM,UAAU,OAAO,qBAAqB;GAC5C,QAAQ,oBAAoB,EAAE,QAAQ,GAAG;IAAE;IAAQ;GAAY,GAAG,CAAC,OAAO,MAAM,CAAC;GACjF,OAAO,MAAM,OAAO,CAAC,QAAQ,OAAO,CAAC,CAAC;GACtC,MAAM,OAAO,SAAS,QAAQ;GAC9B,MAAM,SAAS,IAAI,WAAW,OAAO,eAAe,CAAC;GACrD,MAAM,OAAO,IAAI,WAAW,QAAQ,SAAS,CAAC;GAC9C,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAC1B,KAAK,IAAI,OAAO,SAAS,IAAI,aAAa,IAAI,cAAc,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC;GACvF,IAAI,OAAO,WAAW,MAAM,GAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK,GAAG,CAAC,KAAK,IAAI,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,IAAK,KAAK,EAAG;GAE3F,OAAO;IAAE;IAAO;IAAQ;GAAK;EAC/B,UAAU;GACR,IAAI,OAAO,aAAa,UAAU,OAAO,MAAM;GAC/C,OAAO,QAAQ;EACjB;CACF;CAEA,QAAc;EACZ,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,KAAA;CAClB;AACF;;AAGA,IAAa,iBAAb,MAA4B;CAMjB;CACA;CANT,WAAoB,IAAI,sBAAsB,IAAI;CAClD,QAAyC,CAAC;CAC1C,8BAAuB,IAAI,IAAoB;CAE/C,YACE,QAAe,KACf,SAAgB,KAChB;EAFO,KAAA,QAAA;EACA,KAAA,SAAA;CACN;CAEH,IAAI,cAAsB;EACxB,OAAO,KAAK;CACd;CAEA,IAAI,eAAuB;EACzB,OAAO,KAAK;CACd;CAIA,WAAW,IAA0C;EACnD,OAAO,OAAO,WAAW,KAAK,WAAW;CAC3C;CAEA,aAAiC;EAC/B,OAAO,KAAK,SAAS,WAAW;CAClC;;CAGA,YAA+B;EAC7B,OAAO;CACT;CAEA,wBAA4G;EAC1G,OAAO;GAAE,GAAG;GAAG,GAAG;GAAG,KAAK;GAAG,MAAM;GAAG,OAAO,KAAK;GAAO,QAAQ,KAAK;EAAO;CAC/E;CAEA,aAAa,MAAc,OAAqB;EAC9C,KAAK,YAAY,IAAI,MAAM,KAAK;CAClC;CAEA,aAAa,MAA6B;EACxC,OAAO,KAAK,YAAY,IAAI,IAAI,KAAK;CACvC;CAEA,mBAAyB,CAAC;CAC1B,sBAA4B,CAAC;CAC7B,gBAAyB;EACvB,OAAO;CACT;AACF;AAEA,MAAa,gBAAgB,OAAe,WAAmC,IAAI,eAAe,OAAO,MAAM;;;AChH/G,MAAM,YACJ,QACA,oBAC4B;CAC5B,MAAM,yBAAyB,aAAkD;EAC/E,MAAM,SAAS,OAAO,iBAAiB;GACrC,gBAAgB,OAAO,MAAM;GAC7B,SAAS,OAAO,YAAY,IAAI,CAAC;EACnC,GAAG,EAAE;EACL,gBAAgB,IAAI,QAAQ,OAAO,YAAY;EAC/C,OAA2B,QAAQ;EACnC,OAAO;CACT;CACA,MAAM,wBAAwB,WAA8B;EAC1D,MAAM,QAAQ,gBAAgB,IAAI,MAAM,KAAK,OAAO;EACpD,gBAAgB,OAAO,MAAM;EAC7B,MAAM,MAAM;CACd;CAOA,OAAO;EACL,UAAA;GANA,gBAAgB,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC5E,kBAAkB,KAAa,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC3F,mBAAmB,CAAC;GACpB,sBAAsB,CAAC;EAGhB;EACP,kBAAkB;EAClB;EACA;EACA,mBAAmB,CAAC;EACpB,sBAAsB,CAAC;EACvB,mBAAmB;EACnB,QAAQ;EACR,MAAM;CACR;AACF;;;;;;AAOA,IAAqB,oBAArB,cAA+CA,sBAAAA,gBAAgB;CAC7D,SAAmB,CAAC;CACpB;CACA;CACA;CACA,mCAAmB,IAAI,IAAsC;CAE7D,YAAY,QAA+B,SAA6B;EAGtE,MAAM,EAAE,aAAa,GAAG,2BAA2B,OAAO,cAAc;EACxE,MAAM;GAAE,GAAG;GAAQ,eAAe;IAAE,GAAG,OAAO;IAAe;GAAuB;EAAE,GAAG,OAAO;EAChG,KAAK,eAAe;EAQpB,KAAK,eAAe,CAAC,OAAO,oBAAoB,KAAK,MAAM,CAAC,CAAC,SAAS,SAAS;EAC/E,IAAI,KAAK,cACP,OAAO,eAAe,KAAK,QAAQ,WAAW;GAC5C,OAAO,KAAA;GACP,UAAU;GACV,cAAc;GACd,YAAY;EACd,CAAC;CAEL;CAQA,MAAe,QAAuB;EACpC,MAAM,MAAM,MAAM;EAClB,MAAM,EAAE,QAAQ,YAAY,MAAM,OAAO;EACzC,MAAM,SAAS,KAAK;EACpB,MAAM,QAAQ;GAAE,GAAG;GAAS,GAAG,SAAS,QAAQ,KAAK,gBAAgB;GAAG;EAAa;EACrF,KAAK,SAAS,OAAO,KAAK,KAAK,CAAC,CAAC,QAAQ,QAAQ,EAAE,OAAO,OAAO;EACjE,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO,MAAM;EACnD,IAAI,KAAK,cAAc,KAAK,OAAO,KAAK,SAAS;EAKjD,MAAM,MAAM,OAAO,KAAK,gBAAgB,CAAC,CAAC;EAC1C,MAAM,OAAQ,OAAO,aAAa,CAAC;EACnC,KAAK,iBAAiB,OAAO,yBAAyB,QAAQ,WAAW;EACzE,OAAO,eAAe,QAAQ,aAAa;GACzC,OAAO,IAAI,MAAM,MAAM;IACrB,MAAM,QAAQ,QAAS,QAAQ,QAAQ,MAAM,QAAQ,IAAI,QAAQ,KAAK,MAAM;IAC5E,MAAM,QAAQ,QAAQ,QAAQ,SAAS,QAAQ,IAAI,QAAQ,GAAG;GAChE,CAAC;GACD,cAAc;GACd,YAAY;GACZ,UAAU;EACZ,CAAC;CACH;CAEA,MAAe,WAA0B;EACvC,MAAM,SAAS,KAAK;EACpB,KAAK,MAAM,CAAC,QAAQ,UAAU,KAAK,kBAAkB,MAAM,MAAM;EACjE,KAAK,iBAAiB,MAAM;EAC5B,IAAI,KAAK,gBAAgB,OAAO,eAAe,QAAQ,aAAa,KAAK,cAAc;OAClF,OAAO,OAAO;EACnB,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO;EAC7C,MAAM,MAAM,SAAS;CACvB;AACF"}

package/dist/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"mappings":";;;KACY;EAAc;EAAe;EAAgB,MAAM;;;qBASlD;;WAIU,QAAQ;EAA7B,YAAqB,QAAQ;EAE7B,UAAU,QAAQ;EAKlB;EAKA,oBAAoB;EAIpB,qBAAqB;;EAef,cAAc,QAAQ;;;qBAqCjB;;EAMF;EACA;WALA,OAAO;EAGhB,YACS,gBACA;MAGL;MAIA;EAIJ,WAAW,eAAe;EAC1B,WAAW,aAAa;EAKxB,cAAc,QAAQ;;EAKtB,aAAa;EAIb;IAA2B;IAAW;IAAW;IAAe;IAAgB;IAAa;;EAI7F,aAAa,cAAc;EAI3B,aAAa;EAIb;EACA;EACA;;qBAKW,eAAY,eAAiB,mBAAmB;;;KCnIxD,wBAAwB,6BAA6B;KACrD,qBAAqB,6BAA6B;;YAK3C;;EAEV;;;;;;;qBAqDmB,0BAA0B;;EAO7C,YAAY,QAAQ,uBAAuB,SAAS;EA2BrC,SAAS;EAYT,YAAY"}
+{"version":3,"file":"index.d.cts","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"mappings":";;;KACY;EAAc;EAAe;EAAgB,MAAM;;;qBASlD;;WAIU,QAAQ;EAA7B,YAAqB,QAAQ;EAE7B,UAAU,QAAQ;EAKlB;EAKA,oBAAoB;EAIpB,qBAAqB;;EAef,cAAc,QAAQ;;;qBAqCjB;;EAMF;EACA;WALA,OAAO;EAGhB,YACS,gBACA;MAGL;MAIA;EAIJ,WAAW,eAAe;EAC1B,WAAW,aAAa;EAKxB,cAAc,QAAQ;;EAKtB,aAAa;EAIb;IAA2B;IAAW;IAAW;IAAe;IAAgB;IAAa;;EAI7F,aAAa,cAAc;EAI3B,aAAa;EAIb;EACA;EACA;;qBAKW,eAAY,eAAiB,mBAAmB;;;KCnIxD,wBAAwB,6BAA6B;KACrD,qBAAqB,6BAA6B;;YAK3C;;EAEV;;;;;;;qBAqDmB,0BAA0B;;EAO7C,YAAY,QAAQ,uBAAuB,SAAS;EA8BrC,SAAS;EA0BT,YAAY"}

package/dist/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"mappings":";;;KACY;EAAc;EAAe;EAAgB,MAAM;;;qBASlD;;WAIU,QAAQ;EAA7B,YAAqB,QAAQ;EAE7B,UAAU,QAAQ;EAKlB;EAKA,oBAAoB;EAIpB,qBAAqB;;EAef,cAAc,QAAQ;;;qBAqCjB;;EAMF;EACA;WALA,OAAO;EAGhB,YACS,gBACA;MAGL;MAIA;EAIJ,WAAW,eAAe;EAC1B,WAAW,aAAa;EAKxB,cAAc,QAAQ;;EAKtB,aAAa;EAIb;IAA2B;IAAW;IAAW;IAAe;IAAgB;IAAa;;EAI7F,aAAa,cAAc;EAI3B,aAAa;EAIb;EACA;EACA;;qBAKW,eAAY,eAAiB,mBAAmB;;;KCnIxD,wBAAwB,6BAA6B;KACrD,qBAAqB,6BAA6B;;YAK3C;;EAEV;;;;;;;qBAqDmB,0BAA0B;;EAO7C,YAAY,QAAQ,uBAAuB,SAAS;EA2BrC,SAAS;EAYT,YAAY"}
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"mappings":";;;KACY;EAAc;EAAe;EAAgB,MAAM;;;qBASlD;;WAIU,QAAQ;EAA7B,YAAqB,QAAQ;EAE7B,UAAU,QAAQ;EAKlB;EAKA,oBAAoB;EAIpB,qBAAqB;;EAef,cAAc,QAAQ;;;qBAqCjB;;EAMF;EACA;WALA,OAAO;EAGhB,YACS,gBACA;MAGL;MAIA;EAIJ,WAAW,eAAe;EAC1B,WAAW,aAAa;EAKxB,cAAc,QAAQ;;EAKtB,aAAa;EAIb;IAA2B;IAAW;IAAW;IAAe;IAAgB;IAAa;;EAI7F,aAAa,cAAc;EAI3B,aAAa;EAIb;EACA;EACA;;qBAKW,eAAY,eAAiB,mBAAmB;;;KCnIxD,wBAAwB,6BAA6B;KACrD,qBAAqB,6BAA6B;;YAK3C;;EAEV;;;;;;;qBAqDmB,0BAA0B;;EAO7C,YAAY,QAAQ,uBAAuB,SAAS;EA8BrC,SAAS;EA0BT,YAAY"}

package/dist/index.mjs

@@ -171,12 +171,20 @@ const domShims = (global, animationFrames) => {
 */
 var WebgpuEnvironment = class extends TestEnvironment {
 	#added = [];
-	#navigator = {};
+	#hostNavigator;
 	#nativeAdded;
 	#dawnOptions;
 	#animationFrames = /* @__PURE__ */ new Map();
 	constructor(config, context) {
-		super(config, context);
+		const { dawnOptions, ...testEnvironmentOptions } = config.projectConfig.testEnvironmentOptions;
+		super({
+			...config,
+			projectConfig: {
+				...config.projectConfig,
+				testEnvironmentOptions
+			}
+		}, context);
+		this.#dawnOptions = dawnOptions;
 		this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes("_native");
 		if (this.#nativeAdded) Object.defineProperty(this.global, "_native", {
 			value: void 0,
@@ -184,7 +192,6 @@ var WebgpuEnvironment = class extends TestEnvironment {
 			configurable: true,
 			enumerable: false
 		});
-		this.#dawnOptions = config.projectConfig.testEnvironmentOptions.dawnOptions;
 	}
 	async setup() {
 		await super.setup();
@@ -198,17 +205,25 @@ var WebgpuEnvironment = class extends TestEnvironment {
 		this.#added = Object.keys(shims).filter((key) => !(key in global));
 		for (const key of this.#added) global[key] = shims[key];
 		if (this.#nativeAdded) this.#added.push("_native");
-		this.#navigator = global.navigator ??= {};
-		Object.defineProperty(this.#navigator, "gpu", {
-			value: create(this.#dawnOptions ?? []),
-			configurable: true
+		const gpu = create(this.#dawnOptions ?? []);
+		const host = global.navigator ?? {};
+		this.#hostNavigator = Object.getOwnPropertyDescriptor(global, "navigator");
+		Object.defineProperty(global, "navigator", {
+			value: new Proxy(host, {
+				get: (target, key) => key === "gpu" ? gpu : Reflect.get(target, key, target),
+				has: (target, key) => key === "gpu" || Reflect.has(target, key)
+			}),
+			configurable: true,
+			enumerable: true,
+			writable: true
 		});
 	}
 	async teardown() {
 		const global = this.global;
 		for (const [handle, clear] of this.#animationFrames) clear(handle);
 		this.#animationFrames.clear();
-		delete this.#navigator.gpu;
+		if (this.#hostNavigator) Object.defineProperty(global, "navigator", this.#hostNavigator);
+		else delete global.navigator;
 		for (const key of this.#added) delete global[key];
 		await super.teardown();
 	}

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"sourcesContent":["/** RGBA8 pixels, top row first. Feed it to a screenshot matcher such as jest-image-snapshot after encoding to PNG. */\nexport type RgbaImage = { width: number; height: number; data: Uint8Array };\n\nconst COPY_SRC = 0x01;\nconst RENDER_ATTACHMENT = 0x10;\nconst MAP_READ = 0x01;\nconst COPY_DST = 0x08;\nconst READABLE_FORMATS = new Set<GPUTextureFormat>(['rgba8unorm', 'rgba8unorm-srgb', 'bgra8unorm', 'bgra8unorm-srgb']);\n\n/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */\nexport class HeadlessCanvasContext {\n  #config: GPUCanvasConfiguration | undefined;\n  #texture: GPUTexture | undefined;\n\n  constructor(readonly canvas: HeadlessCanvas) {}\n\n  configure(config: GPUCanvasConfiguration): void {\n    this.#config = config;\n    this.#drop();\n  }\n\n  unconfigure(): void {\n    this.#config = undefined;\n    this.#drop();\n  }\n\n  getConfiguration(): GPUCanvasConfiguration | null {\n    return this.#config ?? null;\n  }\n\n  getCurrentTexture(): GPUTexture {\n    const config = this.#config;\n    if (!config) throw new Error('getCurrentTexture() called before configure()');\n    const { width, height } = this.canvas;\n    if (this.#texture && (this.#texture.width !== width || this.#texture.height !== height)) this.#drop();\n    this.#texture ??= config.device.createTexture({\n      size: [width, height],\n      format: config.format,\n      usage: (config.usage ?? RENDER_ATTACHMENT) | RENDER_ATTACHMENT | COPY_SRC,\n      viewFormats: config.viewFormats,\n    });\n    return this.#texture;\n  }\n\n  /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */\n  async readPixels(): Promise<RgbaImage> {\n    const config = this.#config;\n    if (!config) throw new Error('readPixels() called before configure()');\n    const { device, format } = config;\n    if (!READABLE_FORMATS.has(format)) {\n      throw new Error(`readPixels() supports only 8-bit RGBA/BGRA formats; received ${format}`);\n    }\n    const texture = this.getCurrentTexture();\n    const { width, height } = texture;\n    const bytesPerRow = Math.ceil((width * 4) / 256) * 256;\n    const buffer = device.createBuffer({ size: bytesPerRow * height, usage: MAP_READ | COPY_DST });\n    try {\n      const encoder = device.createCommandEncoder();\n      encoder.copyTextureToBuffer({ texture }, { buffer, bytesPerRow }, [width, height]);\n      device.queue.submit([encoder.finish()]);\n      await buffer.mapAsync(MAP_READ);\n      const padded = new Uint8Array(buffer.getMappedRange());\n      const data = new Uint8Array(width * height * 4);\n      for (let y = 0; y < height; y++)\n        data.set(padded.subarray(y * bytesPerRow, y * bytesPerRow + width * 4), y * width * 4);\n      if (format.startsWith('bgra')) {\n        for (let i = 0; i < data.length; i += 4) [data[i], data[i + 2]] = [data[i + 2]!, data[i]!];\n      }\n      return { width, height, data };\n    } finally {\n      if (buffer.mapState === 'mapped') buffer.unmap();\n      buffer.destroy();\n    }\n  }\n\n  #drop(): void {\n    this.#texture?.destroy();\n    this.#texture = undefined;\n  }\n}\n\n/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */\nexport class HeadlessCanvas {\n  readonly #context = new HeadlessCanvasContext(this);\n  readonly style: Record<string, string> = {};\n  readonly #attributes = new Map<string, string>();\n\n  constructor(\n    public width = 300,\n    public height = 150,\n  ) {}\n\n  get clientWidth(): number {\n    return this.width;\n  }\n\n  get clientHeight(): number {\n    return this.height;\n  }\n\n  getContext(id: 'webgpu'): HeadlessCanvasContext;\n  getContext(id: string): HeadlessCanvasContext | null;\n  getContext(id: string): HeadlessCanvasContext | null {\n    return id === 'webgpu' ? this.#context : null;\n  }\n\n  readPixels(): Promise<RgbaImage> {\n    return this.#context.readPixels();\n  }\n\n  /** This same object, typed as an element for libraries whose signatures demand one. */\n  asElement(): HTMLCanvasElement {\n    return this as unknown as HTMLCanvasElement;\n  }\n\n  getBoundingClientRect(): { x: number; y: number; width: number; height: number; top: number; left: number } {\n    return { x: 0, y: 0, top: 0, left: 0, width: this.width, height: this.height };\n  }\n\n  setAttribute(name: string, value: string): void {\n    this.#attributes.set(name, value);\n  }\n\n  getAttribute(name: string): string | null {\n    return this.#attributes.get(name) ?? null;\n  }\n\n  addEventListener(): void {}\n  removeEventListener(): void {}\n  dispatchEvent(): boolean {\n    return true;\n  }\n}\n\nexport const createCanvas = (width: number, height: number): HeadlessCanvas => new HeadlessCanvas(width, height);\n","import { TestEnvironment } from 'jest-environment-node';\nimport { createCanvas, HeadlessCanvas } from './canvas.js';\n\ntype JestEnvironmentConfig = ConstructorParameters<typeof TestEnvironment>[0];\ntype EnvironmentContext = ConstructorParameters<typeof TestEnvironment>[1];\n\nexport { createCanvas, HeadlessCanvas, HeadlessCanvasContext, type RgbaImage } from './canvas.js';\n\n/** Set under `testEnvironmentOptions` in your Jest config. */\nexport type WebgpuNodeOptions = {\n  /** Dawn options, e.g. `['backend=vulkan', 'enable-dawn-features=allow_unsafe_apis']`. */\n  dawnOptions?: string[];\n};\n\n/** Just enough DOM for libraries that make their own canvas and drive a frame loop. Only installed when missing. */\ntype TimerHandle = ReturnType<typeof setTimeout>;\ntype SandboxGlobal = Record<string, unknown> & {\n  clearTimeout: typeof clearTimeout;\n  performance: Performance;\n  setTimeout: typeof setTimeout;\n};\n\nconst domShims = (\n  global: SandboxGlobal,\n  animationFrames: Map<TimerHandle, typeof clearTimeout>,\n): Record<string, unknown> => {\n  const requestAnimationFrame = (callback: (time: number) => void): TimerHandle => {\n    const handle = global.setTimeout(() => {\n      animationFrames.delete(handle);\n      callback(global.performance.now());\n    }, 16);\n    animationFrames.set(handle, global.clearTimeout);\n    (handle as NodeJS.Timeout).unref?.();\n    return handle;\n  };\n  const cancelAnimationFrame = (handle: TimerHandle): void => {\n    const clear = animationFrames.get(handle) ?? global.clearTimeout;\n    animationFrames.delete(handle);\n    clear(handle);\n  };\n  const document = {\n    createElement: (tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    createElementNS: (_ns: string, tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    addEventListener() {},\n    removeEventListener() {},\n  };\n  return {\n    document,\n    devicePixelRatio: 1,\n    requestAnimationFrame,\n    cancelAnimationFrame,\n    addEventListener() {},\n    removeEventListener() {},\n    HTMLCanvasElement: HeadlessCanvas,\n    window: global,\n    self: global,\n  };\n};\n\n/**\n * Jest environment that puts real headless WebGPU (Dawn via the `webgpu` package) on the sandbox\n * global: `navigator.gpu`, every `GPU*` class and constant, and a `HeadlessCanvas` shim. CommonJS\n * tests can rely on `createCanvas` being a global; ESM tests can also `import` it from this package.\n */\nexport default class WebgpuEnvironment extends TestEnvironment {\n  #added: string[] = [];\n  #navigator: { gpu?: GPU } = {};\n  #nativeAdded: boolean;\n  #dawnOptions: string[] | undefined;\n  #animationFrames = new Map<TimerHandle, typeof clearTimeout>();\n\n  constructor(config: JestEnvironmentConfig, context: EnvironmentContext) {\n    super(config, context);\n    // @babylonjs/core defines a `_native` accessor on `self` at module load; on Jest's proxied\n    // sandbox global that defineProperty violates a proxy invariant, so pre-seed the property to\n    // satisfy Babylon's hasOwnProperty guard.\n    // `Object.hasOwn`/`in` report false negatives here: jest-environment-node's sandbox global is\n    // a Proxy over a vm context, and reads of freshly-added properties from outside the sandbox\n    // (as this constructor runs) don't go through the fast path those use. getOwnPropertyNames is\n    // unaffected, so use that instead.\n    this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes('_native');\n    if (this.#nativeAdded) {\n      Object.defineProperty(this.global, '_native', {\n        value: undefined,\n        writable: true,\n        configurable: true,\n        enumerable: false,\n      });\n    }\n    this.#dawnOptions = config.projectConfig.testEnvironmentOptions.dawnOptions as string[] | undefined;\n  }\n\n  // `webgpu` is ESM-only. Loading it via a dynamic import here (rather than a static top-level\n  // import) keeps it out of this module's own top-level module graph, so a CommonJS test file's\n  // `require('jest-environment-webgpu-node')` - which only needs `createCanvas` - never forces\n  // Jest's sandboxed CJS loader to load an ESM-only dependency (jestjs/jest#15716). This method\n  // itself runs outside that sandbox, through Jest's own environment setup, where Node's native\n  // `import()` always works.\n  override async setup(): Promise<void> {\n    await super.setup();\n    const { create, globals } = await import('webgpu');\n    const global = this.global as unknown as SandboxGlobal;\n    const shims = { ...globals, ...domShims(global, this.#animationFrames), createCanvas } as Record<string, unknown>;\n    this.#added = Object.keys(shims).filter((key) => !(key in global));\n    for (const key of this.#added) global[key] = shims[key];\n    if (this.#nativeAdded) this.#added.push('_native');\n    this.#navigator = (global.navigator ??= {}) as { gpu?: GPU };\n    Object.defineProperty(this.#navigator, 'gpu', { value: create(this.#dawnOptions ?? []), configurable: true });\n  }\n\n  override async teardown(): Promise<void> {\n    const global = this.global as unknown as Record<string, unknown>;\n    for (const [handle, clear] of this.#animationFrames) clear(handle);\n    this.#animationFrames.clear();\n    delete this.#navigator.gpu;\n    for (const key of this.#added) delete global[key];\n    await super.teardown();\n  }\n}\n"],"mappings":";;AAIA,MAAM,oBAAoB;AAC1B,MAAM,WAAW;AAEjB,MAAM,mCAAmB,IAAI,IAAsB;CAAC;CAAc;CAAmB;CAAc;AAAiB,CAAC;;AAGrH,IAAa,wBAAb,MAAmC;CAIZ;CAHrB;CACA;CAEA,YAAY,QAAiC;EAAxB,KAAA,SAAA;CAAyB;CAE9C,UAAU,QAAsC;EAC9C,KAAK,UAAU;EACf,KAAK,MAAM;CACb;CAEA,cAAoB;EAClB,KAAK,UAAU,KAAA;EACf,KAAK,MAAM;CACb;CAEA,mBAAkD;EAChD,OAAO,KAAK,WAAW;CACzB;CAEA,oBAAgC;EAC9B,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,+CAA+C;EAC5E,MAAM,EAAE,OAAO,WAAW,KAAK;EAC/B,IAAI,KAAK,aAAa,KAAK,SAAS,UAAU,SAAS,KAAK,SAAS,WAAW,SAAS,KAAK,MAAM;EACpG,KAAK,aAAa,OAAO,OAAO,cAAc;GAC5C,MAAM,CAAC,OAAO,MAAM;GACpB,QAAQ,OAAO;GACf,QAAQ,OAAO,SAAS,qBAAqB;GAC7C,aAAa,OAAO;EACtB,CAAC;EACD,OAAO,KAAK;CACd;;CAGA,MAAM,aAAiC;EACrC,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,wCAAwC;EACrE,MAAM,EAAE,QAAQ,WAAW;EAC3B,IAAI,CAAC,iBAAiB,IAAI,MAAM,GAC9B,MAAM,IAAI,MAAM,gEAAgE,QAAQ;EAE1F,MAAM,UAAU,KAAK,kBAAkB;EACvC,MAAM,EAAE,OAAO,WAAW;EAC1B,MAAM,cAAc,KAAK,KAAM,QAAQ,IAAK,GAAG,IAAI;EACnD,MAAM,SAAS,OAAO,aAAa;GAAE,MAAM,cAAc;GAAQ,OAAO;EAAoB,CAAC;EAC7F,IAAI;GACF,MAAM,UAAU,OAAO,qBAAqB;GAC5C,QAAQ,oBAAoB,EAAE,QAAQ,GAAG;IAAE;IAAQ;GAAY,GAAG,CAAC,OAAO,MAAM,CAAC;GACjF,OAAO,MAAM,OAAO,CAAC,QAAQ,OAAO,CAAC,CAAC;GACtC,MAAM,OAAO,SAAS,QAAQ;GAC9B,MAAM,SAAS,IAAI,WAAW,OAAO,eAAe,CAAC;GACrD,MAAM,OAAO,IAAI,WAAW,QAAQ,SAAS,CAAC;GAC9C,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAC1B,KAAK,IAAI,OAAO,SAAS,IAAI,aAAa,IAAI,cAAc,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC;GACvF,IAAI,OAAO,WAAW,MAAM,GAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK,GAAG,CAAC,KAAK,IAAI,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,IAAK,KAAK,EAAG;GAE3F,OAAO;IAAE;IAAO;IAAQ;GAAK;EAC/B,UAAU;GACR,IAAI,OAAO,aAAa,UAAU,OAAO,MAAM;GAC/C,OAAO,QAAQ;EACjB;CACF;CAEA,QAAc;EACZ,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,KAAA;CAClB;AACF;;AAGA,IAAa,iBAAb,MAA4B;CAMjB;CACA;CANT,WAAoB,IAAI,sBAAsB,IAAI;CAClD,QAAyC,CAAC;CAC1C,8BAAuB,IAAI,IAAoB;CAE/C,YACE,QAAe,KACf,SAAgB,KAChB;EAFO,KAAA,QAAA;EACA,KAAA,SAAA;CACN;CAEH,IAAI,cAAsB;EACxB,OAAO,KAAK;CACd;CAEA,IAAI,eAAuB;EACzB,OAAO,KAAK;CACd;CAIA,WAAW,IAA0C;EACnD,OAAO,OAAO,WAAW,KAAK,WAAW;CAC3C;CAEA,aAAiC;EAC/B,OAAO,KAAK,SAAS,WAAW;CAClC;;CAGA,YAA+B;EAC7B,OAAO;CACT;CAEA,wBAA4G;EAC1G,OAAO;GAAE,GAAG;GAAG,GAAG;GAAG,KAAK;GAAG,MAAM;GAAG,OAAO,KAAK;GAAO,QAAQ,KAAK;EAAO;CAC/E;CAEA,aAAa,MAAc,OAAqB;EAC9C,KAAK,YAAY,IAAI,MAAM,KAAK;CAClC;CAEA,aAAa,MAA6B;EACxC,OAAO,KAAK,YAAY,IAAI,IAAI,KAAK;CACvC;CAEA,mBAAyB,CAAC;CAC1B,sBAA4B,CAAC;CAC7B,gBAAyB;EACvB,OAAO;CACT;AACF;AAEA,MAAa,gBAAgB,OAAe,WAAmC,IAAI,eAAe,OAAO,MAAM;;;AChH/G,MAAM,YACJ,QACA,oBAC4B;CAC5B,MAAM,yBAAyB,aAAkD;EAC/E,MAAM,SAAS,OAAO,iBAAiB;GACrC,gBAAgB,OAAO,MAAM;GAC7B,SAAS,OAAO,YAAY,IAAI,CAAC;EACnC,GAAG,EAAE;EACL,gBAAgB,IAAI,QAAQ,OAAO,YAAY;EAC/C,OAA2B,QAAQ;EACnC,OAAO;CACT;CACA,MAAM,wBAAwB,WAA8B;EAC1D,MAAM,QAAQ,gBAAgB,IAAI,MAAM,KAAK,OAAO;EACpD,gBAAgB,OAAO,MAAM;EAC7B,MAAM,MAAM;CACd;CAOA,OAAO;EACL,UAAA;GANA,gBAAgB,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC5E,kBAAkB,KAAa,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC3F,mBAAmB,CAAC;GACpB,sBAAsB,CAAC;EAGhB;EACP,kBAAkB;EAClB;EACA;EACA,mBAAmB,CAAC;EACpB,sBAAsB,CAAC;EACvB,mBAAmB;EACnB,QAAQ;EACR,MAAM;CACR;AACF;;;;;;AAOA,IAAqB,oBAArB,cAA+C,gBAAgB;CAC7D,SAAmB,CAAC;CACpB,aAA4B,CAAC;CAC7B;CACA;CACA,mCAAmB,IAAI,IAAsC;CAE7D,YAAY,QAA+B,SAA6B;EACtE,MAAM,QAAQ,OAAO;EAQrB,KAAK,eAAe,CAAC,OAAO,oBAAoB,KAAK,MAAM,CAAC,CAAC,SAAS,SAAS;EAC/E,IAAI,KAAK,cACP,OAAO,eAAe,KAAK,QAAQ,WAAW;GAC5C,OAAO,KAAA;GACP,UAAU;GACV,cAAc;GACd,YAAY;EACd,CAAC;EAEH,KAAK,eAAe,OAAO,cAAc,uBAAuB;CAClE;CAQA,MAAe,QAAuB;EACpC,MAAM,MAAM,MAAM;EAClB,MAAM,EAAE,QAAQ,YAAY,MAAM,OAAO;EACzC,MAAM,SAAS,KAAK;EACpB,MAAM,QAAQ;GAAE,GAAG;GAAS,GAAG,SAAS,QAAQ,KAAK,gBAAgB;GAAG;EAAa;EACrF,KAAK,SAAS,OAAO,KAAK,KAAK,CAAC,CAAC,QAAQ,QAAQ,EAAE,OAAO,OAAO;EACjE,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO,MAAM;EACnD,IAAI,KAAK,cAAc,KAAK,OAAO,KAAK,SAAS;EACjD,KAAK,aAAc,OAAO,cAAc,CAAC;EACzC,OAAO,eAAe,KAAK,YAAY,OAAO;GAAE,OAAO,OAAO,KAAK,gBAAgB,CAAC,CAAC;GAAG,cAAc;EAAK,CAAC;CAC9G;CAEA,MAAe,WAA0B;EACvC,MAAM,SAAS,KAAK;EACpB,KAAK,MAAM,CAAC,QAAQ,UAAU,KAAK,kBAAkB,MAAM,MAAM;EACjE,KAAK,iBAAiB,MAAM;EAC5B,OAAO,KAAK,WAAW;EACvB,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO;EAC7C,MAAM,MAAM,SAAS;CACvB;AACF"}
+{"version":3,"file":"index.mjs","names":[],"sources":["../src/canvas.ts","../src/index.ts"],"sourcesContent":["/** RGBA8 pixels, top row first. Feed it to a screenshot matcher such as jest-image-snapshot after encoding to PNG. */\nexport type RgbaImage = { width: number; height: number; data: Uint8Array };\n\nconst COPY_SRC = 0x01;\nconst RENDER_ATTACHMENT = 0x10;\nconst MAP_READ = 0x01;\nconst COPY_DST = 0x08;\nconst READABLE_FORMATS = new Set<GPUTextureFormat>(['rgba8unorm', 'rgba8unorm-srgb', 'bgra8unorm', 'bgra8unorm-srgb']);\n\n/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */\nexport class HeadlessCanvasContext {\n  #config: GPUCanvasConfiguration | undefined;\n  #texture: GPUTexture | undefined;\n\n  constructor(readonly canvas: HeadlessCanvas) {}\n\n  configure(config: GPUCanvasConfiguration): void {\n    this.#config = config;\n    this.#drop();\n  }\n\n  unconfigure(): void {\n    this.#config = undefined;\n    this.#drop();\n  }\n\n  getConfiguration(): GPUCanvasConfiguration | null {\n    return this.#config ?? null;\n  }\n\n  getCurrentTexture(): GPUTexture {\n    const config = this.#config;\n    if (!config) throw new Error('getCurrentTexture() called before configure()');\n    const { width, height } = this.canvas;\n    if (this.#texture && (this.#texture.width !== width || this.#texture.height !== height)) this.#drop();\n    this.#texture ??= config.device.createTexture({\n      size: [width, height],\n      format: config.format,\n      usage: (config.usage ?? RENDER_ATTACHMENT) | RENDER_ATTACHMENT | COPY_SRC,\n      viewFormats: config.viewFormats,\n    });\n    return this.#texture;\n  }\n\n  /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */\n  async readPixels(): Promise<RgbaImage> {\n    const config = this.#config;\n    if (!config) throw new Error('readPixels() called before configure()');\n    const { device, format } = config;\n    if (!READABLE_FORMATS.has(format)) {\n      throw new Error(`readPixels() supports only 8-bit RGBA/BGRA formats; received ${format}`);\n    }\n    const texture = this.getCurrentTexture();\n    const { width, height } = texture;\n    const bytesPerRow = Math.ceil((width * 4) / 256) * 256;\n    const buffer = device.createBuffer({ size: bytesPerRow * height, usage: MAP_READ | COPY_DST });\n    try {\n      const encoder = device.createCommandEncoder();\n      encoder.copyTextureToBuffer({ texture }, { buffer, bytesPerRow }, [width, height]);\n      device.queue.submit([encoder.finish()]);\n      await buffer.mapAsync(MAP_READ);\n      const padded = new Uint8Array(buffer.getMappedRange());\n      const data = new Uint8Array(width * height * 4);\n      for (let y = 0; y < height; y++)\n        data.set(padded.subarray(y * bytesPerRow, y * bytesPerRow + width * 4), y * width * 4);\n      if (format.startsWith('bgra')) {\n        for (let i = 0; i < data.length; i += 4) [data[i], data[i + 2]] = [data[i + 2]!, data[i]!];\n      }\n      return { width, height, data };\n    } finally {\n      if (buffer.mapState === 'mapped') buffer.unmap();\n      buffer.destroy();\n    }\n  }\n\n  #drop(): void {\n    this.#texture?.destroy();\n    this.#texture = undefined;\n  }\n}\n\n/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */\nexport class HeadlessCanvas {\n  readonly #context = new HeadlessCanvasContext(this);\n  readonly style: Record<string, string> = {};\n  readonly #attributes = new Map<string, string>();\n\n  constructor(\n    public width = 300,\n    public height = 150,\n  ) {}\n\n  get clientWidth(): number {\n    return this.width;\n  }\n\n  get clientHeight(): number {\n    return this.height;\n  }\n\n  getContext(id: 'webgpu'): HeadlessCanvasContext;\n  getContext(id: string): HeadlessCanvasContext | null;\n  getContext(id: string): HeadlessCanvasContext | null {\n    return id === 'webgpu' ? this.#context : null;\n  }\n\n  readPixels(): Promise<RgbaImage> {\n    return this.#context.readPixels();\n  }\n\n  /** This same object, typed as an element for libraries whose signatures demand one. */\n  asElement(): HTMLCanvasElement {\n    return this as unknown as HTMLCanvasElement;\n  }\n\n  getBoundingClientRect(): { x: number; y: number; width: number; height: number; top: number; left: number } {\n    return { x: 0, y: 0, top: 0, left: 0, width: this.width, height: this.height };\n  }\n\n  setAttribute(name: string, value: string): void {\n    this.#attributes.set(name, value);\n  }\n\n  getAttribute(name: string): string | null {\n    return this.#attributes.get(name) ?? null;\n  }\n\n  addEventListener(): void {}\n  removeEventListener(): void {}\n  dispatchEvent(): boolean {\n    return true;\n  }\n}\n\nexport const createCanvas = (width: number, height: number): HeadlessCanvas => new HeadlessCanvas(width, height);\n","import { TestEnvironment } from 'jest-environment-node';\nimport { createCanvas, HeadlessCanvas } from './canvas.js';\n\ntype JestEnvironmentConfig = ConstructorParameters<typeof TestEnvironment>[0];\ntype EnvironmentContext = ConstructorParameters<typeof TestEnvironment>[1];\n\nexport { createCanvas, HeadlessCanvas, HeadlessCanvasContext, type RgbaImage } from './canvas.js';\n\n/** Set under `testEnvironmentOptions` in your Jest config. */\nexport type WebgpuNodeOptions = {\n  /** Dawn options, e.g. `['backend=vulkan', 'enable-dawn-features=allow_unsafe_apis']`. */\n  dawnOptions?: string[];\n};\n\n/** Just enough DOM for libraries that make their own canvas and drive a frame loop. Only installed when missing. */\ntype TimerHandle = ReturnType<typeof setTimeout>;\ntype SandboxGlobal = Record<string, unknown> & {\n  clearTimeout: typeof clearTimeout;\n  performance: Performance;\n  setTimeout: typeof setTimeout;\n};\n\nconst domShims = (\n  global: SandboxGlobal,\n  animationFrames: Map<TimerHandle, typeof clearTimeout>,\n): Record<string, unknown> => {\n  const requestAnimationFrame = (callback: (time: number) => void): TimerHandle => {\n    const handle = global.setTimeout(() => {\n      animationFrames.delete(handle);\n      callback(global.performance.now());\n    }, 16);\n    animationFrames.set(handle, global.clearTimeout);\n    (handle as NodeJS.Timeout).unref?.();\n    return handle;\n  };\n  const cancelAnimationFrame = (handle: TimerHandle): void => {\n    const clear = animationFrames.get(handle) ?? global.clearTimeout;\n    animationFrames.delete(handle);\n    clear(handle);\n  };\n  const document = {\n    createElement: (tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    createElementNS: (_ns: string, tag: string) => (tag === 'canvas' ? new HeadlessCanvas() : {}),\n    addEventListener() {},\n    removeEventListener() {},\n  };\n  return {\n    document,\n    devicePixelRatio: 1,\n    requestAnimationFrame,\n    cancelAnimationFrame,\n    addEventListener() {},\n    removeEventListener() {},\n    HTMLCanvasElement: HeadlessCanvas,\n    window: global,\n    self: global,\n  };\n};\n\n/**\n * Jest environment that puts real headless WebGPU (Dawn via the `webgpu` package) on the sandbox\n * global: `navigator.gpu`, every `GPU*` class and constant, and a `HeadlessCanvas` shim. CommonJS\n * tests can rely on `createCanvas` being a global; ESM tests can also `import` it from this package.\n */\nexport default class WebgpuEnvironment extends TestEnvironment {\n  #added: string[] = [];\n  #hostNavigator: PropertyDescriptor | undefined;\n  #nativeAdded: boolean;\n  #dawnOptions: string[] | undefined;\n  #animationFrames = new Map<TimerHandle, typeof clearTimeout>();\n\n  constructor(config: JestEnvironmentConfig, context: EnvironmentContext) {\n    // jest-environment-node assigns every `testEnvironmentOptions` entry onto the sandbox global, so\n    // strip this environment's own option first to keep `dawnOptions` off `globalThis`.\n    const { dawnOptions, ...testEnvironmentOptions } = config.projectConfig.testEnvironmentOptions;\n    super({ ...config, projectConfig: { ...config.projectConfig, testEnvironmentOptions } }, context);\n    this.#dawnOptions = dawnOptions as string[] | undefined;\n    // @babylonjs/core defines a `_native` accessor on `self` at module load; on Jest's proxied\n    // sandbox global that defineProperty violates a proxy invariant, so pre-seed the property to\n    // satisfy Babylon's hasOwnProperty guard.\n    // `Object.hasOwn`/`in` report false negatives here: jest-environment-node's sandbox global is\n    // a Proxy over a vm context, and reads of freshly-added properties from outside the sandbox\n    // (as this constructor runs) don't go through the fast path those use. getOwnPropertyNames is\n    // unaffected, so use that instead.\n    this.#nativeAdded = !Object.getOwnPropertyNames(this.global).includes('_native');\n    if (this.#nativeAdded) {\n      Object.defineProperty(this.global, '_native', {\n        value: undefined,\n        writable: true,\n        configurable: true,\n        enumerable: false,\n      });\n    }\n  }\n\n  // `webgpu` is ESM-only. Loading it via a dynamic import here (rather than a static top-level\n  // import) keeps it out of this module's own top-level module graph, so a CommonJS test file's\n  // `require('jest-environment-webgpu-node')` - which only needs `createCanvas` - never forces\n  // Jest's sandboxed CJS loader to load an ESM-only dependency (jestjs/jest#15716). This method\n  // itself runs outside that sandbox, through Jest's own environment setup, where Node's native\n  // `import()` always works.\n  override async setup(): Promise<void> {\n    await super.setup();\n    const { create, globals } = await import('webgpu');\n    const global = this.global as unknown as SandboxGlobal;\n    const shims = { ...globals, ...domShims(global, this.#animationFrames), createCanvas } as Record<string, unknown>;\n    this.#added = Object.keys(shims).filter((key) => !(key in global));\n    for (const key of this.#added) global[key] = shims[key];\n    if (this.#nativeAdded) this.#added.push('_native');\n    // jest-environment-node hands the sandbox Node's own `navigator` by reference, so adding `gpu` to it\n    // would leak onto the host process and be shared by every environment in this worker. Give the\n    // sandbox its own proxy instead; `Reflect.get(target, key, target)` keeps Node's private-field\n    // getters (hardwareConcurrency, userAgent, ...) working through it.\n    const gpu = create(this.#dawnOptions ?? []);\n    const host = (global.navigator ?? {}) as object;\n    this.#hostNavigator = Object.getOwnPropertyDescriptor(global, 'navigator');\n    Object.defineProperty(global, 'navigator', {\n      value: new Proxy(host, {\n        get: (target, key) => (key === 'gpu' ? gpu : Reflect.get(target, key, target)),\n        has: (target, key) => key === 'gpu' || Reflect.has(target, key),\n      }),\n      configurable: true,\n      enumerable: true,\n      writable: true,\n    });\n  }\n\n  override async teardown(): Promise<void> {\n    const global = this.global as unknown as Record<string, unknown>;\n    for (const [handle, clear] of this.#animationFrames) clear(handle);\n    this.#animationFrames.clear();\n    if (this.#hostNavigator) Object.defineProperty(global, 'navigator', this.#hostNavigator);\n    else delete global.navigator;\n    for (const key of this.#added) delete global[key];\n    await super.teardown();\n  }\n}\n"],"mappings":";;AAIA,MAAM,oBAAoB;AAC1B,MAAM,WAAW;AAEjB,MAAM,mCAAmB,IAAI,IAAsB;CAAC;CAAc;CAAmB;CAAc;AAAiB,CAAC;;AAGrH,IAAa,wBAAb,MAAmC;CAIZ;CAHrB;CACA;CAEA,YAAY,QAAiC;EAAxB,KAAA,SAAA;CAAyB;CAE9C,UAAU,QAAsC;EAC9C,KAAK,UAAU;EACf,KAAK,MAAM;CACb;CAEA,cAAoB;EAClB,KAAK,UAAU,KAAA;EACf,KAAK,MAAM;CACb;CAEA,mBAAkD;EAChD,OAAO,KAAK,WAAW;CACzB;CAEA,oBAAgC;EAC9B,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,+CAA+C;EAC5E,MAAM,EAAE,OAAO,WAAW,KAAK;EAC/B,IAAI,KAAK,aAAa,KAAK,SAAS,UAAU,SAAS,KAAK,SAAS,WAAW,SAAS,KAAK,MAAM;EACpG,KAAK,aAAa,OAAO,OAAO,cAAc;GAC5C,MAAM,CAAC,OAAO,MAAM;GACpB,QAAQ,OAAO;GACf,QAAQ,OAAO,SAAS,qBAAqB;GAC7C,aAAa,OAAO;EACtB,CAAC;EACD,OAAO,KAAK;CACd;;CAGA,MAAM,aAAiC;EACrC,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,wCAAwC;EACrE,MAAM,EAAE,QAAQ,WAAW;EAC3B,IAAI,CAAC,iBAAiB,IAAI,MAAM,GAC9B,MAAM,IAAI,MAAM,gEAAgE,QAAQ;EAE1F,MAAM,UAAU,KAAK,kBAAkB;EACvC,MAAM,EAAE,OAAO,WAAW;EAC1B,MAAM,cAAc,KAAK,KAAM,QAAQ,IAAK,GAAG,IAAI;EACnD,MAAM,SAAS,OAAO,aAAa;GAAE,MAAM,cAAc;GAAQ,OAAO;EAAoB,CAAC;EAC7F,IAAI;GACF,MAAM,UAAU,OAAO,qBAAqB;GAC5C,QAAQ,oBAAoB,EAAE,QAAQ,GAAG;IAAE;IAAQ;GAAY,GAAG,CAAC,OAAO,MAAM,CAAC;GACjF,OAAO,MAAM,OAAO,CAAC,QAAQ,OAAO,CAAC,CAAC;GACtC,MAAM,OAAO,SAAS,QAAQ;GAC9B,MAAM,SAAS,IAAI,WAAW,OAAO,eAAe,CAAC;GACrD,MAAM,OAAO,IAAI,WAAW,QAAQ,SAAS,CAAC;GAC9C,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAC1B,KAAK,IAAI,OAAO,SAAS,IAAI,aAAa,IAAI,cAAc,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC;GACvF,IAAI,OAAO,WAAW,MAAM,GAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK,GAAG,CAAC,KAAK,IAAI,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,IAAK,KAAK,EAAG;GAE3F,OAAO;IAAE;IAAO;IAAQ;GAAK;EAC/B,UAAU;GACR,IAAI,OAAO,aAAa,UAAU,OAAO,MAAM;GAC/C,OAAO,QAAQ;EACjB;CACF;CAEA,QAAc;EACZ,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,KAAA;CAClB;AACF;;AAGA,IAAa,iBAAb,MAA4B;CAMjB;CACA;CANT,WAAoB,IAAI,sBAAsB,IAAI;CAClD,QAAyC,CAAC;CAC1C,8BAAuB,IAAI,IAAoB;CAE/C,YACE,QAAe,KACf,SAAgB,KAChB;EAFO,KAAA,QAAA;EACA,KAAA,SAAA;CACN;CAEH,IAAI,cAAsB;EACxB,OAAO,KAAK;CACd;CAEA,IAAI,eAAuB;EACzB,OAAO,KAAK;CACd;CAIA,WAAW,IAA0C;EACnD,OAAO,OAAO,WAAW,KAAK,WAAW;CAC3C;CAEA,aAAiC;EAC/B,OAAO,KAAK,SAAS,WAAW;CAClC;;CAGA,YAA+B;EAC7B,OAAO;CACT;CAEA,wBAA4G;EAC1G,OAAO;GAAE,GAAG;GAAG,GAAG;GAAG,KAAK;GAAG,MAAM;GAAG,OAAO,KAAK;GAAO,QAAQ,KAAK;EAAO;CAC/E;CAEA,aAAa,MAAc,OAAqB;EAC9C,KAAK,YAAY,IAAI,MAAM,KAAK;CAClC;CAEA,aAAa,MAA6B;EACxC,OAAO,KAAK,YAAY,IAAI,IAAI,KAAK;CACvC;CAEA,mBAAyB,CAAC;CAC1B,sBAA4B,CAAC;CAC7B,gBAAyB;EACvB,OAAO;CACT;AACF;AAEA,MAAa,gBAAgB,OAAe,WAAmC,IAAI,eAAe,OAAO,MAAM;;;AChH/G,MAAM,YACJ,QACA,oBAC4B;CAC5B,MAAM,yBAAyB,aAAkD;EAC/E,MAAM,SAAS,OAAO,iBAAiB;GACrC,gBAAgB,OAAO,MAAM;GAC7B,SAAS,OAAO,YAAY,IAAI,CAAC;EACnC,GAAG,EAAE;EACL,gBAAgB,IAAI,QAAQ,OAAO,YAAY;EAC/C,OAA2B,QAAQ;EACnC,OAAO;CACT;CACA,MAAM,wBAAwB,WAA8B;EAC1D,MAAM,QAAQ,gBAAgB,IAAI,MAAM,KAAK,OAAO;EACpD,gBAAgB,OAAO,MAAM;EAC7B,MAAM,MAAM;CACd;CAOA,OAAO;EACL,UAAA;GANA,gBAAgB,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC5E,kBAAkB,KAAa,QAAiB,QAAQ,WAAW,IAAI,eAAe,IAAI,CAAC;GAC3F,mBAAmB,CAAC;GACpB,sBAAsB,CAAC;EAGhB;EACP,kBAAkB;EAClB;EACA;EACA,mBAAmB,CAAC;EACpB,sBAAsB,CAAC;EACvB,mBAAmB;EACnB,QAAQ;EACR,MAAM;CACR;AACF;;;;;;AAOA,IAAqB,oBAArB,cAA+C,gBAAgB;CAC7D,SAAmB,CAAC;CACpB;CACA;CACA;CACA,mCAAmB,IAAI,IAAsC;CAE7D,YAAY,QAA+B,SAA6B;EAGtE,MAAM,EAAE,aAAa,GAAG,2BAA2B,OAAO,cAAc;EACxE,MAAM;GAAE,GAAG;GAAQ,eAAe;IAAE,GAAG,OAAO;IAAe;GAAuB;EAAE,GAAG,OAAO;EAChG,KAAK,eAAe;EAQpB,KAAK,eAAe,CAAC,OAAO,oBAAoB,KAAK,MAAM,CAAC,CAAC,SAAS,SAAS;EAC/E,IAAI,KAAK,cACP,OAAO,eAAe,KAAK,QAAQ,WAAW;GAC5C,OAAO,KAAA;GACP,UAAU;GACV,cAAc;GACd,YAAY;EACd,CAAC;CAEL;CAQA,MAAe,QAAuB;EACpC,MAAM,MAAM,MAAM;EAClB,MAAM,EAAE,QAAQ,YAAY,MAAM,OAAO;EACzC,MAAM,SAAS,KAAK;EACpB,MAAM,QAAQ;GAAE,GAAG;GAAS,GAAG,SAAS,QAAQ,KAAK,gBAAgB;GAAG;EAAa;EACrF,KAAK,SAAS,OAAO,KAAK,KAAK,CAAC,CAAC,QAAQ,QAAQ,EAAE,OAAO,OAAO;EACjE,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO,MAAM;EACnD,IAAI,KAAK,cAAc,KAAK,OAAO,KAAK,SAAS;EAKjD,MAAM,MAAM,OAAO,KAAK,gBAAgB,CAAC,CAAC;EAC1C,MAAM,OAAQ,OAAO,aAAa,CAAC;EACnC,KAAK,iBAAiB,OAAO,yBAAyB,QAAQ,WAAW;EACzE,OAAO,eAAe,QAAQ,aAAa;GACzC,OAAO,IAAI,MAAM,MAAM;IACrB,MAAM,QAAQ,QAAS,QAAQ,QAAQ,MAAM,QAAQ,IAAI,QAAQ,KAAK,MAAM;IAC5E,MAAM,QAAQ,QAAQ,QAAQ,SAAS,QAAQ,IAAI,QAAQ,GAAG;GAChE,CAAC;GACD,cAAc;GACd,YAAY;GACZ,UAAU;EACZ,CAAC;CACH;CAEA,MAAe,WAA0B;EACvC,MAAM,SAAS,KAAK;EACpB,KAAK,MAAM,CAAC,QAAQ,UAAU,KAAK,kBAAkB,MAAM,MAAM;EACjE,KAAK,iBAAiB,MAAM;EAC5B,IAAI,KAAK,gBAAgB,OAAO,eAAe,QAAQ,aAAa,KAAK,cAAc;OAClF,OAAO,OAAO;EACnB,KAAK,MAAM,OAAO,KAAK,QAAQ,OAAO,OAAO;EAC7C,MAAM,MAAM,SAAS;CACvB;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jest-environment-webgpu-node",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Jest environment that provides real headless WebGPU (Dawn via the webgpu package) in Node",
   "keywords": [
     "dawn",
@@ -30,7 +30,7 @@
     "dist/**/*.d.cts.map",
     "dist/**/*.d.mts",
     "dist/**/*.d.mts.map",
-    "src/globals.d.ts",
+    "globals.d.ts",
     "CHANGELOG.md"
   ],
   "type": "module",
@@ -49,7 +49,7 @@
       }
     },
     "./globals": {
-      "types": "./src/globals.d.ts"
+      "types": "./globals.d.ts"
     }
   },
   "publishConfig": {