test: add tests for setting default headers

tests/client.rs

@@ -79,9 +79,9 @@ fn test_post() {
         request: b"\
             POST /2 HTTP/1.1\r\n\
             Host: $HOST\r\n\
-            Content-Length: 5\r\n\
             User-Agent: $USERAGENT\r\n\
             Accept: */*\r\n\
+            Content-Length: 5\r\n\
             Accept-Encoding: gzip\r\n\
             \r\n\
             Hello\
@@ -170,3 +170,111 @@ fn test_error_for_status_5xx() {
     assert!(err.is_server_error());
     assert_eq!(err.status(), Some(reqwest::StatusCode::InternalServerError));
 }
+
+#[test]
+fn test_default_headers() {
+    use reqwest::header;
+    let mut headers = header::Headers::with_capacity(1);
+    let mut cookies = header::Cookie::new();
+    cookies.set("a", "b");
+    cookies.set("c", "d");
+    headers.set(cookies);
+    let client = reqwest::Client::builder()
+        .default_headers(headers)
+        .build().unwrap();
+
+    let server = server! {
+        request: b"\
+            GET /1 HTTP/1.1\r\n\
+            Host: $HOST\r\n\
+            User-Agent: $USERAGENT\r\n\
+            Accept: */*\r\n\
+            Cookie: a=b; c=d\r\n\
+            Accept-Encoding: gzip\r\n\
+            \r\n\
+            ",
+        response: b"\
+            HTTP/1.1 200 OK\r\n\
+            Server: test\r\n\
+            Content-Length: 0\r\n\
+            \r\n\
+            "
+    };
+
+    let url = format!("http://{}/1", server.addr());
+    let res = client.get(&url).send().unwrap();
+
+    assert_eq!(res.url().as_str(), &url);
+    assert_eq!(res.status(), reqwest::StatusCode::Ok);
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::Server::new("test")));
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::ContentLength(0)));
+
+    let server = server! {
+        request: b"\
+            GET /2 HTTP/1.1\r\n\
+            Host: $HOST\r\n\
+            User-Agent: $USERAGENT\r\n\
+            Accept: */*\r\n\
+            Cookie: a=b; c=d\r\n\
+            Accept-Encoding: gzip\r\n\
+            \r\n\
+            ",
+        response: b"\
+            HTTP/1.1 200 OK\r\n\
+            Server: test\r\n\
+            Content-Length: 0\r\n\
+            \r\n\
+            "
+    };
+
+    let url = format!("http://{}/2", server.addr());
+    let res = client.get(&url).send().unwrap();
+
+    assert_eq!(res.url().as_str(), &url);
+    assert_eq!(res.status(), reqwest::StatusCode::Ok);
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::Server::new("test")));
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::ContentLength(0)));
+}
+
+#[test]
+fn test_override_default_headers() {
+    use reqwest::header;
+    let mut headers = header::Headers::with_capacity(1);
+    headers.set(header::Authorization("iamatoken".to_string()));
+    let client = reqwest::Client::builder()
+        .default_headers(headers)
+        .build().unwrap();
+
+    let server = server! {
+        request: b"\
+            GET /3 HTTP/1.1\r\n\
+            Host: $HOST\r\n\
+            User-Agent: $USERAGENT\r\n\
+            Accept: */*\r\n\
+            Authorization: secret\r\n\
+            Accept-Encoding: gzip\r\n\
+            \r\n\
+            ",
+        response: b"\
+            HTTP/1.1 200 OK\r\n\
+            Server: test\r\n\
+            Content-Length: 0\r\n\
+            \r\n\
+            "
+    };
+
+    let url = format!("http://{}/3", server.addr());
+    let res = client.get(&url).header(header::Authorization("secret".to_string())).send().unwrap();
+
+    assert_eq!(res.url().as_str(), &url);
+    assert_eq!(res.status(), reqwest::StatusCode::Ok);
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::Server::new("test")));
+    assert_eq!(res.headers().get(),
+               Some(&reqwest::header::ContentLength(0)));
+
+}