test: body-plz - chunked

Author: darkseid-is

body-plz/src/reader/chunked_reader.rs

@@ -488,16 +488,38 @@ pub(crate) mod tests {
     }
 
     #[test]
-    fn test_get_size() {
+    fn test_mark_size_chunk_true() {
         let data = "7\r\n";
         let mut buf = BytesMut::from(data);
         let mut cbuf = Cursor::new(&mut buf);
         let result = ChunkReaderState::mark_size_chunk(&mut cbuf);
         assert!(result);
+        assert_eq!(cbuf.position(), 1);
         let size = ChunkReaderState::try_get_size(&mut cbuf).unwrap();
         assert_eq!(size, 7);
     }
 
+    #[test]
+    fn test_mark_size_chunk_true_with_extension() {
+        let data = "7; hola amigo\r\n";
+        let mut buf = BytesMut::from(data);
+        let mut cbuf = Cursor::new(&mut buf);
+        let result = ChunkReaderState::mark_size_chunk(&mut cbuf);
+        assert!(result);
+        assert_eq!(cbuf.position(), data.len() - 2);
+        let size = ChunkReaderState::try_get_size(&mut cbuf).unwrap();
+        assert_eq!(size, 7);
+    }
+
+    #[test]
+    fn test_mark_size_chunk_false() {
+        let data = "7\r";
+        let mut buf = BytesMut::from(data);
+        let mut cbuf = Cursor::new(&mut buf);
+        let result = ChunkReaderState::mark_size_chunk(&mut cbuf);
+        assert!(!result);
+    }
+
     #[test]
     fn test_chunked_reader_extra_data() {
         let data = "\r\n\

body-plz/src/variants/chunked.rs

@@ -81,6 +81,32 @@ mod tests {
         chunk_vec
     }
 
+    #[test]
+    fn test_chunk_type_len() {
+        let buf = BytesMut::from("data\r\n");
+        let size_chunk = ChunkType::Size(buf.clone());
+        assert_eq!(size_chunk.len(), 6);
+
+        let chunk = ChunkType::Chunk(buf.clone());
+        assert_eq!(chunk.len(), 6);
+
+        let extra = ChunkType::Extra(buf.clone());
+        assert_eq!(extra.len(), 6);
+
+        let last_chunk = ChunkType::LastChunk(buf.clone());
+        assert_eq!(last_chunk.len(), 3);
+
+        let end_crlf = ChunkType::EndCRLF(buf.clone());
+        assert_eq!(end_crlf.len(), 2);
+
+        let raw_headers = "content-type: application/json\r\n\
+                           content-encoding: gzip\r\n\r\n";
+
+        let header_map = HeaderMap::from(BytesMut::from(raw_headers));
+        let trailers = ChunkType::Trailers(header_map);
+        assert_eq!(trailers.len(), 58);
+    }
+
     #[test]
     fn test_total_chunk_size() {
         let buf = BytesMut::from("data\r\n");

body-plz/src/variants/mod.rs

@@ -34,3 +34,51 @@ impl Body {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_variants_body_push_chunk() {
+        let mut body = Body::Chunked(Vec::new());
+        let buf = BytesMut::from("data\r\n");
+        body.push_chunk(ChunkType::Chunk(buf.clone()));
+        assert_eq!(body, Body::Chunked(vec![ChunkType::Chunk(buf.clone())]));
+
+        body.push_chunk(ChunkType::Size(buf.clone()));
+        assert_eq!(
+            body,
+            Body::Chunked(vec![
+                ChunkType::Chunk(buf.clone()),
+                ChunkType::Size(buf.clone())
+            ])
+        );
+    }
+
+    #[test]
+    fn test_variants_body_into_bytes_raw() {
+        let buf = BytesMut::from("data\r\n");
+        let body = Body::Raw(buf.clone());
+        assert_eq!(body.into_bytes(), Some(buf));
+    }
+
+    #[test]
+    fn test_variants_body_into_bytes_chunk() {
+        let body = Body::Chunked(Vec::new());
+        assert_eq!(body.into_bytes(), None);
+    }
+
+    #[test]
+    fn test_variants_body_into_chunks_raw() {
+        let body = Body::Raw(BytesMut::from("data\r\n"));
+        assert_eq!(body.into_chunks(), Vec::new());
+    }
+
+    #[test]
+    fn test_variants_body_into_chunks_chunk() {
+        let buf = BytesMut::from("data\r\n");
+        let body = Body::Chunked(vec![ChunkType::Chunk(buf.clone())]);
+        assert_eq!(body.into_chunks(), vec![ChunkType::Chunk(buf.clone())]);
+    }
+}