From 2cb3d4f0668bc804495ec9a773d38ff5d07f13d0 Mon Sep 17 00:00:00 2001
From: Martin Algesten <martin@algesten.se>
Date: Mon, 11 Jun 2018 23:00:45 +0200
Subject: [PATCH] reuse agent

---
 src/test/agent.rs | 28 ++++++++++++++++++++++++++++
 src/test/mod.rs   |  1 +
 2 files changed, 29 insertions(+)
 create mode 100644 src/test/agent.rs

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;