mirror of
https://github.com/ducaale/xh.git
synced 2025-05-05 15:32:50 +00:00
move compress request body test cases to own file
This commit is contained in:
zuisong
parent
c4bbdba60e
commit
abc8c03b10
197
tests/cases/compress_request_body.rs
Normal file
197
tests/cases/compress_request_body.rs
Normal file
@ -0,0 +1,197 @@
|
||||
use std::{fs::OpenOptions, io::Read as _};
|
||||
|
||||
use hyper::header::HeaderValue;
|
||||
|
||||
use crate::prelude::*;
|
||||
use std::io::Write;
|
||||
|
||||
fn zlib_decode(bytes: Vec<u8>) -> std::io::Result<String> {
|
||||
let mut z = flate2::read::ZlibDecoder::new(&bytes[..]);
|
||||
let mut s = String::new();
|
||||
z.read_to_string(&mut s)?;
|
||||
Ok(s)
|
||||
}
|
||||
|
||||
fn server() -> server::Server {
|
||||
server::http(|req| async move {
|
||||
match req.uri().path() {
|
||||
"/deflate" => {
|
||||
assert_eq!(
|
||||
req.headers().get(hyper::header::CONTENT_ENCODING),
|
||||
Some(HeaderValue::from_static("deflate")).as_ref()
|
||||
);
|
||||
|
||||
let compressed_body = req.body().await;
|
||||
let body = zlib_decode(compressed_body).unwrap();
|
||||
hyper::Response::builder()
|
||||
.header("date", "N/A")
|
||||
.header("Content-Type", "text/plain")
|
||||
.body(body.into())
|
||||
.unwrap()
|
||||
}
|
||||
"/normal" => {
|
||||
let body = req.body_as_string().await;
|
||||
hyper::Response::builder()
|
||||
.header("date", "N/A")
|
||||
.header("Content-Type", "text/plain")
|
||||
.body(body.into())
|
||||
.unwrap()
|
||||
}
|
||||
_ => panic!("unknown path"),
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_request_body_json() {
|
||||
let server = server();
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"-x",
|
||||
"-j",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1010
|
||||
|
||||
{{"key":"{c}"}}
|
||||
"#, c = "1".repeat(1000),});
|
||||
}
|
||||
#[test]
|
||||
fn compress_request_body_form() {
|
||||
let server = server();
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"-x",
|
||||
"-x",
|
||||
"-f",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1004
|
||||
|
||||
key={c}
|
||||
"#, c = "1".repeat(1000),});
|
||||
}
|
||||
#[test]
|
||||
fn skip_compression_when_compression_ratio_is_negative() {
|
||||
let server = server();
|
||||
get_command()
|
||||
.arg(format!("{}/normal", server.base_url()))
|
||||
.args([&format!("key={}", "1"), "-x", "-f", "--pretty=none"])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 5
|
||||
|
||||
key={c}
|
||||
"#, c = "1"});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_request_body_force() {
|
||||
let server = server();
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([&format!("key={}", "1"), "-xx", "-f", "--pretty=none"])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 5
|
||||
|
||||
key={c}
|
||||
"#, c = "1"});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn dont_compress_request_body_if_content_encoding_have_value() {
|
||||
let server = server();
|
||||
get_command()
|
||||
.arg(format!("{}/normal", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"content-encoding:gzip",
|
||||
"-xx",
|
||||
"-f",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1004
|
||||
|
||||
key={c}
|
||||
"#, c = "1".repeat(1000),});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_body_from_file() {
|
||||
let server = server::http(|req| async move {
|
||||
assert_eq!("Hello world\n", zlib_decode(req.body().await).unwrap());
|
||||
hyper::Response::default()
|
||||
});
|
||||
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let filename = dir.path().join("input.txt");
|
||||
OpenOptions::new()
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.write(true)
|
||||
.open(&filename)
|
||||
.unwrap()
|
||||
.write_all(b"Hello world\n")
|
||||
.unwrap();
|
||||
|
||||
get_command()
|
||||
.arg(server.base_url())
|
||||
.arg("-xx")
|
||||
.arg(format!("@{}", filename.to_string_lossy()))
|
||||
.assert()
|
||||
.success();
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_body_from_file_unless_compress_rate_less_1() {
|
||||
let server = server::http(|req| async move {
|
||||
assert_eq!("Hello world\n", req.body_as_string().await);
|
||||
hyper::Response::default()
|
||||
});
|
||||
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let filename = dir.path().join("input.txt");
|
||||
OpenOptions::new()
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.write(true)
|
||||
.open(&filename)
|
||||
.unwrap()
|
||||
.write_all(b"Hello world\n")
|
||||
.unwrap();
|
||||
|
||||
get_command()
|
||||
.arg(server.base_url())
|
||||
.arg("-x")
|
||||
.arg(format!("@{}", filename.to_string_lossy()))
|
||||
.assert()
|
||||
.success();
|
||||
}
|
||||
@ -1,2 +1,3 @@
|
||||
mod compress_request_body;
|
||||
mod download;
|
||||
mod logging;
|
||||
|
||||
194
tests/cli.rs
194
tests/cli.rs
@ -6,7 +6,7 @@ mod server;
|
||||
use std::collections::{HashMap, HashSet};
|
||||
use std::fs::{self, File, OpenOptions};
|
||||
use std::future::Future;
|
||||
use std::io::{Read as _, Write};
|
||||
use std::io::Write;
|
||||
use std::iter::FromIterator;
|
||||
use std::net::IpAddr;
|
||||
use std::pin::Pin;
|
||||
@ -3521,198 +3521,6 @@ fn zstd() {
|
||||
"#});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_body_from_file() {
|
||||
let server = server::http(|req| async move {
|
||||
assert_eq!("Hello world\n", zlib_decode(req.body().await).unwrap());
|
||||
hyper::Response::default()
|
||||
});
|
||||
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let filename = dir.path().join("input.txt");
|
||||
OpenOptions::new()
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.write(true)
|
||||
.open(&filename)
|
||||
.unwrap()
|
||||
.write_all(b"Hello world\n")
|
||||
.unwrap();
|
||||
|
||||
get_command()
|
||||
.arg(server.base_url())
|
||||
.arg("-xx")
|
||||
.arg(format!("@{}", filename.to_string_lossy()))
|
||||
.assert()
|
||||
.success();
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_body_from_file_unless_compress_rate_less_1() {
|
||||
let server = server::http(|req| async move {
|
||||
assert_eq!("Hello world\n", req.body_as_string().await);
|
||||
hyper::Response::default()
|
||||
});
|
||||
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let filename = dir.path().join("input.txt");
|
||||
OpenOptions::new()
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.write(true)
|
||||
.open(&filename)
|
||||
.unwrap()
|
||||
.write_all(b"Hello world\n")
|
||||
.unwrap();
|
||||
|
||||
get_command()
|
||||
.arg(server.base_url())
|
||||
.arg("-x")
|
||||
.arg(format!("@{}", filename.to_string_lossy()))
|
||||
.assert()
|
||||
.success();
|
||||
}
|
||||
|
||||
fn zlib_decode(bytes: Vec<u8>) -> std::io::Result<String> {
|
||||
let mut z = flate2::read::ZlibDecoder::new(&bytes[..]);
|
||||
let mut s = String::new();
|
||||
z.read_to_string(&mut s)?;
|
||||
Ok(s)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compress_request_body() {
|
||||
let server = server::http(|req| async move {
|
||||
match req.uri().path() {
|
||||
"/deflate" => {
|
||||
assert_eq!(
|
||||
req.headers().get(hyper::header::CONTENT_ENCODING),
|
||||
Some(HeaderValue::from_static("deflate")).as_ref()
|
||||
);
|
||||
|
||||
let compressed_body = req.body().await;
|
||||
let body = zlib_decode(compressed_body).unwrap();
|
||||
hyper::Response::builder()
|
||||
.header("date", "N/A")
|
||||
.header("Content-Type", "text/plain")
|
||||
.body(body.into())
|
||||
.unwrap()
|
||||
}
|
||||
"/normal" => {
|
||||
let body = req.body_as_string().await;
|
||||
hyper::Response::builder()
|
||||
.header("date", "N/A")
|
||||
.header("Content-Type", "text/plain")
|
||||
.body(body.into())
|
||||
.unwrap()
|
||||
}
|
||||
_ => panic!("unknown path"),
|
||||
}
|
||||
});
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"-x",
|
||||
"-j",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1010
|
||||
|
||||
{{"key":"{c}"}}
|
||||
"#, c = "1".repeat(1000),});
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"-x",
|
||||
"-x",
|
||||
"-f",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1004
|
||||
|
||||
key={c}
|
||||
"#, c = "1".repeat(1000),});
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"-x",
|
||||
"-x",
|
||||
"-f",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1004
|
||||
|
||||
key={c}
|
||||
"#, c = "1".repeat(1000),});
|
||||
|
||||
get_command()
|
||||
.arg(format!("{}/normal", server.base_url()))
|
||||
.args([&format!("key={}", "1"), "-x", "-f", "--pretty=none"])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 5
|
||||
|
||||
key={c}
|
||||
"#, c = "1"});
|
||||
|
||||
// force compress
|
||||
get_command()
|
||||
.arg(format!("{}/deflate", server.base_url()))
|
||||
.args([&format!("key={}", "1"), "-xx", "-f", "--pretty=none"])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 5
|
||||
|
||||
key={c}
|
||||
"#, c = "1"});
|
||||
// dont compress_request_body_if_content_encoding_have_value
|
||||
get_command()
|
||||
.arg(format!("{}/normal", server.base_url()))
|
||||
.args([
|
||||
&format!("key={}", "1".repeat(1000)),
|
||||
"content-encoding:gzip",
|
||||
"-x",
|
||||
"-f",
|
||||
"--pretty=none",
|
||||
])
|
||||
.assert()
|
||||
.stdout(indoc::formatdoc! {r#"
|
||||
HTTP/1.1 200 OK
|
||||
Date: N/A
|
||||
Content-Type: text/plain
|
||||
Content-Length: 1004
|
||||
|
||||
key={c}
|
||||
"#, c = "1".repeat(1000),});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn empty_response_with_content_encoding() {
|
||||
let server = server::http(|_req| async move {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user