Forbid invalid attributes on assignments (#2412)

src/parser.rs

@@ -480,18 +480,31 @@ impl<'run, 'src> Parser<'run, 'src> {
   fn parse_assignment(
     &mut self,
     export: bool,
-    attributes: BTreeSet<Attribute<'_>>,
+    attributes: BTreeSet<Attribute<'src>>,
   ) -> CompileResult<'src, Assignment<'src>> {
     let name = self.parse_name()?;
     self.presume(ColonEquals)?;
     let value = self.parse_expression()?;
     self.expect_eol()?;
+
+    let private = attributes.contains(&Attribute::Private);
+
+    for attribute in attributes {
+      if attribute != Attribute::Private {
+        return Err(name.error(CompileErrorKind::InvalidAttribute {
+          item_kind: "Assignment",
+          item_name: name.lexeme(),
+          attribute,
+        }));
+      }
+    }
+
     Ok(Assignment {
       file_depth: self.file_depth,
       export,
       name,
       value,
-      private: name.lexeme().starts_with('_') || attributes.contains(&Attribute::Private),
+      private: private || name.lexeme().starts_with('_'),
     })
   }
 

tests/assignment.rs

@@ -3,29 +3,54 @@ use super::*;
 test! {
   name: set_export_parse_error,
   justfile: "
-  set export := fals
+    set export := fals
   ",
   stdout: "",
   stderr: "
     error: Expected keyword `true` or `false` but found identifier `fals`
-      |
-    1 | set export := fals
-      |               ^^^^
+     ——▶ justfile:1:15
+      │
+    1 │ set export := fals
+      │               ^^^^
   ",
   status: EXIT_FAILURE,
 }
 
 test! {
-  name: set_export_parse_error_EOL,
+  name: set_export_parse_error_eol,
   justfile: "
-  set export := fals
+    set export :=
   ",
   stdout: "",
   stderr: "
-    error: Expected keyword `true` or `false` but found `end of line`
-      |
-    1 | set export := 
-      |               ^
+    error: Expected identifier, but found end of line
+     ——▶ justfile:1:14
+      │
+    1 │ set export :=
+      │              ^
   ",
   status: EXIT_FAILURE,
-}
+}
+
+#[test]
+fn invalid_attributes_are_an_error() {
+  Test::new()
+    .justfile(
+      "
+        [group: 'bar']
+        x := 'foo'
+      ",
+    )
+    .args(["--evaluate", "x"])
+    .stderr(
+      "
+        error: Assignment `x` has invalid attribute `group`
+         ——▶ justfile:2:1
+          │
+        2 │ x := 'foo'
+          │ ^
+      ",
+    )
+    .status(EXIT_FAILURE)
+    .run();
+}

tests/lib.rs

@@ -36,6 +36,7 @@ mod allow_duplicate_variables;
 mod assert_stdout;
 mod assert_success;
 mod assertions;
+mod assignment;
 mod attributes;
 mod backticks;
 mod byte_order_mark;