Check no remaining token after the first literal

dtolnay · Oct 2, 2023 · 137ae33 · 137ae33
1 parent 258e9e8
commit 137ae33
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/tests/test_lit.rs b/tests/test_lit.rs
@@ -14,13 +14,12 @@ use std::str::FromStr;
 use syn::{Lit, LitFloat, LitInt, LitStr};
 
 fn lit(s: &str) -> Lit {
-    match TokenStream::from_str(s)
-        .unwrap()
-        .into_iter()
-        .next()
-        .unwrap()
-    {
-        TokenTree::Literal(lit) => Lit::new(lit),
+    let mut tokens = TokenStream::from_str(s).unwrap().into_iter();
+    match tokens.next().unwrap() {
+        TokenTree::Literal(lit) => {
+            assert!(tokens.next().is_none());
+            Lit::new(lit)
+        }
         _ => panic!(),
     }
 }