From 8c42b8ae86ac0389f428b434373167f4fbccd4f9 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Sat, 14 Oct 2017 00:28:49 -0700
Subject: [PATCH] more tests

---
 src/reader.rs | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)

diff --git a/src/reader.rs b/src/reader.rs
index 0da354c8..21d657d8 100644
--- a/src/reader.rs
+++ b/src/reader.rs
@@ -790,6 +790,89 @@ mod tests {
         assert!(payload.eof());
     }
 
+    #[test]
+    fn test_http_request_chunked_payload_and_next_message() {
+        let mut buf = Buffer::new(
+            "GET /test HTTP/1.1\r\n\
+             transfer-encoding: chunked\r\n\r\n");
+
+        let mut reader = Reader::new();
+
+        let (req, mut payload) = reader_parse_ready!(reader.parse(&mut buf));
+        assert!(req.chunked().unwrap());
+        assert!(!payload.eof());
+
+        buf.feed_data(
+            "4\r\ndata\r\n4\r\nline\r\n0\r\n\r\n\
+             POST /test2 HTTP/1.1\r\n\
+             transfer-encoding: chunked\r\n\r\n");
+
+        let (req2, payload2) = reader_parse_ready!(reader.parse(&mut buf));
+        assert_eq!(*req2.method(), Method::POST);
+        assert!(req2.chunked().unwrap());
+        assert!(!payload2.eof());
+
+        assert_eq!(payload.readall().unwrap().as_ref(), b"dataline");
+        assert!(payload.eof());
+    }
+
+    #[test]
+    fn test_http_request_chunked_payload_chunks() {
+        let mut buf = Buffer::new(
+            "GET /test HTTP/1.1\r\n\
+             transfer-encoding: chunked\r\n\r\n");
+
+        let mut reader = Reader::new();
+        let (req, mut payload) = reader_parse_ready!(reader.parse(&mut buf));
+        assert!(req.chunked().unwrap());
+        assert!(!payload.eof());
+
+        buf.feed_data("4\r\ndata\r");
+        not_ready!(reader.parse(&mut buf));
+
+        buf.feed_data("\n4");
+        not_ready!(reader.parse(&mut buf));
+
+        buf.feed_data("\r");
+        not_ready!(reader.parse(&mut buf));
+        buf.feed_data("\n");
+        not_ready!(reader.parse(&mut buf));
+
+        buf.feed_data("li");
+        not_ready!(reader.parse(&mut buf));
+
+        buf.feed_data("ne\r\n0\r\n");
+        not_ready!(reader.parse(&mut buf));
+
+        //buf.feed_data("test: test\r\n");
+        //not_ready!(reader.parse(&mut buf));
+
+        assert_eq!(payload.readall().unwrap().as_ref(), b"dataline");
+        assert!(!payload.eof());
+
+        buf.feed_data("\r\n");
+        not_ready!(reader.parse(&mut buf));
+        assert!(payload.eof());
+    }
+
+    #[test]
+    fn test_parse_chunked_payload_chunk_extension() {
+        let mut buf = Buffer::new(
+            "GET /test HTTP/1.1\r\n\
+             transfer-encoding: chunked\r\n\r\n");
+
+        let mut reader = Reader::new();
+        let (req, mut payload) = reader_parse_ready!(reader.parse(&mut buf));
+        assert!(req.chunked().unwrap());
+        assert!(!payload.eof());
+
+        buf.feed_data("4;test\r\ndata\r\n4\r\nline\r\n0\r\n\r\n"); // test: test\r\n\r\n")
+        not_ready!(reader.parse(&mut buf));
+        assert!(!payload.eof());
+        assert_eq!(payload.readall().unwrap().as_ref(), b"dataline");
+        assert!(payload.eof());
+    }
+
     /*#[test]
     #[should_panic]
     fn test_parse_multiline() {