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

Fix aiohttp tests to include content type when parsing response to json

Browse Source
This commit is contained in:
Luiz Menezes
2018-05-07 09:55:12 -03:00
parent 689d68a0a2
commit cd864b5eca
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration/aiohttp_utils.py
View File

@@ -1,8 +1,8 @@
import aiohttp import aiohttp
async def aiohttp_request(loop, method, url, output='text', encoding='utf-8', **kwargs): # NOQA: E999 async def aiohttp_request(loop, method, url, output='text', encoding='utf-8', headers=None, **kwargs): # NOQA: E999
async with aiohttp.ClientSession(loop=loop) as session: # NOQA: E999 async with aiohttp.ClientSession(loop=loop, headers=headers or {}) as session: # NOQA: E999
async with session.request(method, url, **kwargs) as response: # NOQA: E999 async with session.request(method, url, **kwargs) as response: # NOQA: E999
if output == 'text': if output == 'text':
content = await response.text() # NOQA: E999 content = await response.text() # NOQA: E999

22
tests/integration/test_aiohttp.py
View File

@@ -71,11 +71,13 @@ def test_text(tmpdir, scheme):
def test_json(tmpdir, scheme): def test_json(tmpdir, scheme):
url = scheme + '://httpbin.org/get' url = scheme + '://httpbin.org/get'
headers = {'Content-Type': 'application/json'}
with vcr.use_cassette(str(tmpdir.join('json.yaml'))): with vcr.use_cassette(str(tmpdir.join('json.yaml'))):
_, response_json = get(url, output='json') _, response_json = get(url, output='json', headers=headers)
with vcr.use_cassette(str(tmpdir.join('json.yaml'))) as cassette: with vcr.use_cassette(str(tmpdir.join('json.yaml'))) as cassette:
_, cassette_response_json = get(url, output='json') _, cassette_response_json = get(url, output='json', headers=headers)
assert cassette_response_json == response_json assert cassette_response_json == response_json
assert cassette.play_count == 1 assert cassette.play_count == 1
@@ -105,24 +107,28 @@ def test_post(tmpdir, scheme):
def test_params(tmpdir, scheme): def test_params(tmpdir, scheme):
url = scheme + '://httpbin.org/get' url = scheme + '://httpbin.org/get'
headers = {'Content-Type': 'application/json'}
params = {'a': 1, 'b': False, 'c': 'c'} params = {'a': 1, 'b': False, 'c': 'c'}
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, response_json = get(url, output='json', params=params)
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette: with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, cassette_response_json = get(url, output='json', params=params) _, response_json = get(url, output='json', params=params, headers=headers)
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, cassette_response_json = get(url, output='json', params=params, headers=headers)
assert cassette_response_json == response_json assert cassette_response_json == response_json
assert cassette.play_count == 1 assert cassette.play_count == 1
def test_params_same_url_distinct_params(tmpdir, scheme): def test_params_same_url_distinct_params(tmpdir, scheme):
url = scheme + '://httpbin.org/get' url = scheme + '://httpbin.org/get'
headers = {'Content-Type': 'application/json'}
params = {'a': 1, 'b': False, 'c': 'c'} params = {'a': 1, 'b': False, 'c': 'c'}
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, response_json = get(url, output='json', params=params)
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette: with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, cassette_response_json = get(url, output='json', params=params) _, response_json = get(url, output='json', params=params, headers=headers)
with vcr.use_cassette(str(tmpdir.join('get.yaml'))) as cassette:
_, cassette_response_json = get(url, output='json', params=params, headers=headers)
assert cassette_response_json == response_json assert cassette_response_json == response_json
assert cassette.play_count == 1 assert cassette.play_count == 1