add a test, fix a bug, etc

ut-utp · Sep 21, 2019 · 39d8b06 · 39d8b06
1 parent a8dc2af
commit 39d8b06
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -261,12 +261,13 @@ impl IdentifierReplace<'_> {
 
 impl VisitMut for IdentifierReplace<'_> {
     fn visit_expr_mut(&mut self, expr: &mut Expr) {
-        println!("{:?}", expr);
+        // println!("{:?}", expr);
 
         if let Expr::Path(ExprPath { path, .. }) = expr {
             println!("match!!! {:?}", path);
             if path.is_ident(self.identifier_name) {
-                *expr = parse_quote!(#(self.substitution));
+                let num = self.substitution;
+                *expr = parse_quote!(#num);
                 return;
             }
         // This is unfortunately not permitted:

diff --git a/tests/repeat_n.rs b/tests/repeat_n.rs
@@ -13,13 +13,18 @@ mod tests {
     #[allow(unused)]
     struct Bar<const B: usize>;
 
-    // #[test]
-    // fn test() {
-    //     repeat_with_n!(10, n, { impl Foo for Bar<{n as usize}> {} });
-    // }
+    #[test]
+    fn test() {
+        repeat_with_n!(10, n, { impl Foo for Bar<{n as usize}> {} });
+    }
 
     #[test]
     fn with_macro() {
         repeat_with_n!(10, longer_identifier, { println!("{}", longer_identifier); });
     }
+
+    #[test]
+    fn nested_macro() {
+        repeat_with_n!(10, y, { println!("{}", format!("{}", format!("{}{}{}{}", y, y, y, y))); });
+    }
 }