v0.29.2

src/web_server/tests/part_01.rs

@@ -182,6 +182,43 @@ async fn validators_and_single_ranges_work() {
     assert_eq!(response.headers()[CONTENT_LENGTH], "10");
 }
 
+#[tokio::test]
+async fn multipart_byte_ranges_stream_each_requested_part() {
+    let tmp = TempDir::new().unwrap();
+    std::fs::write(tmp.path().join("file.txt"), b"0123456789").unwrap();
+    let service = service(tmp.path(), false);
+    let mut ranged = request(Method::GET, "/file.txt");
+    ranged
+        .headers_mut()
+        .insert(RANGE, "bytes=0-1,6-8".parse().unwrap());
+    let response = serve_request(&service, &ranged, "/file.txt").await;
+    assert_eq!(response.status(), StatusCode::PARTIAL_CONTENT);
+    let content_type = response.headers()[CONTENT_TYPE].to_str().unwrap().to_owned();
+    assert!(content_type.starts_with("multipart/byteranges; boundary="));
+    assert!(!response.headers().contains_key(CONTENT_RANGE));
+    let body = response.into_body().collect().await.unwrap().to_bytes();
+    let body = String::from_utf8(body.to_vec()).unwrap();
+    assert!(body.contains("Content-Range: bytes 0-1/10"));
+    assert!(body.contains("Content-Range: bytes 6-8/10"));
+    assert!(body.contains("\r\n01\r\n"));
+    assert!(body.contains("\r\n678\r\n"));
+}
+
+#[tokio::test]
+async fn too_many_byte_ranges_are_rejected() {
+    let tmp = TempDir::new().unwrap();
+    std::fs::write(tmp.path().join("file.txt"), b"0123456789").unwrap();
+    let service = service(tmp.path(), false);
+    let ranges = std::iter::repeat("0-0").take(17).collect::<Vec<_>>().join(",");
+    let mut ranged = request(Method::GET, "/file.txt");
+    ranged
+        .headers_mut()
+        .insert(RANGE, format!("bytes={ranges}").parse().unwrap());
+    let response = serve_request(&service, &ranged, "/file.txt").await;
+    assert_eq!(response.status(), StatusCode::RANGE_NOT_SATISFIABLE);
+    assert_eq!(response.headers()[CONTENT_RANGE], "bytes */10");
+}
+
 #[tokio::test]
 async fn listing_escapes_html_and_encodes_urls() {
     let tmp = TempDir::new().unwrap();