chore: add test for Keyword::lookup_keyword completeness (#906)

noir-lang · Feb 24, 2023 · 269269c · 269269c
1 parent 93d83bf
commit 269269c
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/noirc_frontend/Cargo.toml b/crates/noirc_frontend/Cargo.toml
@@ -17,3 +17,7 @@ chumsky.workspace = true
 thiserror.workspace = true
 smol_str.workspace = true
 rustc-hash = "1.1.0"
+
+[dev-dependencies]
+strum = "0.24"
+strum_macros = "0.24"
diff --git a/crates/noirc_frontend/src/lexer/token.rs b/crates/noirc_frontend/src/lexer/token.rs
@@ -402,6 +402,7 @@ impl AsRef<str> for Attribute {
 }
 
 #[derive(PartialEq, Eq, Hash, Debug, Copy, Clone, PartialOrd, Ord)]
+#[cfg_attr(test, derive(strum_macros::EnumIter))]
 // Special Keywords allowed in the target language
 pub enum Keyword {
     As,
@@ -502,6 +503,29 @@ impl Keyword {
     }
 }
 
+#[cfg(test)]
+mod keywords {
+    use strum::IntoEnumIterator;
+
+    use super::{Keyword, Token};
+
+    #[test]
+    fn lookup_consistency() {
+        for keyword in Keyword::iter() {
+            let resolved_token =
+                Keyword::lookup_keyword(&format!("{keyword}")).unwrap_or_else(|| {
+                    panic!("Keyword::lookup_keyword couldn't find Keyword {}", keyword)
+                });
+
+            assert_eq!(
+                resolved_token,
+                Token::Keyword(keyword),
+                "Keyword::lookup_keyword returns unexpected Keyword"
+            )
+        }
+    }
+}
+
 pub struct Tokens(pub Vec<SpannedToken>);
 
 impl<'a> From<Tokens>