vitest-environment-webgpu-node 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Ben Houston
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,59 @@
+# vitest-environment-webgpu-node
+
+Real, headless WebGPU inside Vitest, with no browser. Powered by Google's Dawn through the
+[`webgpu`](https://github.com/dawn-gpu/node-webgpu) npm package.
+
+The environment puts `navigator.gpu` and every `GPU*` class and constant (`GPUBufferUsage`,
+`GPUShaderStage`, ...) on the global object before each test file and removes them afterwards.
+
+## Install
+
+```sh
+pnpm add -D vitest vitest-environment-webgpu-node
+```
+
+## Usage
+
+```ts
+// vitest.config.ts
+export default defineConfig({
+  test: {
+    environment: 'webgpu-node',
+    environmentOptions: { webgpuNode: { dawnOptions: ['backend=vulkan'] } },
+  },
+});
+```
+
+```ts
+import { expect, it } from 'vitest';
+
+it('runs a compute shader', async () => {
+  const device = await (await navigator.gpu.requestAdapter())!.requestDevice();
+  // ...createShaderModule, dispatch, copy to a MAP_READ buffer, mapAsync, assert
+});
+```
+
+`dawnOptions` are passed straight to Dawn: `backend=<null|d3d11|d3d12|metal|vulkan|opengl|opengles>`,
+`adapter=<name>`, `enable-dawn-features=...`, `disable-dawn-features=...`.
+
+## Canvas
+
+Dawn has no `<canvas>`, so the environment ships a headless one whose `getContext('webgpu')` returns a
+`GPUCanvasContext` backed by a texture. `document.createElement('canvas')`, `HTMLCanvasElement`, `window`,
+`self` and `requestAnimationFrame` are installed too (only when missing), which is enough for three.js
+`WebGPURenderer`, Babylon Lite and vgpu to run unchanged.
+
+```ts
+import * as THREE from 'three/webgpu';
+import { createCanvas } from 'vitest-environment-webgpu-node';
+
+const canvas = createCanvas(256, 256);
+const renderer = new THREE.WebGPURenderer({ canvas: canvas.asElement() });
+await renderer.init();
+await renderer.renderAsync(scene, camera);
+expect(await canvas.readPixels()).toMatchScreenshot('scene');
+```
+
+`readPixels()` returns `{ width, height, data }` RGBA8 pixels, ready for
+[`vitest-screenshot`](../vitest-screenshot). `asElement()` is the same object typed as an
+`HTMLCanvasElement` for library signatures that demand one.

package/dist/canvas.d.ts

@@ -0,0 +1,48 @@
+/** RGBA8 pixels, top row first. Feed it to `vitest-screenshot`'s `toMatchScreenshot`. */
+export type RgbaImage = {
+    width: number;
+    height: number;
+    data: Uint8Array;
+};
+/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */
+export declare class HeadlessCanvasContext {
+    #private;
+    readonly canvas: HeadlessCanvas;
+    constructor(canvas: HeadlessCanvas);
+    configure(config: GPUCanvasConfiguration): void;
+    unconfigure(): void;
+    getConfiguration(): GPUCanvasConfiguration | null;
+    getCurrentTexture(): GPUTexture;
+    /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */
+    readPixels(): Promise<RgbaImage>;
+}
+/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */
+export declare class HeadlessCanvas {
+    #private;
+    width: number;
+    height: number;
+    readonly style: Record<string, string>;
+    constructor(width?: number, height?: number);
+    get clientWidth(): number;
+    get clientHeight(): number;
+    getContext(id: 'webgpu'): HeadlessCanvasContext;
+    getContext(id: string): HeadlessCanvasContext | null;
+    readPixels(): Promise<RgbaImage>;
+    /** This same object, typed as an element for libraries whose signatures demand one. */
+    asElement(): HTMLCanvasElement;
+    getBoundingClientRect(): {
+        x: number;
+        y: number;
+        width: number;
+        height: number;
+        top: number;
+        left: number;
+    };
+    setAttribute(name: string, value: string): void;
+    getAttribute(name: string): string | null;
+    addEventListener(): void;
+    removeEventListener(): void;
+    dispatchEvent(): boolean;
+}
+export declare const createCanvas: (width: number, height: number) => HeadlessCanvas;
+//# sourceMappingURL=canvas.d.ts.map

package/dist/canvas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"canvas.d.ts","sourceRoot":"","sources":["../src/canvas.ts"],"names":[],"mappings":"AAAA,yFAAyF;AACzF,MAAM,MAAM,SAAS,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,UAAU,CAAA;CAAE,CAAC;AAO5E,iIAAiI;AACjI,qBAAa,qBAAqB;;IAIpB,QAAQ,CAAC,MAAM,EAAE,cAAc;IAA3C,YAAqB,MAAM,EAAE,cAAc,EAAI;IAE/C,SAAS,CAAC,MAAM,EAAE,sBAAsB,GAAG,IAAI,CAG9C;IAED,WAAW,IAAI,IAAI,CAGlB;IAED,gBAAgB,IAAI,sBAAsB,GAAG,IAAI,CAEhD;IAED,iBAAiB,IAAI,UAAU,CAY9B;IAED,gFAAgF;IAC1E,UAAU,IAAI,OAAO,CAAC,SAAS,CAAC,CAoBrC;CAMF;AAED,6FAA6F;AAC7F,qBAAa,cAAc;;IAMhB,KAAK;IACL,MAAM;IALf,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAM;IAG5C,YACS,KAAK,SAAM,EACX,MAAM,SAAM,EACjB;IAEJ,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,UAAU,CAAC,EAAE,EAAE,QAAQ,GAAG,qBAAqB,CAAC;IAChD,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,qBAAqB,GAAG,IAAI,CAAC;IAKrD,UAAU,IAAI,OAAO,CAAC,SAAS,CAAC,CAE/B;IAED,uFAAuF;IACvF,SAAS,IAAI,iBAAiB,CAE7B;IAED,qBAAqB,IAAI;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAE1G;IAED,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAE9C;IAED,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAExC;IAED,gBAAgB,IAAI,IAAI,CAAG;IAC3B,mBAAmB,IAAI,IAAI,CAAG;IAC9B,aAAa,IAAI,OAAO,CAEvB;CACF;AAED,eAAO,MAAM,YAAY,UAAW,MAAM,UAAU,MAAM,KAAG,cAAmD,CAAC"}

package/dist/canvas.js

@@ -0,0 +1,110 @@
+const COPY_SRC = 0x01;
+const RENDER_ATTACHMENT = 0x10;
+const MAP_READ = 0x01;
+const COPY_DST = 0x08;
+/** Minimal `GPUCanvasContext`: `getCurrentTexture()` is a texture the size of the canvas, kept until reconfigured or resized. */
+export class HeadlessCanvasContext {
+    canvas;
+    #config;
+    #texture;
+    constructor(canvas) {
+        this.canvas = canvas;
+    }
+    configure(config) {
+        this.#config = config;
+        this.#drop();
+    }
+    unconfigure() {
+        this.#config = undefined;
+        this.#drop();
+    }
+    getConfiguration() {
+        return this.#config ?? null;
+    }
+    getCurrentTexture() {
+        const config = this.#config;
+        if (!config)
+            throw new Error('getCurrentTexture() called before configure()');
+        const { width, height } = this.canvas;
+        if (this.#texture && (this.#texture.width !== width || this.#texture.height !== height))
+            this.#drop();
+        this.#texture ??= config.device.createTexture({
+            size: [width, height],
+            format: config.format,
+            usage: (config.usage ?? RENDER_ATTACHMENT) | RENDER_ATTACHMENT | COPY_SRC,
+            viewFormats: config.viewFormats,
+        });
+        return this.#texture;
+    }
+    /** Copies the current texture to the CPU. BGRA formats are swizzled to RGBA. */
+    async readPixels() {
+        const texture = this.getCurrentTexture();
+        const { device, format } = this.#config;
+        const { width, height } = texture;
+        const bytesPerRow = Math.ceil((width * 4) / 256) * 256;
+        const buffer = device.createBuffer({ size: bytesPerRow * height, usage: MAP_READ | COPY_DST });
+        const encoder = device.createCommandEncoder();
+        encoder.copyTextureToBuffer({ texture }, { buffer, bytesPerRow }, [width, height]);
+        device.queue.submit([encoder.finish()]);
+        await buffer.mapAsync(MAP_READ);
+        const padded = new Uint8Array(buffer.getMappedRange());
+        const data = new Uint8Array(width * height * 4);
+        for (let y = 0; y < height; y++)
+            data.set(padded.subarray(y * bytesPerRow, y * bytesPerRow + width * 4), y * width * 4);
+        buffer.unmap();
+        buffer.destroy();
+        if (format.startsWith('bgra')) {
+            for (let i = 0; i < data.length; i += 4)
+                [data[i], data[i + 2]] = [data[i + 2], data[i]];
+        }
+        return { width, height, data };
+    }
+    #drop() {
+        this.#texture?.destroy();
+        this.#texture = undefined;
+    }
+}
+/** Enough of `HTMLCanvasElement` for three.js `WebGPURenderer`, Babylon Lite and friends. */
+export class HeadlessCanvas {
+    width;
+    height;
+    #context = new HeadlessCanvasContext(this);
+    style = {};
+    #attributes = new Map();
+    constructor(width = 300, height = 150) {
+        this.width = width;
+        this.height = height;
+    }
+    get clientWidth() {
+        return this.width;
+    }
+    get clientHeight() {
+        return this.height;
+    }
+    getContext(id) {
+        return id === 'webgpu' ? this.#context : null;
+    }
+    readPixels() {
+        return this.#context.readPixels();
+    }
+    /** This same object, typed as an element for libraries whose signatures demand one. */
+    asElement() {
+        return this;
+    }
+    getBoundingClientRect() {
+        return { x: 0, y: 0, top: 0, left: 0, width: this.width, height: this.height };
+    }
+    setAttribute(name, value) {
+        this.#attributes.set(name, value);
+    }
+    getAttribute(name) {
+        return this.#attributes.get(name) ?? null;
+    }
+    addEventListener() { }
+    removeEventListener() { }
+    dispatchEvent() {
+        return true;
+    }
+}
+export const createCanvas = (width, height) => new HeadlessCanvas(width, height);
+//# sourceMappingURL=canvas.js.map

package/dist/canvas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"canvas.js","sourceRoot":"","sources":["../src/canvas.ts"],"names":[],"mappings":"AAGA,MAAM,QAAQ,GAAG,IAAI,CAAC;AACtB,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC;AACtB,MAAM,QAAQ,GAAG,IAAI,CAAC;AAEtB,iIAAiI;AACjI,MAAM,OAAO,qBAAqB;IAIX,MAAM;IAH3B,OAAO,CAAqC;IAC5C,QAAQ,CAAyB;IAEjC,YAAqB,MAAsB;sBAAtB,MAAM;IAAmB,CAAC;IAE/C,SAAS,CAAC,MAA8B;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,WAAW;QACT,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QACzB,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;IAC9B,CAAC;IAED,iBAAiB;QACf,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QAC9E,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QACtC,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,MAAM,CAAC;YAAE,IAAI,CAAC,KAAK,EAAE,CAAC;QACtG,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,CAAC,MAAM,CAAC,KAAK,IAAI,iBAAiB,CAAC,GAAG,iBAAiB,GAAG,QAAQ;YACzE,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,gFAAgF;IAChF,KAAK,CAAC,UAAU;QACd,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,OAAQ,CAAC;QACzC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,WAAW,GAAG,MAAM,EAAE,KAAK,EAAE,QAAQ,GAAG,QAAQ,EAAE,CAAC,CAAC;QAC/F,MAAM,OAAO,GAAG,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAC9C,OAAO,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACnF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACxC,MAAM,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC;QAChD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE;YAC7B,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,WAAW,EAAE,CAAC,GAAG,WAAW,GAAG,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;QACzF,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC;gBAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,EAAE,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC;QAC7F,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACzB,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC5B,CAAC;CACF;AAED,6FAA6F;AAC7F,MAAM,OAAO,cAAc;IAMhB,KAAK;IACL,MAAM;IANN,QAAQ,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAC3C,KAAK,GAA2B,EAAE,CAAC;IACnC,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEjD,YACS,KAAK,GAAG,GAAG,EACX,MAAM,GAAG,GAAG;qBADZ,KAAK;sBACL,MAAM;IACZ,CAAC;IAEJ,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAID,UAAU,CAAC,EAAU;QACnB,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC;IAChD,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;IACpC,CAAC;IAED,uFAAuF;IACvF,SAAS;QACP,OAAO,IAAoC,CAAC;IAC9C,CAAC;IAED,qBAAqB;QACnB,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IACjF,CAAC;IAED,YAAY,CAAC,IAAY,EAAE,KAAa;QACtC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,YAAY,CAAC,IAAY;QACvB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC;IAC5C,CAAC;IAED,gBAAgB,KAAU,CAAC;IAC3B,mBAAmB,KAAU,CAAC;IAC9B,aAAa;QACX,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,MAAc,EAAkB,EAAE,CAAC,IAAI,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,10 @@
+import type { Environment } from 'vitest/environments';
+export { createCanvas, HeadlessCanvas, HeadlessCanvasContext, type RgbaImage } from './canvas.js';
+/** Set under `test.environmentOptions.webgpuNode` in your Vitest config. */
+export type WebgpuNodeOptions = {
+    /** Dawn options, e.g. `['backend=vulkan', 'enable-dawn-features=allow_unsafe_apis']`. */
+    dawnOptions?: string[];
+};
+declare const _default: Environment;
+export default _default;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAIvD,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,qBAAqB,EAAE,KAAK,SAAS,EAAE,MAAM,aAAa,CAAC;AAElG,4EAA4E;AAC5E,MAAM,MAAM,iBAAiB,GAAG;IAC9B,yFAAyF;IACzF,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB,CAAC;wBA6Bc,WAAW"}

package/dist/index.js

@@ -0,0 +1,47 @@
+import { create, globals } from 'webgpu';
+import { HeadlessCanvas } from './canvas.js';
+export { createCanvas, HeadlessCanvas, HeadlessCanvasContext } from './canvas.js';
+/** Just enough DOM for libraries that make their own canvas and drive a frame loop. Only installed when missing. */
+const requestAnimationFrame = (callback) => setTimeout(() => callback(performance.now()), 16).unref();
+const domShims = () => {
+    const window = {
+        devicePixelRatio: 1,
+        requestAnimationFrame,
+        cancelAnimationFrame: clearTimeout,
+        addEventListener() { },
+        removeEventListener() { },
+    };
+    return {
+        HTMLCanvasElement: HeadlessCanvas,
+        document: {
+            createElement: (tag) => (tag === 'canvas' ? new HeadlessCanvas() : {}),
+            createElementNS: (_ns, tag) => (tag === 'canvas' ? new HeadlessCanvas() : {}),
+            addEventListener() { },
+            removeEventListener() { },
+        },
+        window,
+        self: window,
+        requestAnimationFrame,
+        cancelAnimationFrame: clearTimeout,
+    };
+};
+export default {
+    name: 'webgpu-node',
+    viteEnvironment: 'ssr',
+    setup(global, { webgpuNode = {} }) {
+        const shims = { ...globals, ...domShims() };
+        const added = Object.keys(shims).filter((key) => !(key in global));
+        for (const key of added)
+            global[key] = shims[key];
+        const navigator = (global.navigator ??= {});
+        Object.defineProperty(navigator, 'gpu', { value: create(webgpuNode.dawnOptions ?? []), configurable: true });
+        return {
+            teardown(g) {
+                delete navigator.gpu;
+                for (const key of added)
+                    delete g[key];
+            },
+        };
+    },
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,qBAAqB,EAAkB,MAAM,aAAa,CAAC;AAQlG,oHAAoH;AACpH,MAAM,qBAAqB,GAAG,CAAC,QAAgC,EAAkB,EAAE,CACjF,UAAU,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;AAE5D,MAAM,QAAQ,GAAG,GAA4B,EAAE;IAC7C,MAAM,MAAM,GAAG;QACb,gBAAgB,EAAE,CAAC;QACnB,qBAAqB;QACrB,oBAAoB,EAAE,YAAY;QAClC,gBAAgB,KAAI,CAAC;QACrB,mBAAmB,KAAI,CAAC;KACzB,CAAC;IACF,OAAO;QACL,iBAAiB,EAAE,cAAc;QACjC,QAAQ,EAAE;YACR,aAAa,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9E,eAAe,EAAE,CAAC,GAAW,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7F,gBAAgB,KAAI,CAAC;YACrB,mBAAmB,KAAI,CAAC;SACzB;QACD,MAAM;QACN,IAAI,EAAE,MAAM;QACZ,qBAAqB;QACrB,oBAAoB,EAAE,YAAY;KACnC,CAAC;AACJ,CAAC,CAAC;AAEF,eAA4B;IAC1B,IAAI,EAAE,aAAa;IACnB,eAAe,EAAE,KAAK;IACtB,KAAK,CAAC,MAA+B,EAAE,EAAE,UAAU,GAAG,EAAE,EAAsC;QAC5F,MAAM,KAAK,GAAG,EAAE,GAAG,OAAO,EAAE,GAAG,QAAQ,EAAE,EAA6B,CAAC;QACvE,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,MAAM,CAAC,CAAC,CAAC;QACnE,KAAK,MAAM,GAAG,IAAI,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,CAAC,MAAM,CAAC,SAAS,KAAK,EAAE,CAAkB,CAAC;QAC7D,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7G,OAAO;YACL,QAAQ,CAAC,CAA0B;gBACjC,OAAO,SAAS,CAAC,GAAG,CAAC;gBACrB,KAAK,MAAM,GAAG,IAAI,KAAK;oBAAE,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC;YACzC,CAAC;SACF,CAAC;IACJ,CAAC;CACF,CAAC"}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "vitest-environment-webgpu-node",
+  "version": "0.1.0",
+  "description": "Vitest environment that provides real headless WebGPU (Dawn via the webgpu package) in Node",
+  "keywords": [
+    "dawn",
+    "headless",
+    "testing",
+    "vitest",
+    "vitest-environment",
+    "webgpu"
+  ],
+  "homepage": "https://github.com/bhouston/vitest-gpu#readme",
+  "bugs": {
+    "url": "https://github.com/bhouston/vitest-gpu/issues"
+  },
+  "license": "MIT",
+  "author": "Ben Houston <ben@ben3d.ca> (https://ben3d.ca)",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/bhouston/vitest-gpu.git",
+    "directory": "packages/vitest-environment-webgpu-node"
+  },
+  "files": [
+    "dist/**/*.js",
+    "dist/**/*.js.map",
+    "dist/**/*.d.ts",
+    "dist/**/*.d.ts.map",
+    "CHANGELOG.md"
+  ],
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc -b",
+    "tsc": "tsc -b"
+  },
+  "dependencies": {
+    "@webgpu/types": "^0.1.72",
+    "webgpu": "^0.6.1"
+  },
+  "peerDependencies": {
+    "vitest": ">=3 <5"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  }
+}