diff --git a/src/test/agent.rs b/src/test/agent.rs new file mode 100644 index 0000000..51be78a --- /dev/null +++ b/src/test/agent.rs @@ -0,0 +1,28 @@ +use test; + +use super::super::*; + +#[test] +fn agent_reuse_headers() { + let agent = agent() + .set("Authorization", "Foo 12345") + .build(); + + test::set_handler("/agent_reuse_headers", |req, _url| { + assert!(req.has("Authorization")); + assert_eq!(req.get("Authorization").unwrap(), "Foo 12345"); + test::make_stream(200, "OK", vec!["X-Call: 1"], vec![]) + }); + + let resp = agent.get("test://host/agent_reuse_headers").call(); + assert_eq!(resp.get("X-Call").unwrap(), "1"); + + test::set_handler("/agent_reuse_headers", |req, _url| { + assert!(req.has("Authorization")); + assert_eq!(req.get("Authorization").unwrap(), "Foo 12345"); + test::make_stream(200, "OK", vec!["X-Call: 2"], vec![]) + }); + + let resp = agent.get("test://host/agent_reuse_headers").call(); + assert_eq!(resp.get("X-Call").unwrap(), "2"); +} diff --git a/src/test/mod.rs b/src/test/mod.rs index 40142a9..dae0f56 100644 --- a/src/test/mod.rs +++ b/src/test/mod.rs @@ -8,6 +8,7 @@ use stream::Stream; use url::Url; use util::vecread::VecRead; +mod agent; mod auth; mod simple; mod body_read;