From 47bf17cb0b23ab093185286df450290dd3fcfd0f Mon Sep 17 00:00:00 2001
From: nichkara <nina.reiss@nickr.eu>
Date: Wed, 3 Sep 2025 09:36:54 +0200
Subject: [PATCH] Add tests for comment removal

---
 src/tests.rs | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/tests.rs b/src/tests.rs
index 76202aa..50283a5 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -1,5 +1,5 @@
 #[cfg(test)]
-use crate::tcl::{commands, words};
+use crate::tcl::{commands, comments, words};
 
 #[test]
 fn test_commands_working() {
@@ -66,3 +66,16 @@ fn test_words_quoted() {
     let case: Vec<String> = words(testcase);
     assert_eq!(case, verify);
 }
+
+#[test]
+fn test_remove_comments() {
+    let testcase_1: &str = "test#noreplacement";
+    let testcase_2: &str = ";#replace_this\n";
+    let verify_1: &str = "test#noreplacement";
+    let verify_2: &str = ";";
+    let case_1: &str = comments(testcase_1).as_str();
+    let case_2: &str = comments(testcase_2).as_str();
+
+    assert_eq!(verify_1, case_1);
+    assert_eq!(verify_2, case_2);
+}