1 files changed, 141 insertions, 0 deletions
diff --git a/dom/webgpu/tests/cts/checkout/src/webgpu/api/validation/encoding/cmds/render/setVertexBuffer.spec.ts b/dom/webgpu/tests/cts/checkout/src/webgpu/api/validation/encoding/cmds/render/setVertexBuffer.spec.ts
new file mode 100644
index 0000000000..453281dbdd
--- /dev/null
+++ b/dom/webgpu/tests/cts/checkout/src/webgpu/api/validation/encoding/cmds/render/setVertexBuffer.spec.ts
@@ -0,0 +1,141 @@
+export const description = `
+Validation tests for setVertexBuffer on render pass and render bundle.
+`;
+
+import { makeTestGroup } from '../../../../../../common/framework/test_group.js';
+import { kLimitInfo } from '../../../../../capability_info.js';
+import { GPUConst } from '../../../../../constants.js';
+import { kResourceStates } from '../../../../../gpu_test.js';
+import { ValidationTest } from '../../../validation_test.js';
+
+import { kRenderEncodeTypeParams, buildBufferOffsetAndSizeOOBTestParams } from './render.js';
+
+export const g = makeTestGroup(ValidationTest);
+
+g.test('slot')
+  .desc(
+    `
+Tests slot must be less than the maxVertexBuffers in device limits.
+  `
+  )
+  .paramsSubcasesOnly(
+    kRenderEncodeTypeParams.combine('slot', [
+      0,
+      kLimitInfo.maxVertexBuffers.default - 1,
+      kLimitInfo.maxVertexBuffers.default,
+    ] as const)
+  )
+  .fn(t => {
+    const { encoderType, slot } = t.params;
+    const vertexBuffer = t.createBufferWithState('valid', {
+      size: 16,
+      usage: GPUBufferUsage.VERTEX,
+    });
+
+    const { encoder, validateFinish } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(slot, vertexBuffer);
+    validateFinish(slot < kLimitInfo.maxVertexBuffers.default);
+  });
+
+g.test('vertex_buffer_state')
+  .desc(
+    `
+Tests vertex buffer must be valid.
+  `
+  )
+  .paramsSubcasesOnly(kRenderEncodeTypeParams.combine('state', kResourceStates))
+  .fn(t => {
+    const { encoderType, state } = t.params;
+    const vertexBuffer = t.createBufferWithState(state, {
+      size: 16,
+      usage: GPUBufferUsage.VERTEX,
+    });
+
+    const { encoder, validateFinishAndSubmitGivenState } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(0, vertexBuffer);
+    validateFinishAndSubmitGivenState(state);
+  });
+
+g.test('vertex_buffer,device_mismatch')
+  .desc('Tests setVertexBuffer cannot be called with a vertex buffer created from another device')
+  .paramsSubcasesOnly(kRenderEncodeTypeParams.combine('mismatched', [true, false]))
+  .beforeAllSubcases(t => {
+    t.selectMismatchedDeviceOrSkipTestCase(undefined);
+  })
+  .fn(async t => {
+    const { encoderType, mismatched } = t.params;
+    const sourceDevice = mismatched ? t.mismatchedDevice : t.device;
+
+    const vertexBuffer = sourceDevice.createBuffer({
+      size: 16,
+      usage: GPUBufferUsage.VERTEX,
+    });
+    t.trackForCleanup(vertexBuffer);
+
+    const { encoder, validateFinish } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(0, vertexBuffer);
+    validateFinish(!mismatched);
+  });
+
+g.test('vertex_buffer_usage')
+  .desc(
+    `
+Tests vertex buffer must have 'Vertex' usage.
+  `
+  )
+  .paramsSubcasesOnly(
+    kRenderEncodeTypeParams.combine('usage', [
+      GPUConst.BufferUsage.VERTEX, // control case
+      GPUConst.BufferUsage.COPY_DST,
+      GPUConst.BufferUsage.COPY_DST | GPUConst.BufferUsage.VERTEX,
+    ] as const)
+  )
+  .fn(t => {
+    const { encoderType, usage } = t.params;
+    const vertexBuffer = t.device.createBuffer({
+      size: 16,
+      usage,
+    });
+
+    const { encoder, validateFinish } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(0, vertexBuffer);
+    validateFinish((usage & GPUBufferUsage.VERTEX) !== 0);
+  });
+
+g.test('offset_alignment')
+  .desc(
+    `
+Tests offset must be a multiple of 4.
+  `
+  )
+  .paramsSubcasesOnly(kRenderEncodeTypeParams.combine('offset', [0, 2, 4] as const))
+  .fn(t => {
+    const { encoderType, offset } = t.params;
+    const vertexBuffer = t.device.createBuffer({
+      size: 16,
+      usage: GPUBufferUsage.VERTEX,
+    });
+
+    const { encoder, validateFinish: finish } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(0, vertexBuffer, offset);
+    finish(offset % 4 === 0);
+  });
+
+g.test('offset_and_size_oob')
+  .desc(
+    `
+Tests offset and size cannot be larger than vertex buffer size.
+  `
+  )
+  .paramsSubcasesOnly(buildBufferOffsetAndSizeOOBTestParams(4, 256))
+  .fn(t => {
+    const { encoderType, offset, size, _valid } = t.params;
+    const vertexBuffer = t.device.createBuffer({
+      size: 256,
+      usage: GPUBufferUsage.VERTEX,
+    });
+
+    const { encoder, validateFinish } = t.createEncoder(encoderType);
+    encoder.setVertexBuffer(0, vertexBuffer, offset, size);
+    validateFinish(_valid);
+  });