1 files changed, 70 insertions, 0 deletions
diff --git a/dom/webgpu/tests/cts/checkout/src/webgpu/shader/validation/static_assert/static_assert.spec.ts b/dom/webgpu/tests/cts/checkout/src/webgpu/shader/validation/static_assert/static_assert.spec.ts
new file mode 100644
index 0000000000..24db6bd5af
--- /dev/null
+++ b/dom/webgpu/tests/cts/checkout/src/webgpu/shader/validation/static_assert/static_assert.spec.ts
@@ -0,0 +1,70 @@
+export const description = `Validation tests for static_assert`;
+
+import { makeTestGroup } from '../../../../common/framework/test_group.js';
+import { ShaderValidationTest } from '../shader_validation_test.js';
+
+export const g = makeTestGroup(ShaderValidationTest);
+
+/**
+ * Builds a static_assert() statement, which checks that @p expr is equal to @p expect_true.
+ * @param expect_true true if @p expr should evaluate to true
+ * @param expr the constant expression
+ * @param scope module-scope or function-scope constant expression
+ * @returns the WGSL code
+ */
+function buildStaticAssert(expect_true: boolean, expr: string, scope: 'module' | 'function') {
+  const stmt = expect_true ? `static_assert ${expr};` : `static_assert !(${expr});`;
+  return scope === 'module' ? stmt : `fn f() { ${stmt} }`;
+}
+
+const kConditionCases = {
+  true_literal: `true`,
+  not_false: `!false`,
+  const_eq_literal_int: `one == 1`,
+  const_eq_literal_float: `one == 1.0`,
+  binary_op_eq_const: `one+1 == two`,
+  any: `any(vec3(false, true, false))`,
+  min_max: `min(three, max(two, one)) == 2`,
+};
+
+g.test('constant_expression')
+  .desc(`Test that static_assert validates the condition expression.`)
+  .params(u =>
+    u
+      .combine('case', Object.keys(kConditionCases) as Array<keyof typeof kConditionCases>)
+      .combine('scope', ['module', 'function'] as const)
+      .beginSubcases()
+  )
+  .fn(t => {
+    const constants = `
+const one = 1;
+const two = 2;
+const three = 2;
+`;
+    const expr = kConditionCases[t.params.case];
+    t.expectCompileResult(true, constants + buildStaticAssert(true, expr, t.params.scope));
+    t.expectCompileResult(false, constants + buildStaticAssert(false, expr, t.params.scope));
+  });
+
+g.test('evaluation_stage')
+  .desc(`Test that the static_assert expression must be a constant expression.`)
+  .params(u =>
+    u
+      .combine('scope', ['module', 'function'] as const)
+      .combine('stage', ['constant', 'override', 'runtime'] as const)
+      .beginSubcases()
+  )
+  .fn(t => {
+    const staticAssert = buildStaticAssert(true, 'value', t.params.scope);
+    switch (t.params.stage) {
+      case 'constant':
+        t.expectCompileResult(true, `const value = true;\n${staticAssert}`);
+        break;
+      case 'override':
+        t.expectCompileResult(false, `override value = true;\n${staticAssert}`);
+        break;
+      case 'runtime':
+        t.expectCompileResult(false, `var<private> value = true;\n${staticAssert}`);
+        break;
+    }
+  });