GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-10 02:09:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: log request cookies

Browse Source
This commit is contained in:
debugtalk
2020-05-22 13:28:56 +08:00
parent 988e7e6e78
commit 41a188df3f
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
httprunner/client.py
View File

@@ -41,6 +41,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
# record actual request info # record actual request info
request_headers = dict(resp_obj.request.headers) request_headers = dict(resp_obj.request.headers)
request_cookies = dict(resp_obj.request._cookies)
request_body = resp_obj.request.body request_body = resp_obj.request.body
try: try:
request_body = json.loads(request_body) request_body = json.loads(request_body)
@@ -57,6 +58,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
method=resp_obj.request.method, method=resp_obj.request.method,
url=resp_obj.request.url, url=resp_obj.request.url,
headers=request_headers, headers=request_headers,
cookies=request_cookies,
body=request_body, body=request_body,
) )

7
httprunner/schema.py
View File

@@ -13,6 +13,7 @@ BaseUrl = Union[HttpUrl, Text]
VariablesMapping = Dict[Text, Any] VariablesMapping = Dict[Text, Any]
FunctionsMapping = Dict[Text, Callable] FunctionsMapping = Dict[Text, Callable]
Headers = Dict[Text, Text] Headers = Dict[Text, Text]
Cookies = Dict[Text, Text]
Verify = bool Verify = bool
Hook = List[Text] Hook = List[Text]
Export = List[Text] Export = List[Text]
@@ -53,7 +54,7 @@ class Request(BaseModel):
headers: Headers = {} headers: Headers = {}
req_json: Dict = Field({}, alias="json") req_json: Dict = Field({}, alias="json")
data: Union[Text, Dict[Text, Any]] = "" data: Union[Text, Dict[Text, Any]] = ""
cookies: Dict[Text, Text] = {} cookies: Cookies = {}
timeout: int = 120 timeout: int = 120
allow_redirects: bool = True allow_redirects: bool = True
verify: Verify = False verify: Verify = False
@@ -108,13 +109,13 @@ class RequestData(BaseModel):
method: MethodEnum = MethodEnum.GET method: MethodEnum = MethodEnum.GET
url: Url url: Url
headers: Headers = {} headers: Headers = {}
# TODO: add cookies cookies: Cookies = {}
body: Union[Text, bytes, Dict, None] = {} body: Union[Text, bytes, Dict, None] = {}
class ResponseData(BaseModel): class ResponseData(BaseModel):
status_code: int status_code: int
cookies: Dict cookies: Cookies
encoding: Union[Text, None] = None encoding: Union[Text, None] = None
headers: Dict headers: Dict
content_type: Text content_type: Text