Add tests for mapping behavior around device.destroy()

Author: kainino0x

src/webgpu/api/validation/state/device_lost/destroy.spec.ts

@@ -21,6 +21,7 @@ import {
   kTextureUsageCopyInfo,
   kShaderStageKeys,
 } from '../../../../capability_info.js';
+import { GPUConst } from '../../../../constants.js';
 import {
   getBlockInfoForColorTextureFormat,
   kCompressedTextureFormats,
@@ -52,16 +53,15 @@ class DeviceDestroyTests extends AllFeaturesMaxLimitsValidationTest {
    * `fn` after the device is destroyed without any specific expectation. If `awaitLost` is true, we
    * also wait for device.lost to resolve before executing `fn` in the destroy case.
    */
-  async executeAfterDestroy(fn: () => void, awaitLost: boolean): Promise<void> {
+  async executeAfterDestroy(fn: () => void | Promise<void>, awaitLost: boolean): Promise<void> {
     this.expectDeviceLost('destroyed');
 
-    this.expectValidationError(fn, false);
     this.device.destroy();
     if (awaitLost) {
       const lostInfo = await this.device.lost;
       this.expect(lostInfo.reason === 'destroyed');
     }
-    fn();
+    await fn();
   }
 
   /**
@@ -146,6 +146,109 @@ Tests creating buffers on destroyed device. Tests valid combinations of:
     }, awaitLost);
   });
 
+g.test('mapping,mappedAtCreation')
+  .desc(
+    `
+Tests behavior of mappedAtCreation buffers when destroying the device (multiple times).
+- Various usages
+- Wait for .lost or not
+  `
+  )
+  .params(u =>
+    u
+      .combine('usage', [
+        GPUConst.BufferUsage.MAP_READ,
+        GPUConst.BufferUsage.MAP_WRITE,
+        GPUConst.BufferUsage.COPY_SRC,
+      ])
+      .combine('awaitLost', [true, false])
+  )
+  .fn(async t => {
+    const { awaitLost, usage } = t.params;
+
+    const b1 = t.createBufferTracked({ size: 16, usage, mappedAtCreation: true });
+    t.expect(b1.mapState === 'mapped', 'b1 before destroy 1');
+
+    await t.executeAfterDestroy(() => {
+      // Destroy should have unmapped everything.
+      t.expect(b1.mapState === 'unmapped', 'b1 after destroy 1');
+      // But unmap just in case, to reset state before continuing.
+      b1.unmap();
+
+      // mappedAtCreation should still work.
+      const b2 = t.createBufferTracked({ size: 16, usage, mappedAtCreation: true });
+      t.expect(b2.mapState === 'mapped', 'b2 at creation after destroy 1');
+
+      // Destroying again should unmap.
+      t.device.destroy();
+      t.expect(b2.mapState === 'unmapped', 'b2 after destroy 2');
+    }, awaitLost);
+  });
+
+g.test('mapping,mapAsync')
+  .desc(
+    `
+Tests behavior of mapAsync'd buffers when destroying the device.
+- Various usages
+- Wait for .lost or not
+
+TODO(https://github.com/gpuweb/gpuweb/issues/5101): Test which error we got at [1].
+  `
+  )
+  .params(u =>
+    u
+      .combine('usage', [
+        GPUConst.BufferUsage.MAP_READ,
+        GPUConst.BufferUsage.MAP_WRITE,
+        GPUConst.BufferUsage.COPY_SRC,
+      ])
+      .combine('awaitLost', [true, false])
+  )
+  .fn(async t => {
+    const { awaitLost, usage } = t.params;
+
+    const bAsync = t.createBufferTracked({ size: 16, usage });
+    const mode =
+      usage === GPUBufferUsage.MAP_READ
+        ? GPUMapMode.READ
+        : usage === GPUBufferUsage.MAP_WRITE
+        ? GPUMapMode.WRITE
+        : 0;
+    if (mode) {
+      await bAsync.mapAsync(mode);
+      t.expect(bAsync.mapState === 'mapped', 'bAsync before destroy 1');
+    } else {
+      t.expect(bAsync.mapState === 'unmapped', 'bAsync before destroy 1');
+    }
+
+    await t.executeAfterDestroy(async () => {
+      // Destroy should have unmapped everything.
+      t.expect(bAsync.mapState === 'unmapped', 'bAsync after destroy 1');
+      // But unmap just in case, to reset state before continuing.
+      bAsync.unmap();
+
+      // Check that mapping after destroy fails.
+      // Also check that if mapAsync fails validation, it still produces an OperationError.
+      try {
+        await bAsync.mapAsync(mode);
+        t.expect(false, 'bAsync mapAsync after destroy 1 should reject');
+      } catch (ex) {
+        if (mode) {
+          // The mapAsync call is valid except for the fact that the device is lost.
+          t.expect(ex instanceof DOMException && ex.name === 'AbortError');
+        } else {
+          // The mapAsync call is also invalid for other reasons.
+          // [1] Test which error type we got. (And maybe switch to shouldReject().)
+          t.expect(ex instanceof DOMException);
+        }
+      }
+      const mapPromise = bAsync.mapAsync(mode);
+      t.shouldReject('AbortError', mapPromise);
+      await mapPromise.catch(() => {});
+      t.expect(bAsync.mapState === 'unmapped', 'bAsync after mapAsync after destroy 1');
+    }, awaitLost);
+  });
+
 g.test('createTexture,2d,uncompressed_format')
   .desc(
     `