From bba3950c37d4fa32ccf18ed1b093ecf48aec419d Mon Sep 17 00:00:00 2001
From: Dirkjan Ochtman <dirkjan@ochtman.nl>
Date: Wed, 8 Feb 2017 10:59:57 +0100
Subject: Add unit tests for whitespace splitter

---
 askama/src/parser.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/askama/src/parser.rs b/askama/src/parser.rs
index f338456..bb59b48 100644
--- a/askama/src/parser.rs
+++ b/askama/src/parser.rs
@@ -186,3 +186,26 @@ pub fn parse(src: &str) -> Vec<Node> {
         IResult::Incomplete(_) => panic!("parsing incomplete"),
     }
 }
+
+#[cfg(test)]
+mod tests {
+    fn check_ws_split(s: &str, res: &(&[u8], &[u8], &[u8])) {
+        let node = super::split_ws_parts(s.as_bytes());
+        match node {
+            super::Node::Lit(lws, s, rws) => {
+                assert_eq!(lws, res.0);
+                assert_eq!(s, res.1);
+                assert_eq!(rws, res.2);
+            },
+            _ => { panic!("fail"); },
+        }
+    }
+    #[test]
+    fn test_ws_splitter() {
+        check_ws_split("a", &(b"", b"a", b""));
+        check_ws_split("", &(b"", b"", b""));
+        check_ws_split("\ta", &(b"\t", b"a", b""));
+        check_ws_split("b\n", &(b"", b"b", b"\n"));
+        check_ws_split(" \t\r\n", &(b" \t\r\n", b"", b""));
+    }
+}
-- 
cgit