@quake2ts/test-utils 0.0.848 → 0.0.850

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@quake2ts/test-utils",
-  "version": "0.0.848",
+  "version": "0.0.850",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -55,10 +55,10 @@
     "serve-handler": "^6.1.6",
     "vitest": "^1.6.0",
     "webgpu": "^0.3.8",
-    "@quake2ts/engine": "^0.0.848",
-    "@quake2ts/server": "0.0.848",
-    "@quake2ts/shared": "0.0.848",
-    "@quake2ts/game": "0.0.848"
+    "@quake2ts/engine": "^0.0.850",
+    "@quake2ts/game": "0.0.850",
+    "@quake2ts/shared": "0.0.850",
+    "@quake2ts/server": "0.0.850"
   },
   "peerDependenciesMeta": {
     "@quake2ts/engine": {
@@ -114,10 +114,10 @@
     "typescript": "^5.9.3",
     "vitest": "^4.0.16",
     "webgpu": "^0.3.8",
-    "@quake2ts/engine": "^0.0.848",
-    "@quake2ts/game": "0.0.848",
-    "@quake2ts/server": "0.0.848",
-    "@quake2ts/shared": "0.0.848"
+    "@quake2ts/engine": "^0.0.850",
+    "@quake2ts/game": "0.0.850",
+    "@quake2ts/server": "0.0.850",
+    "@quake2ts/shared": "0.0.850"
   },
   "dependencies": {
     "upng-js": "^2.1.0"

package/src/engine/builders/visual-testing.ts

@@ -1,10 +1,10 @@
 import { test as base } from 'vitest';
+import { expectSnapshot } from '../../visual/snapshots.js';
 import {
-    expectSnapshot,
     createRenderTestSetup,
     RenderTestSetup,
     captureTexture
-} from '@quake2ts/test-utils';
+} from '../helpers/webgpu-rendering.js';
 import path from 'path';
 
 interface VisualTestContext {

package/src/index.ts

@@ -30,6 +30,7 @@ export * from './server/helpers/bandwidth.js';
 export * from './setup/browser.js';
 export * from './setup/canvas.js';
 export * from './setup/webgpu.js';
+export * from './setup/webgpu-lifecycle.js';
 // Safe to export now as gl is lazy loaded
 export * from './setup/headless-webgl.js';
 export * from './engine/mocks/webgpu.js';
@@ -72,6 +73,7 @@ export * from './client/helpers/prediction.js';
 // Visual Testing
 export * from './visual/snapshots.js';
 export * from './visual/animation-snapshots.js';
+export { test } from './engine/builders/visual-testing.js';
 
 // E2E
 export * from './e2e/playwright.js';

package/src/setup/webgpu-lifecycle.ts

@@ -0,0 +1,102 @@
+/**
+ * WebGPU Device Lifecycle Management for Integration Tests
+ *
+ * Provides utilities to track and cleanup GPUDevice instances created during tests,
+ * preventing resource leaks.
+ *
+ * Usage:
+ * ```ts
+ * describe('My WebGPU Tests', () => {
+ *   const lifecycle = createWebGPULifecycle();
+ *
+ *   beforeAll(async () => {
+ *     await initHeadlessWebGPU();
+ *   });
+ *
+ *   afterAll(lifecycle.cleanup);
+ *
+ *   it('test', async () => {
+ *     const context = await createWebGPUContext();
+ *     lifecycle.track(context.device);
+ *     // ... test code
+ *   });
+ * });
+ * ```
+ */
+
+export interface WebGPULifecycle {
+  /**
+   * Track a GPUDevice for cleanup
+   */
+  track(device: GPUDevice): void;
+
+  /**
+   * Track a renderer with a device property for cleanup
+   */
+  trackRenderer(renderer: { device: GPUDevice; dispose: () => void }): void;
+
+  /**
+   * Cleanup all tracked devices
+   * Call this in afterAll hook
+   */
+  cleanup(): Promise<void>;
+
+  /**
+   * Get count of tracked devices
+   */
+  count(): number;
+}
+
+/**
+ * Create a WebGPU lifecycle manager for tracking and cleaning up devices
+ */
+export function createWebGPULifecycle(): WebGPULifecycle {
+  const devices: GPUDevice[] = [];
+  const renderers: Array<{ device: GPUDevice; dispose: () => void }> = [];
+
+  return {
+    track(device: GPUDevice): void {
+      devices.push(device);
+    },
+
+    trackRenderer(renderer: { device: GPUDevice; dispose: () => void }): void {
+      renderers.push(renderer);
+    },
+
+    async cleanup(): Promise<void> {
+      // Dispose renderers first (they may destroy their devices)
+      for (const renderer of renderers) {
+        try {
+          renderer.dispose();
+        } catch (e) {
+          console.warn('Error disposing renderer:', e);
+        }
+      }
+      renderers.length = 0;
+
+      // Collect all devices and their lost promises
+      const allDevices = [...devices];
+      devices.length = 0;
+
+      if (allDevices.length === 0) return;
+
+      // Wait for all devices to be destroyed
+      const lostPromises = allDevices.map(d => d.lost);
+      for (const device of allDevices) {
+        try {
+          device.destroy();
+        } catch (e) {
+          // Device might already be destroyed
+          console.warn('Error destroying device:', e);
+        }
+      }
+
+      // Wait for all devices to report as lost
+      await Promise.all(lostPromises);
+    },
+
+    count(): number {
+      return devices.length + renderers.length;
+    }
+  };
+}