From 592fa2c0f17b32a1534e46fda2d851667c2ac4d7 Mon Sep 17 00:00:00 2001
From: "yihua.huang" <code4crafter@gmail.com>
Date: Sat, 3 Jun 2017 11:56:18 +0800
Subject: [PATCH] add site header test

---
 .../downloader/HttpClientDownloaderTest.java     | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/webmagic-core/src/test/java/us/codecraft/webmagic/downloader/HttpClientDownloaderTest.java b/webmagic-core/src/test/java/us/codecraft/webmagic/downloader/HttpClientDownloaderTest.java
index 8e36c05..3f9c838 100644
--- a/webmagic-core/src/test/java/us/codecraft/webmagic/downloader/HttpClientDownloaderTest.java
+++ b/webmagic-core/src/test/java/us/codecraft/webmagic/downloader/HttpClientDownloaderTest.java
@@ -206,6 +206,22 @@ public class HttpClientDownloaderTest {
         });
     }
 
+    @Test
+    public void test_set_site_header() throws Exception {
+        HttpServer server = httpServer(13423);
+        server.get(eq(header("header"), "header-webmagic")).response("ok");
+        Runner.running(server, new Runnable() {
+            @Override
+            public void run() throws Exception {
+                HttpClientDownloader httpClientDownloader = new HttpClientDownloader();
+                Request request = new Request();
+                request.setUrl("http://127.0.0.1:13423");
+                Page page = httpClientDownloader.download(request, Site.me().addHeader("header","header-webmagic").toTask());
+                assertThat(page.getRawText()).isEqualTo("ok");
+            }
+        });
+    }
+
     @Test
     public void test_set_site_cookie() throws Exception {
         HttpServer server = httpServer(13423);