From ba0dc8bd8f655483ed9b5a831778a1f818ca1808 Mon Sep 17 00:00:00 2001
From: Luni-4 <luni-4@hotmail.it>
Date: Thu, 25 Mar 2021 15:44:42 +0100
Subject: [PATCH] Add a test for the remove comments API

---
 src/comment_rm.rs | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/comment_rm.rs b/src/comment_rm.rs
index a5c45afd6..f30351a9c 100644
--- a/src/comment_rm.rs
+++ b/src/comment_rm.rs
@@ -91,3 +91,42 @@ impl Callback for CommentRm {
         Ok(())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use std::path::PathBuf;
+
+    use crate::{CcommentParser, ParserTrait};
+
+    use super::rm_comments;
+
+    const SOURCE_CODE: &str = "/* Remove this code block */\n\
+                               int a = 42; // Remove this comment\n\
+                               // Remove this comment\n\
+                               int b = 42;\n\
+                               /* Remove\n\
+                                * this\n\
+                                * comment\n\
+                                */";
+
+    const SOURCE_CODE_NO_COMMENTS: &str = "\n\
+                                           int a = 42; \n\
+                                           \n\
+                                           int b = 42;\n\
+                                           \n\
+                                           \n\
+                                           \n\
+                                           \n";
+
+    #[test]
+    fn ccomment_remove_comments() {
+        let path = PathBuf::from("foo.c");
+        let mut trimmed_bytes = SOURCE_CODE.as_bytes().to_vec();
+        trimmed_bytes.push(b'\n');
+        let parser = CcommentParser::new(trimmed_bytes, &path, None);
+
+        let no_comments = rm_comments(&parser).unwrap();
+
+        assert_eq!(no_comments.as_slice(), SOURCE_CODE_NO_COMMENTS.as_bytes());
+    }
+}