From d0be1f2ea97342ef577ebe879971776422b08038 Mon Sep 17 00:00:00 2001 From: Damian Date: Tue, 17 Dec 2019 21:00:04 +0100 Subject: Added option to disable archives --- tests/archive.rs | 46 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 tests/archive.rs (limited to 'tests/archive.rs') diff --git a/tests/archive.rs b/tests/archive.rs new file mode 100644 index 0000000..c692d90 --- /dev/null +++ b/tests/archive.rs @@ -0,0 +1,46 @@ +mod fixtures; + +use assert_cmd::prelude::*; +use assert_fs::fixture::TempDir; +use fixtures::{port, tmpdir, Error}; +use rstest::rstest; +use select::document::Document; +use select::predicate::Text; +use std::process::{Command, Stdio}; +use std::thread::sleep; +use std::time::Duration; +use reqwest::StatusCode; + +#[rstest] +fn archives_are_disabled(tmpdir: TempDir, port: u16) -> Result<(), Error> { + let mut child = Command::cargo_bin("miniserve")? + .arg(tmpdir.path()) + .arg("-p") + .arg(port.to_string()) + .arg("-r") + .stdout(Stdio::null()) + .spawn()?; + + sleep(Duration::from_secs(1)); + + // Ensure the links to the archives are not present + let body = reqwest::get(format!("http://localhost:{}", port).as_str())?.error_for_status()?; + let parsed = Document::from_read(body)?; + assert!(parsed.find(Text).all(|x| x.text() != "Download .tar.gz" && x.text() != "Download .tar")); + + // Try to download anyway, ensure it's forbidden + assert_eq!( + reqwest::get( + format!("http://localhost:{}/?download=tar_gz", port).as_str())? + .status(), + StatusCode::FORBIDDEN); + assert_eq!( + reqwest::get( + format!("http://localhost:{}/?download=tar", port).as_str())? + .status(), + StatusCode::FORBIDDEN); + + child.kill()?; + + Ok(()) +} \ No newline at end of file -- cgit v1.2.3