From 41a188df3f99fa7850c249b3b85eec5ceb9cbedf Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Fri, 22 May 2020 13:28:56 +0800
Subject: [PATCH] feat: log request cookies

---
 httprunner/client.py | 2 ++
 httprunner/schema.py | 7 ++++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/httprunner/client.py b/httprunner/client.py
index a72e770f..f5474ce3 100644
--- a/httprunner/client.py
+++ b/httprunner/client.py
@@ -41,6 +41,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
 
     # record actual request info
     request_headers = dict(resp_obj.request.headers)
+    request_cookies = dict(resp_obj.request._cookies)
     request_body = resp_obj.request.body
     try:
         request_body = json.loads(request_body)
@@ -57,6 +58,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
         method=resp_obj.request.method,
         url=resp_obj.request.url,
         headers=request_headers,
+        cookies=request_cookies,
         body=request_body,
     )
 
diff --git a/httprunner/schema.py b/httprunner/schema.py
index cf0d5dc3..86c1a13c 100644
--- a/httprunner/schema.py
+++ b/httprunner/schema.py
@@ -13,6 +13,7 @@ BaseUrl = Union[HttpUrl, Text]
 VariablesMapping = Dict[Text, Any]
 FunctionsMapping = Dict[Text, Callable]
 Headers = Dict[Text, Text]
+Cookies = Dict[Text, Text]
 Verify = bool
 Hook = List[Text]
 Export = List[Text]
@@ -53,7 +54,7 @@ class Request(BaseModel):
     headers: Headers = {}
     req_json: Dict = Field({}, alias="json")
     data: Union[Text, Dict[Text, Any]] = ""
-    cookies: Dict[Text, Text] = {}
+    cookies: Cookies = {}
     timeout: int = 120
     allow_redirects: bool = True
     verify: Verify = False
@@ -108,13 +109,13 @@ class RequestData(BaseModel):
     method: MethodEnum = MethodEnum.GET
     url: Url
     headers: Headers = {}
-    # TODO: add cookies
+    cookies: Cookies = {}
     body: Union[Text, bytes, Dict, None] = {}
 
 
 class ResponseData(BaseModel):
     status_code: int
-    cookies: Dict
+    cookies: Cookies
     encoding: Union[Text, None] = None
     headers: Dict
     content_type: Text