Add SetHeaders Method

client_test.go

@@ -120,19 +120,6 @@ func TestRequest(t *testing.T) {
 	}
 }
 
-func TestRequest_Header(t *testing.T) {
-	{
-		var req = Post("http://%s", nextAddr()).SetEncoder(FormEncoder)
-		var typ = req.Header().Get("Content-Type")
-		assert.Equal(t, typ, MimeForm)
-	}
-	{
-		var req = Get("http://%s", nextAddr())
-		var typ = req.Header().Get("Content-Type")
-		assert.Equal(t, typ, MimeJson)
-	}
-}
-
 func TestRequest_SetContext(t *testing.T) {
 	addr := nextAddr()
 	srv := &http.Server{Addr: addr}
@@ -385,3 +372,15 @@ func TestResponse(t *testing.T) {
 		assert.Error(t, resp.Err())
 	})
 }
+
+func TestRequest_SetHeaders(t *testing.T) {
+	var h = http.Header{}
+	h.Set("Content-Type", MimeJson)
+	h.Set("cookie", "123")
+	var req = Get("https://api.github.com").
+		SetHeader("Cookie", "456").
+		SetHeader("encoding", "none").
+		SetHeaders(h)
+	assert.Equal(t, req.headers.Get("cookie"), "123")
+	assert.Equal(t, req.headers.Get("encoding"), "none")
+}

request.go

@@ -99,10 +99,13 @@ func (c *Request) SetHeader(k, v string) *Request {
 	return c
 }
 
-// Header 获取请求头
-// Get request header
-func (c *Request) Header() http.Header {
-	return c.headers
+// SetHeaders 批量设置请求头
+// Set Request Header
+func (c *Request) SetHeaders(headers http.Header) *Request {
+	for k, v := range headers {
+		c.headers[k] = v
+	}
+	return c
 }
 
 // SetContext 设置请求上下文