pauljd/craft-libs
1
0
Fork 0
mirror of https://github.com/storytold/craft-libs.git synced 2026-11-10 10:03:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update budget test decoders to the buffered input type

Browse Source
This commit is contained in:
FredApps
2026-11-07 15:45:26 +01:00
parent c8ea8be292
commit c3456fbf78
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/doc/src/vectorcraft.rs
+3 -3
View File
@@ -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);
}