From c3456fbf7851bc44d554d8b5ab8d43ab7f69f03e Mon Sep 17 00:00:00 2001 From: FredApps <153173854+FredApps@users.noreply.github.com> Date: Sat, 7 Nov 2026 15:45:26 +0100 Subject: [PATCH] Update budget test decoders to the buffered input type --- crates/doc/src/vectorcraft.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/crates/doc/src/vectorcraft.rs b/crates/doc/src/vectorcraft.rs index 6007824f..1f18c7e6 100644 --- a/crates/doc/src/vectorcraft.rs +++ b/crates/doc/src/vectorcraft.rs @@ -632,7 +632,7 @@ mod tests { fn duplicate_copy_cost_is_checked_before_the_stream_is_read() { let limits = crate::Limits { max_total: 12, ..Default::default() }; let mut budget = Budget::new(&limits); - let tail = Tail::Stream { dec: Box::new(GzDecoder::new(&b"not a gzip stream"[..])), read: vec![] }; + let tail = Tail::Stream { dec: Box::new(gzip_decoder(&b"not a gzip stream"[..], usize::MAX)), read: vec![] }; let result = tail.ranges(&[[0, 8], [0, 8]], &mut budget); assert!(matches!(result, Err(Error::LimitExceeded(_))), "{result:?}"); assert_eq!(budget.left(), 12); @@ -642,7 +642,7 @@ mod tests { fn duplicate_copies_can_use_exactly_the_remaining_budget() { let limits = crate::Limits { max_total: 16, ..Default::default() }; let mut budget = Budget::new(&limits); - let tail = Tail::Stream { dec: Box::new(GzDecoder::new(&b"not a gzip stream"[..])), read: b"12345678".to_vec() }; + let tail = Tail::Stream { dec: Box::new(gzip_decoder(&b"not a gzip stream"[..], usize::MAX)), read: b"12345678".to_vec() }; let result = tail.ranges(&[[0, 8], [0, 8]], &mut budget).unwrap(); assert_eq!(result, [Some(b"12345678".to_vec()), Some(b"12345678".to_vec())]); assert_eq!(budget.left(), 0); @@ -652,7 +652,7 @@ mod tests { fn rendition_copy_size_overflow_is_rejected_before_reading() { let limits = crate::Limits { max_total: u64::MAX, ..Default::default() }; let mut budget = Budget::new(&limits); - let tail = Tail::Stream { dec: Box::new(GzDecoder::new(&b"not a gzip stream"[..])), read: vec![] }; + let tail = Tail::Stream { dec: Box::new(gzip_decoder(&b"not a gzip stream"[..], usize::MAX)), read: vec![] }; assert!(matches!(tail.ranges(&[[0, u64::MAX], [0, 1]], &mut budget), Err(Error::LimitExceeded(_)))); assert_eq!(budget.left(), u64::MAX); }