From 0c4be634de862e68ef7498db96ac336239f2e5e2 Mon Sep 17 00:00:00 2001
From: JIe <depressedjie@outlook.com>
Date: Thu, 21 Nov 2024 09:54:57 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8D=95=E5=85=83=E6=B5=8B?=
 =?UTF-8?q?=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/prehandle.rs | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/prehandle.rs b/src/prehandle.rs
index 6eb93a6..1eace6d 100644
--- a/src/prehandle.rs
+++ b/src/prehandle.rs
@@ -23,9 +23,6 @@ fn filter_by_string(reader: Vec<String>, filter_str: &str) ->Vec<String>{
 }
 
 fn take(reader: &mut Vec<String>, start_index: u32, end_index: u32) -> Vec<String> {
-    if end_index < 0 && start_index < 0{
-        panic!("start_index and end_index must be positive");
-    }
     if end_index == 0 && start_index == 0{
         return reader.to_owned();
     }
@@ -40,4 +37,24 @@ fn take(reader: &mut Vec<String>, start_index: u32, end_index: u32) -> Vec<Strin
         body.append(&mut reader[start_index as usize..].to_owned());
     }
     body
+}
+
+#[cfg(test)]
+mod test{
+    #[test]
+    fn test_filter_by_string(){
+        let reader = vec!["a".to_string(), "b".to_string(), "c".to_string()];
+        let filter_str = "a";
+        let result = super::filter_by_string(reader, filter_str);
+        assert_eq!(result, vec!["a".to_string()]);
+    }
+
+    #[test]
+    fn test_take(){
+        let reader = vec!["a".to_string(), "b".to_string(), "c".to_string()];
+        let start_index = 1;
+        let end_index = 2;
+        let result = super::take(&mut reader.to_owned(), start_index, end_index);
+        assert_eq!(result, vec!["a".to_string(), "b".to_string()]);
+    }
 }
\ No newline at end of file