1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
from unittest.mock import patch
import requests
from requests.sessions import Session as OriginalSession
import requests_cache
from requests_cache import CachedSession
from requests_cache.backends import BaseCache, SQLiteCache
from tests.conftest import CACHE_NAME, ignore_deprecation
def test_install_uninstall():
for _ in range(2):
requests_cache.install_cache(name=CACHE_NAME, use_temp=True)
assert isinstance(requests.Session(), CachedSession)
assert isinstance(requests.sessions.Session(), CachedSession)
requests_cache.uninstall_cache()
assert not isinstance(requests.Session(), CachedSession)
assert not isinstance(requests.sessions.Session(), CachedSession)
def test_session_is_a_class_with_original_attributes(installed_session):
assert isinstance(requests.Session, type)
for attribute in dir(OriginalSession):
assert hasattr(requests.Session, attribute)
assert isinstance(requests.Session(), CachedSession)
def test_inheritance_after_monkey_patch(installed_session):
class FooSession(requests.Session):
__attrs__ = requests.Session.__attrs__ + ["new_one"]
def __init__(self, param):
self.param = param
super(FooSession, self).__init__()
s = FooSession(1)
assert isinstance(s, CachedSession)
assert s.param == 1
assert "new_one" in s.__attrs__
@patch.object(SQLiteCache, 'clear')
def test_clear(mock_clear, installed_session):
requests_cache.clear()
mock_clear.assert_called()
@patch.object(SQLiteCache, 'clear')
def test_clear__not_installed(mock_clear):
"""If clear is called without a cache installed, it should just fail silently"""
requests_cache.clear()
mock_clear.assert_not_called()
@patch.object(OriginalSession, 'request')
@patch.object(CachedSession, 'request')
def test_disabled(cached_request, original_request, installed_session):
with requests_cache.disabled():
for i in range(3):
requests.get('some_url')
assert cached_request.call_count == 0
assert original_request.call_count == 3
@patch.object(OriginalSession, 'request')
@patch.object(CachedSession, 'request')
def test_enabled(cached_request, original_request, tempfile_path):
with requests_cache.enabled(tempfile_path):
for i in range(3):
requests.get('some_url')
assert cached_request.call_count == 3
assert original_request.call_count == 0
def test_is_installed():
assert requests_cache.is_installed() is False
requests_cache.install_cache(name=CACHE_NAME, use_temp=True)
assert requests_cache.is_installed() is True
requests_cache.uninstall_cache()
assert requests_cache.is_installed() is False
@patch.object(BaseCache, 'delete')
def test_delete__expired_responses(mock_delete):
requests_cache.install_cache(backend='memory', expire_after=360)
requests_cache.delete(expired=True)
assert mock_delete.called is True
requests_cache.uninstall_cache()
@patch.object(BaseCache, 'delete')
def test_delete__cache_not_installed(mock_delete):
requests_cache.delete(expired=True)
assert mock_delete.called is False
@patch.object(BaseCache, 'delete')
def test_remove_expired_responses(mock_delete):
requests_cache.install_cache(backend='memory', expire_after=360)
with ignore_deprecation():
requests_cache.remove_expired_responses()
assert mock_delete.called is True
requests_cache.uninstall_cache()
|
