support set cookie.

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

gofight.go

@@ -99,6 +99,7 @@ type RequestConfig struct {
 	Path    string
 	Body    string
 	Headers H
+	Cookies H
 	Debug   bool
 }
 
@@ -245,6 +246,15 @@ func (rc *RequestConfig) SetBody(body string) *RequestConfig {
 	return rc
 }
 
+// SetCookie supply cookies what you defined.
+func (rc *RequestConfig) SetCookie(cookies H) *RequestConfig {
+	if len(cookies) > 0 {
+		rc.Cookies = cookies
+	}
+
+	return rc
+}
+
 func (rc *RequestConfig) initTest() (*http.Request, *httptest.ResponseRecorder) {
 	qs := ""
 	if strings.Contains(rc.Path, "?") {
@@ -278,11 +288,18 @@ func (rc *RequestConfig) initTest() (*http.Request, *httptest.ResponseRecorder)
 		}
 	}
 
+	if len(rc.Cookies) > 0 {
+		for k, v := range rc.Cookies {
+			req.AddCookie(&http.Cookie{Name: k, Value: v})
+		}
+	}
+
 	if rc.Debug {
 		log.Printf("Request Method: %s", rc.Method)
 		log.Printf("Request Path: %s", rc.Path)
 		log.Printf("Request Body: %s", rc.Body)
 		log.Printf("Request Headers: %s", rc.Headers)
+		log.Printf("Request Cookies: %s", rc.Cookies)
 		log.Printf("Request Header: %s", req.Header)
 	}
 

gofight_test.go

@@ -76,6 +76,21 @@ func TestGinHeader(t *testing.T) {
 		})
 }
 
+func TestGinCookie(t *testing.T) {
+	r := New()
+
+	r.GET("/text").
+		SetCookie(H{
+			"foo":   "bar",
+			"hello": "world",
+		}).
+		Run(framework.GinEngine(), func(r HTTPResponse, rq HTTPRequest) {
+
+			assert.Equal(t, http.StatusOK, r.Code)
+			assert.Equal(t, "foo=bar; hello=world", rq.Header.Get("cookie"))
+		})
+}
+
 func TestGinQuery(t *testing.T) {
 	r := New()