def test_redirects(endpoint, n_redirects, mock_session):
"""Test all types of redirect endpoints with different numbers of consecutive redirects"""
mock_session.get(httpbin(f'redirect/{n_redirects}'))
r2 = mock_session.get(httpbin('get'))
assert r2.from_cache is True
assert len(mock_session.cache.redirects) == n_redirects
def test_all_response_formats(response_format, tempfile_session):
"""Test that all relevant response formats are cached correctly"""
r1 = tempfile_session.get(httpbin(response_format))
r2 = tempfile_session.get(httpbin(response_format))
assert r1.from_cache is False
assert r2.from_cache is True
assert r1.content == r2.content
def test_response_decode_stream(tempfile_session):
"""Test that streamed gzip-compressed responses can be uncompressed with decode_content"""
response_uncached = tempfile_session.get(httpbin('gzip'), stream=True)
response_cached = tempfile_session.get(httpbin('gzip'), stream=True)
for res in (response_uncached, response_cached):
assert b'gzipped' in res.content
assert b'gzipped' in res.raw.read(None, decode_content=True)
def test_all_response_formats(response_format, tempfile_session):
"""Test that all relevant response formats are cached correctly"""
# Temporary workaround for this issue: https://github.com/kevin1024/pytest-httpbin/issues/60
if response_format == 'json' and USE_PYTEST_HTTPBIN:
tempfile_session.allowable_codes = (200, 404)
r1 = tempfile_session.get(httpbin(response_format))
r2 = tempfile_session.get(httpbin(response_format))
assert r1.from_cache is False
assert r2.from_cache is True
assert r1.content == r2.content
def test_response_decode(tempfile_session):
"""Test that a gzip-compressed raw response can be manually uncompressed with decode_content"""
response = tempfile_session.get(httpbin('gzip'))
assert b'gzipped' in response.content
cached = CachedResponse(response)
assert b'gzipped' in cached.content
assert b'gzipped' in cached.raw.read(None, decode_content=True)
def test_all_methods(field, method, tempfile_session):
"""Test all relevant combinations of methods and data fields. Requests with different request
params, data, or json should be cached under different keys.
"""
url = httpbin(method.lower())
for params in [{'param_1': 1}, {'param_1': 2}, {'param_2': 2}]:
assert tempfile_session.request(method, url, **{field: params}).from_cache is False
assert tempfile_session.request(method, url, **{field: params}).from_cache is True
def test_cookies(tempfile_session):
def get_json(url):
return json.loads(tempfile_session.get(url).text)
response_1 = get_json(httpbin('cookies/set/test1/test2'))
with tempfile_session.cache_disabled():
assert get_json(httpbin('cookies')) == response_1
# From cache
response_2 = get_json(httpbin('cookies'))
assert response_2 == get_json(httpbin('cookies'))
# Not from cache
with tempfile_session.cache_disabled():
response_3 = get_json(httpbin('cookies/set/test3/test4'))
assert response_3 == get_json(httpbin('cookies'))