YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_http_protocol: add benchmark for request from bytes

Browse source
This commit is contained in:
Joakim Frostegård 2020-07-20 20:27:42 +02:00
parent 95e25710dc
commit ea4b560faa
9 changed files with 1071 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

38
aquatic_http_protocol/src/request.rs
View file

@ -109,6 +109,7 @@ impl ScrapeRequest {
}
#[derive(Debug)]
pub enum RequestParseError {
NeedMoreData,
Invalid(anyhow::Error),
@ -408,13 +409,8 @@ mod tests {
assert!(f("%å7").is_err());
}
#[test]
fn test_announce_request_from_path(){
let parsed_request = Request::from_http_get_path(
ANNOUNCE_REQUEST_PATH
).unwrap();
let reference_request = Request::Announce(AnnounceRequest {
fn get_reference_announce_request() -> Request {
Request::Announce(AnnounceRequest {
info_hash: InfoHash(REFERENCE_INFO_HASH),
peer_id: PeerId(REFERENCE_PEER_ID),
port: 12345,
@ -423,7 +419,33 @@ mod tests {
compact: true,
numwant: Some(0),
key: Some("4ab4b877".into())
});
})
}
#[test]
fn test_announce_request_from_bytes(){
let mut bytes = Vec::new();
bytes.extend_from_slice(b"GET ");
bytes.extend_from_slice(&ANNOUNCE_REQUEST_PATH.as_bytes());
bytes.extend_from_slice(b" HTTP/1.1\r\n\r\n");
let parsed_request = Request::from_bytes(
&bytes[..]
).unwrap();
let reference_request = get_reference_announce_request();
assert_eq!(parsed_request, reference_request);
}
#[test]
fn test_announce_request_from_path(){
let parsed_request = Request::from_http_get_path(
ANNOUNCE_REQUEST_PATH
).unwrap();
let reference_request = get_reference_announce_request();
assert_eq!(parsed_request, reference_request);
}