Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
Code Releases Activity

aiohttp: Fix code linting

Browse Source
This commit is contained in:
Paulo Romeira
2020-06-04 15:12:42 -03:00
committed by Kevin McCarthy
parent 5362db2ebb
commit 77d838e0fc
2 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/integration/test_aiohttp.py
View File

@@ -339,18 +339,14 @@ def test_cookies(scheme, tmpdir):
with vcr.use_cassette(tmp) as cassette: with vcr.use_cassette(tmp) as cassette:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
home_resp = await session.get( home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers)
home_url, cookies=req_cookies, headers=req_headers
)
assert_responses(cookies_resp, home_resp) assert_responses(cookies_resp, home_resp)
# -------------------------- Play --------------------------- # # -------------------------- Play --------------------------- #
with vcr.use_cassette(tmp, record_mode="none") as cassette: with vcr.use_cassette(tmp, record_mode="none") as cassette:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
home_resp = await session.get( home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers)
home_url, cookies=req_cookies, headers=req_headers
)
assert_responses(cookies_resp, home_resp) assert_responses(cookies_resp, home_resp)
def assert_responses(cookies_resp, home_resp): def assert_responses(cookies_resp, home_resp):

4
vcr/stubs/aiohttp_stubs/__init__.py
View File

@@ -238,9 +238,7 @@ def vcr_request(cassette, real_request):
request_url = URL(url).with_query(params) request_url = URL(url).with_query(params)
c_header = headers.pop(hdrs.COOKIE, None) c_header = headers.pop(hdrs.COOKIE, None)
cookie_header = _build_cookie_header( cookie_header = _build_cookie_header(self, cookies, c_header, request_url)
self, cookies, c_header, request_url
)
if cookie_header: if cookie_header:
headers[hdrs.COOKIE] = cookie_header headers[hdrs.COOKIE] = cookie_header