[wgsl-in] Fix error message for invalid textureStore() on 2d_array

Fixes #2431

src/front/wgsl/lower/mod.rs

@@ -1937,7 +1937,10 @@ impl<'source, 'temp> Lowerer<'source, 'temp> {
 
                             let (_, arrayed) = ctx.image_data(image, image_span)?;
                             let array_index = arrayed
-                                .then(|| self.expression(args.next()?, ctx.reborrow()))
+                                .then(|| {
+                                    args.min_args += 1;
+                                    self.expression(args.next()?, ctx.reborrow())
+                                })
                                 .transpose()?;
 
                             let value = self.expression(args.next()?, ctx.reborrow())?;

src/front/wgsl/tests.rs

@@ -481,3 +481,22 @@ fn parse_alias() {
     )
     .unwrap();
 }
+
+#[test]
+fn parse_texture_store_expecting_four_args() {
+    let error = parse_str(
+        "
+        @group(0) @binding(0) var tex_los_res: texture_storage_2d_array<rg11b10float, write>;
+        @compute
+        @workgroup_size(1)
+        fn main(@builtin(global_invocation_id) id: vec3<u32>) {
+            var color = vec4(1, 1, 1, 1);
+            textureStore(tex_los_res, id, color);
+        }",
+    )
+    .unwrap_err();
+    assert_eq!(
+        error.message(),
+        "wrong number of arguments: expected 4, found 3"
+    );
+}