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 tests

Browse Source
This commit is contained in:
Paulo Romeira
2020-06-04 16:12:48 -03:00
committed by Kevin McCarthy
parent cd0907ffaf
commit 837992767e
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration/test_aiohttp.py
View File

@@ -1,4 +1,3 @@
import asyncio
import contextlib import contextlib
import logging import logging
import urllib.parse import urllib.parse
@@ -324,7 +323,7 @@ def test_double_requests(tmpdir):
def test_cookies(scheme, tmpdir): def test_cookies(scheme, tmpdir):
async def run(scheme, tmpdir): async def run(loop):
cookies_url = scheme + ( cookies_url = scheme + (
"://httpbin.org/response-headers?" "://httpbin.org/response-headers?"
"set-cookie=" + urllib.parse.quote("cookie_1=val_1; Path=/") + "&" "set-cookie=" + urllib.parse.quote("cookie_1=val_1; Path=/") + "&"
@@ -337,16 +336,18 @@ def test_cookies(scheme, tmpdir):
# ------------------------- Record -------------------------- # # ------------------------- Record -------------------------- #
with vcr.use_cassette(tmp) as cassette: with vcr.use_cassette(tmp) as cassette:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession(loop=loop) as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers) home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers)
assert cassette.play_count == 0
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(loop=loop) as session:
cookies_resp = await session.get(cookies_url) cookies_resp = await session.get(cookies_url)
home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers) home_resp = await session.get(home_url, cookies=req_cookies, headers=req_headers)
assert cassette.play_count == 2
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):
@@ -358,4 +359,4 @@ def test_cookies(scheme, tmpdir):
assert "Cookie_3=Val_3" in request_cookies assert "Cookie_3=Val_3" in request_cookies
assert "Cookie_4=Val_4" in request_cookies assert "Cookie_4=Val_4" in request_cookies
asyncio.run(run(scheme, tmpdir)) run_in_loop(run)