Fix empty trim_newline panic, add impl macro test

diff --git a/src/utils.rs b/src/utils.rs
index 195ca265c1fb43a51e186b0268cbef19938e6362..aaa838bf043c179f9dabe1530ec5d15817b99d7d 100644 (file)
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -146,7 +146,11 @@ pub fn semicolon_for_stmt(stmt: &ast::Stmt) -> bool {
 pub fn trim_newlines(input: &str) -> &str {
     let start = input.find(|c| c != '\n' && c != '\r').unwrap_or(0);
     let end = input.rfind(|c| c != '\n' && c != '\r').unwrap_or(0) + 1;
-    &input[start..end]
+    if start == 0 && end == 1 {
+        input
+    } else {
+        &input[start..end]
+    }
 }
 
 #[inline]

diff --git a/tests/source/impls.rs b/tests/source/impls.rs
index c15ae12325e0927f4a4c7b42680219f0173f5033..4382c4fd0e61899464945e284321945caf20babb 100644 (file)
--- a/tests/source/impls.rs
+++ b/tests/source/impls.rs
@@ -51,3 +51,10 @@ fn foo() {}
         }
     }
 }
+
+impl Foo { add_fun!(); }
+
+impl Blah {
+    fn boop() {}
+    add_fun!();
+}

diff --git a/tests/target/impls.rs b/tests/target/impls.rs
index 972991572a50aa10e112e97ebd55a69c9da77d94..35ddc23a20a5e669ee19566fca1598e913f87247 100644 (file)
--- a/tests/target/impls.rs
+++ b/tests/target/impls.rs
@@ -63,3 +63,12 @@ fn foo() {}
         }
     }
 }
+
+impl Foo {
+    add_fun!();
+}
+
+impl Blah {
+    fn boop() {}
+    add_fun!();
+}