From da97e91f2eb02ace7f3e6cb3522d4bd55aa45caf Mon Sep 17 00:00:00 2001 From: Dean Li Date: Mon, 22 Feb 2021 13:13:53 +0800 Subject: Add test for custom header --- tests/header.rs | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 tests/header.rs (limited to 'tests/header.rs') diff --git a/tests/header.rs b/tests/header.rs new file mode 100644 index 0000000..a0734e5 --- /dev/null +++ b/tests/header.rs @@ -0,0 +1,33 @@ +mod fixtures; + +use assert_cmd::prelude::*; +use assert_fs::fixture::TempDir; +use fixtures::{port, tmpdir, Error}; +use rstest::rstest; +use std::process::{Command, Stdio}; +use std::thread::sleep; +use std::time::Duration; + +#[rstest] +fn custom_header_set(tmpdir: TempDir, port: u16) -> Result<(), Error> { + let header_name = "x-info"; + let header_value = "123"; + let header_str = format!("{}: {}", header_name, header_value); + + let _ = Command::cargo_bin("miniserve")? + .arg(tmpdir.path()) + .arg("-p") + .arg(port.to_string()) + .arg("--header") + .arg(header_str) + .stdout(Stdio::null()) + .spawn()?; + + sleep(Duration::from_secs(1)); + + let resp = reqwest::blocking::get(format!("http://localhost:{}", port).as_str())?; + + assert_eq!(resp.headers().get(header_name).unwrap(), header_value); + + Ok(()) +} -- cgit v1.2.3