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 linting

Browse Source
This commit is contained in:
Paulo Romeira
2020-06-18 07:13:41 -03:00
committed by Kevin McCarthy
parent 15d79e5b78
commit 4897a8e692
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/integration/test_aiohttp.py
View File

@@ -364,7 +364,7 @@ def test_cookies(scheme, tmpdir):
def test_cookies_redirect(scheme, tmpdir): def test_cookies_redirect(scheme, tmpdir):
async def run(loop): async def run(loop):
# Sets cookie as provided by the query string and redirects # Sets cookie as provided by the query string and redirects
cookies_url = scheme + '://httpbin.org/cookies/set?Cookie_1=Val_1' cookies_url = scheme + "://httpbin.org/cookies/set?Cookie_1=Val_1"
tmp = str(tmpdir.join("cookies.yaml")) tmp = str(tmpdir.join("cookies.yaml"))
# ------------------------- Record -------------------------- # # ------------------------- Record -------------------------- #
@@ -373,29 +373,29 @@ def test_cookies_redirect(scheme, tmpdir):
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
assert not cookies_resp.cookies assert not cookies_resp.cookies
cookies = session.cookie_jar.filter_cookies(cookies_url) cookies = session.cookie_jar.filter_cookies(cookies_url)
assert cookies['Cookie_1'].value == 'Val_1' assert cookies["Cookie_1"].value == "Val_1"
assert cassette.play_count == 0 assert cassette.play_count == 0
cassette.requests[1].headers['Cookie'] == 'Cookie_1=Val_1' cassette.requests[1].headers["Cookie"] == "Cookie_1=Val_1"
# -------------------------- 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(loop=loop) as session: async with aiohttp.ClientSession(loop=loop) as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
assert not cookies_resp.cookies assert not cookies_resp.cookies
cookies = session.cookie_jar.filter_cookies(cookies_url) cookies = session.cookie_jar.filter_cookies(cookies_url)
assert cookies['Cookie_1'].value == 'Val_1' assert cookies["Cookie_1"].value == "Val_1"
assert cassette.play_count == 2 assert cassette.play_count == 2
cassette.requests[1].headers['Cookie'] == 'Cookie_1=Val_1' cassette.requests[1].headers["Cookie"] == "Cookie_1=Val_1"
# Assert that it's ignoring expiration date # Assert that it's ignoring expiration date
with vcr.use_cassette(tmp, record_mode='none') as cassette: with vcr.use_cassette(tmp, record_mode="none") as cassette:
cassette.responses[0]['headers']['set-cookie'] = [ cassette.responses[0]["headers"]["set-cookie"] = [
'Cookie_1=Val_1; Expires=Wed, 21 Oct 2015 07:28:00 GMT' "Cookie_1=Val_1; Expires=Wed, 21 Oct 2015 07:28:00 GMT"
] ]
async with aiohttp.ClientSession(loop=loop) as session: async with aiohttp.ClientSession(loop=loop) as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
assert not cookies_resp.cookies assert not cookies_resp.cookies
cookies = session.cookie_jar.filter_cookies(cookies_url) cookies = session.cookie_jar.filter_cookies(cookies_url)
assert cookies['Cookie_1'].value == 'Val_1' assert cookies["Cookie_1"].value == "Val_1"
run_in_loop(run) run_in_loop(run)

8
vcr/stubs/aiohttp_stubs.py
View File

@@ -77,11 +77,11 @@ def build_response(vcr_request, vcr_response, history):
try: try:
cookies = SimpleCookie(hdr) cookies = SimpleCookie(hdr)
for cookie_name, cookie in cookies.items(): for cookie_name, cookie in cookies.items():
expires = cookie.get('expires', '').strip() expires = cookie.get("expires", "").strip()
if expires: if expires:
log.debug("Ignoring expiration date: %s=\"%s\"", cookie_name, expires) log.debug('Ignoring expiration date: %s="%s"', cookie_name, expires)
cookie['expires'] = '' cookie["expires"] = ""
response.cookies.load(cookie.output(header='').strip()) response.cookies.load(cookie.output(header="").strip())
except CookieError as exc: except CookieError as exc:
log.warning("Can not load response cookies: %s", exc) log.warning("Can not load response cookies: %s", exc)