use crate::fetch::Cursor; fn mock_reader(path: &str) -> git_packetline::StreamingPeekableIter<Cursor> { use crate::fixture_bytes; let buf = fixture_bytes(path); git_packetline::StreamingPeekableIter::new(Cursor::new(buf), &[git_packetline::PacketLine::Flush])} fn id(hex: &str) -> git_hash::ObjectId { git_hash::ObjectId::from_hex(hex.as_bytes()).expect("expect valid hex id")} mod v1 { mod from_line_reader { use crate::fetch::response::{id, mock_reader}; #[cfg(feature = "async-client")] use futures_lite::io::AsyncReadExt; use git_protocol::fetch::{ self, response::{Acknowledgement, ShallowUpdate}, }; use git_transport::Protocol; #[cfg(feature = "blocking-client")] use std::io::Read; #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn clone() -> crate::Result { let mut provider = mock_reader("v1/clone-only.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?; assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]); assert!(r.has_pack()); let mut buf = Vec::new(); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 1090, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn shallow_clone() -> crate::Result { let mut provider = mock_reader("v1/clone-deepen-1.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?; assert_eq!( r.shallow_updates(), &[ShallowUpdate::Shallow(id("808e50d724f604f69ab93c6da2919c014667bedb"))] ); assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]); assert!(r.has_pack()); let mut buf = Vec::new(); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 1989, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn empty_shallow_clone_due_to_depth_being_too_high() -> crate::Result { let mut provider = mock_reader("v1/clone-deepen-5.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?; assert!(r.shallow_updates().is_empty()); assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]); assert!(r.has_pack()); let mut buf = Vec::new(); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 1988, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn fetch_acks_without_pack() -> crate::Result { let mut provider = mock_reader("v1/fetch-no-pack.response"); let r = fetch::Response::from_line_reader(Protocol::V1, &mut provider.as_read_without_sidebands()).await?; assert_eq!( r.acknowledgements(), &[ Acknowledgement::Common(id("47ee0b7fe4f3a7d776c78794873e6467e1c47e59")), Acknowledgement::Common(id("3f02c0ad360d96e8dbba92f97b42ebbaa4319db1")), Acknowledgement::Nak, ] ); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn fetch_acks_and_pack() -> crate::Result { let mut provider = mock_reader("v1/fetch.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?; assert_eq!( r.acknowledgements(), &[ Acknowledgement::Common(id("6504930888c9c5337e7e065c964f87b60d16a7d7")), Acknowledgement::Common(id("fe17165c392110d1305674c06e4aec35728bfab7")), Acknowledgement::Common(id("f22743895a3024bb0c958335981439f1fa747d57")), Acknowledgement::Ready, Acknowledgement::Nak, ] ); assert!(r.has_pack()); let mut buf = Vec::new(); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 9703, "should be able to read the whole pack"); Ok(()) } }}mod v2 { mod from_line_reader { use crate::fetch::response::{id, mock_reader}; #[cfg(feature = "async-client")] use futures_lite::io::AsyncReadExt; use git_protocol::fetch::{ self, response::{Acknowledgement, ShallowUpdate}, }; use git_transport::Protocol; #[cfg(feature = "blocking-client")] use std::io::Read; #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn clone() -> crate::Result { let mut provider = mock_reader("v2/clone-only.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?; assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile"); assert!(r.has_pack()); let mut buf = Vec::new(); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 1089, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn shallow_clone() -> crate::Result { let mut provider = mock_reader("v2/clone-deepen-1.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?; assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile"); assert_eq!( r.shallow_updates(), &[ShallowUpdate::Shallow(id("808e50d724f604f69ab93c6da2919c014667bedb"))] ); assert!(r.has_pack()); let mut pack = Vec::new(); reader.read_to_end(&mut pack).await?; assert_eq!(pack.len(), 1991, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn empty_shallow_clone() -> crate::Result { let mut provider = mock_reader("v2/clone-deepen-5.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?; assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile"); assert!(r.shallow_updates().is_empty(), "it should go straight to the packfile"); assert!(r.has_pack()); let mut pack = Vec::new(); let bytes_read = reader.read_to_end(&mut pack).await?; assert_eq!(bytes_read, 1989, "should be able to read the whole pack"); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn clone_with_sidebands() -> crate::Result { let mut provider = mock_reader("v2/clone-only-2.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?; assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile"); assert!(r.has_pack()); let mut buf = Vec::new(); reader.set_progress_handler(Some(Box::new(|is_err: bool, _data: &[u8]| { assert!(!is_err, "fixture does not have an error"); }) as git_transport::client::HandleProgress)); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 1643, "should be able to read the whole pack"); assert_eq!(&buf[..4], b"PACK"); assert_eq!( git_hash::ObjectId::from_20_bytes(&buf[buf.len() - 20..]).to_string(), "f34c9be7e0c3ef2c3ed7c62cc7791dbf6dc5ec9a" ); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn fetch_acks_without_pack() -> crate::Result { let mut provider = mock_reader("v2/fetch-no-pack.response"); let r = fetch::Response::from_line_reader(Protocol::V2, &mut provider.as_read_without_sidebands()).await?; assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak,]); Ok(()) } #[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))] async fn fetch_acks_and_pack() -> crate::Result { let mut provider = mock_reader("v2/fetch.response"); let mut reader = provider.as_read_without_sidebands(); let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?; assert_eq!( r.acknowledgements(), &[ Acknowledgement::Common(id("190c3f6b2319c1f4ec854215533caf8623f8f870")), Acknowledgement::Common(id("97c5a932b3940a09683e924ef6a92b31a6f7c6de")), Acknowledgement::Ready, ] ); assert!(r.has_pack()); let mut buf = Vec::new(); reader.set_progress_handler(Some(Box::new(|a: bool, b: &[u8]| { git_protocol::RemoteProgress::translate_to_progress(a, b, &mut git_features::progress::Discard) }) as git_transport::client::HandleProgress)); let bytes_read = reader.read_to_end(&mut buf).await?; assert_eq!(bytes_read, 5360, "should be able to read the whole pack"); Ok(()) } }}
use crate::fetch::Cursor;
fn mock_reader(path: &str) -> git_packetline::StreamingPeekableIter<Cursor> {
use crate::fixture_bytes;
let buf = fixture_bytes(path);
git_packetline::StreamingPeekableIter::new(Cursor::new(buf), &[git_packetline::PacketLine::Flush])
}
fn id(hex: &str) -> git_hash::ObjectId {
git_hash::ObjectId::from_hex(hex.as_bytes()).expect("expect valid hex id")
}
mod v1 {
mod from_line_reader {
use crate::fetch::response::{id, mock_reader};
#[cfg(feature = "async-client")]
use futures_lite::io::AsyncReadExt;
use git_protocol::fetch::{
self,
response::{Acknowledgement, ShallowUpdate},
};
use git_transport::Protocol;
#[cfg(feature = "blocking-client")]
use std::io::Read;
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn clone() -> crate::Result {
let mut provider = mock_reader("v1/clone-only.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?;
assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]);
assert!(r.has_pack());
let mut buf = Vec::new();
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 1090, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn shallow_clone() -> crate::Result {
let mut provider = mock_reader("v1/clone-deepen-1.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?;
assert_eq!(
r.shallow_updates(),
&[ShallowUpdate::Shallow(id("808e50d724f604f69ab93c6da2919c014667bedb"))]
);
assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]);
assert!(r.has_pack());
let mut buf = Vec::new();
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 1989, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn empty_shallow_clone_due_to_depth_being_too_high() -> crate::Result {
let mut provider = mock_reader("v1/clone-deepen-5.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?;
assert!(r.shallow_updates().is_empty());
assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak]);
assert!(r.has_pack());
let mut buf = Vec::new();
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 1988, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn fetch_acks_without_pack() -> crate::Result {
let mut provider = mock_reader("v1/fetch-no-pack.response");
let r = fetch::Response::from_line_reader(Protocol::V1, &mut provider.as_read_without_sidebands()).await?;
assert_eq!(
r.acknowledgements(),
&[
Acknowledgement::Common(id("47ee0b7fe4f3a7d776c78794873e6467e1c47e59")),
Acknowledgement::Common(id("3f02c0ad360d96e8dbba92f97b42ebbaa4319db1")),
Acknowledgement::Nak,
]
);
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn fetch_acks_and_pack() -> crate::Result {
let mut provider = mock_reader("v1/fetch.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V1, &mut reader).await?;
assert_eq!(
r.acknowledgements(),
&[
Acknowledgement::Common(id("6504930888c9c5337e7e065c964f87b60d16a7d7")),
Acknowledgement::Common(id("fe17165c392110d1305674c06e4aec35728bfab7")),
Acknowledgement::Common(id("f22743895a3024bb0c958335981439f1fa747d57")),
Acknowledgement::Ready,
Acknowledgement::Nak,
]
);
assert!(r.has_pack());
let mut buf = Vec::new();
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 9703, "should be able to read the whole pack");
Ok(())
}
}
}
mod v2 {
mod from_line_reader {
use crate::fetch::response::{id, mock_reader};
#[cfg(feature = "async-client")]
use futures_lite::io::AsyncReadExt;
use git_protocol::fetch::{
self,
response::{Acknowledgement, ShallowUpdate},
};
use git_transport::Protocol;
#[cfg(feature = "blocking-client")]
use std::io::Read;
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn clone() -> crate::Result {
let mut provider = mock_reader("v2/clone-only.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?;
assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile");
assert!(r.has_pack());
let mut buf = Vec::new();
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 1089, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn shallow_clone() -> crate::Result {
let mut provider = mock_reader("v2/clone-deepen-1.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?;
assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile");
assert_eq!(
r.shallow_updates(),
&[ShallowUpdate::Shallow(id("808e50d724f604f69ab93c6da2919c014667bedb"))]
);
assert!(r.has_pack());
let mut pack = Vec::new();
reader.read_to_end(&mut pack).await?;
assert_eq!(pack.len(), 1991, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn empty_shallow_clone() -> crate::Result {
let mut provider = mock_reader("v2/clone-deepen-5.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?;
assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile");
assert!(r.shallow_updates().is_empty(), "it should go straight to the packfile");
assert!(r.has_pack());
let mut pack = Vec::new();
let bytes_read = reader.read_to_end(&mut pack).await?;
assert_eq!(bytes_read, 1989, "should be able to read the whole pack");
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn clone_with_sidebands() -> crate::Result {
let mut provider = mock_reader("v2/clone-only-2.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?;
assert!(r.acknowledgements().is_empty(), "it should go straight to the packfile");
assert!(r.has_pack());
let mut buf = Vec::new();
reader.set_progress_handler(Some(Box::new(|is_err: bool, _data: &[u8]| {
assert!(!is_err, "fixture does not have an error");
}) as git_transport::client::HandleProgress));
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 1643, "should be able to read the whole pack");
assert_eq!(&buf[..4], b"PACK");
assert_eq!(
git_hash::ObjectId::from_20_bytes(&buf[buf.len() - 20..]).to_string(),
"f34c9be7e0c3ef2c3ed7c62cc7791dbf6dc5ec9a"
);
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn fetch_acks_without_pack() -> crate::Result {
let mut provider = mock_reader("v2/fetch-no-pack.response");
let r = fetch::Response::from_line_reader(Protocol::V2, &mut provider.as_read_without_sidebands()).await?;
assert_eq!(r.acknowledgements(), &[Acknowledgement::Nak,]);
Ok(())
}
#[maybe_async::test(feature = "blocking-client", async(feature = "async-client", async_std::test))]
async fn fetch_acks_and_pack() -> crate::Result {
let mut provider = mock_reader("v2/fetch.response");
let mut reader = provider.as_read_without_sidebands();
let r = fetch::Response::from_line_reader(Protocol::V2, &mut reader).await?;
assert_eq!(
r.acknowledgements(),
&[
Acknowledgement::Common(id("190c3f6b2319c1f4ec854215533caf8623f8f870")),
Acknowledgement::Common(id("97c5a932b3940a09683e924ef6a92b31a6f7c6de")),
Acknowledgement::Ready,
]
);
assert!(r.has_pack());
let mut buf = Vec::new();
reader.set_progress_handler(Some(Box::new(|a: bool, b: &[u8]| {
git_protocol::RemoteProgress::translate_to_progress(a, b, &mut git_features::progress::Discard)
}) as git_transport::client::HandleProgress));
let bytes_read = reader.read_to_end(&mut buf).await?;
assert_eq!(bytes_read, 5360, "should be able to read the whole pack");
Ok(())
}
}
}