Skip to content

Commit a8e371b

Browse files
committed
Update async_zip
1 parent 9608969 commit a8e371b

File tree

2 files changed

+19
-8
lines changed

2 files changed

+19
-8
lines changed

Cargo.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ tracing = "0.1.40"
2626
axum = { version = "0.7.5", default-features = false, features = ["tokio", "http1"] }
2727
tokio = { version = "1.33.0", default-features = false, features = ["macros", "test-util"] }
2828
tower-http = { version = "0.6.1", default-features = false, features = ["fs"] }
29-
async_zip = { version = "0.0.15", default-features = false, features = ["tokio"] }
29+
async_zip = { version = "0.0.17", default-features = false, features = ["tokio"] }
3030
assert_matches = "1.5.0"
3131
rstest = { version = "0.23.0" }
3232
url = { version = "2.4.1" }

src/lib.rs

+18-7
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,6 @@ mod test {
663663
use rstest::*;
664664
use std::path::Path;
665665
use tokio::io::AsyncReadExt as _;
666-
use tokio_util::compat::TokioAsyncReadCompatExt;
667666

668667
#[rstest]
669668
#[case(CheckSupportMethod::Head)]
@@ -703,14 +702,16 @@ mod test {
703702

704703
assert_eq!(range.len(), file_size);
705704

706-
let mut reader = ZipFileReader::new(range.compat()).await.unwrap();
705+
let mut reader = ZipFileReader::with_tokio(tokio::io::BufReader::with_capacity(0, range))
706+
.await
707+
.unwrap();
707708

708709
assert_eq!(
709710
reader
710711
.file()
711712
.entries()
712713
.iter()
713-
.map(|e| e.entry().filename().as_str().unwrap_or(""))
714+
.map(|e| e.filename().as_str().unwrap_or(""))
714715
.collect::<Vec<_>>(),
715716
vec![
716717
"metadata.json",
@@ -720,7 +721,12 @@ mod test {
720721
);
721722

722723
// Get the number of performed requests so far
723-
let request_ranges = reader.inner_mut().get_mut().requested_ranges().await;
724+
let request_ranges = reader
725+
.inner_mut()
726+
.get_mut()
727+
.get_mut()
728+
.requested_ranges()
729+
.await;
724730
assert_eq!(request_ranges.len(), 1);
725731
assert_eq!(
726732
request_ranges[0].end - request_ranges[0].start,
@@ -737,8 +743,7 @@ mod test {
737743
let offset = entry.header_offset();
738744
// Get the size of the entry plus the header + size of the filename. We should also actually
739745
// include bytes for the extra fields but we don't have that information.
740-
let size =
741-
entry.entry().compressed_size() + 30 + entry.entry().filename().as_bytes().len() as u64;
746+
let size = entry.compressed_size() + 30 + entry.filename().as_bytes().len() as u64;
742747

743748
// The zip archive uses as BufReader which reads in chunks of 8192. To ensure we prefetch
744749
// enough data we round the size up to the nearest multiple of the buffer size.
@@ -749,6 +754,7 @@ mod test {
749754
reader
750755
.inner_mut()
751756
.get_mut()
757+
.get_mut()
752758
.prefetch(offset..offset + size as u64)
753759
.await;
754760

@@ -763,7 +769,12 @@ mod test {
763769
.unwrap();
764770

765771
// Get the number of performed requests
766-
let request_ranges = reader.inner_mut().get_mut().requested_ranges().await;
772+
let request_ranges = reader
773+
.inner_mut()
774+
.get_mut()
775+
.get_mut()
776+
.requested_ranges()
777+
.await;
767778

768779
assert_eq!(contents, r#"{"conda_pkg_format_version": 2}"#);
769780
assert_eq!(request_ranges.len(), 2);

0 commit comments

Comments
 (0)