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
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Path hack
import os, sys
try:
import unittest2 as unittest
except ImportError:
import unittest
import requests
from requests.sessions import Session as OriginalSession
import requests_cache
from requests_cache import CachedSession
from requests_cache.backends import BaseCache
CACHE_NAME = 'requests_cache_test'
CACHE_BACKEND = 'sqlite'
FAST_SAVE = False
class MonkeyPatchTestCase(unittest.TestCase):
def setUp(self):
requests_cache.install_cache(name=CACHE_NAME, backend=CACHE_BACKEND)
requests.Session().cache.clear()
requests_cache.uninstall_cache()
def test_install_uninstall(self):
for _ in range(2):
requests_cache.install_cache(name=CACHE_NAME, backend=CACHE_BACKEND)
self.assertTrue(isinstance(requests.Session(), CachedSession))
self.assertTrue(isinstance(requests.sessions.Session(), CachedSession))
self.assertTrue(isinstance(requests.session(), CachedSession))
requests_cache.uninstall_cache()
self.assertFalse(isinstance(requests.Session(), CachedSession))
self.assertFalse(isinstance(requests.sessions.Session(), CachedSession))
self.assertFalse(isinstance(requests.session(), CachedSession))
def test_requests_from_cache(self):
requests_cache.install_cache(name=CACHE_NAME, backend=CACHE_BACKEND)
r = requests.get('http://httpbin.org/get')
self.assertFalse(r.from_cache)
r = requests.get('http://httpbin.org/get')
self.assertTrue(r.from_cache)
def test_session_is_a_class_with_original_attributes(self):
requests_cache.install_cache(name=CACHE_NAME, backend=CACHE_BACKEND)
self.assertTrue(isinstance(requests.Session, type))
for attribute in dir(OriginalSession):
self.assertTrue(hasattr(requests.Session, attribute))
self.assertTrue(isinstance(requests.Session(), CachedSession))
def test_inheritance_after_monkey_patch(self):
requests_cache.install_cache(name=CACHE_NAME, backend=CACHE_BACKEND)
class FooSession(requests.Session):
__attrs__ = requests.Session.__attrs__ + ["new_one"]
def __init__(self, param):
self.param = param
super(FooSession, self).__init__()
s = FooSession(1)
self.assertEquals(s.param, 1)
self.assertIn("new_one", s.__attrs__)
self.assertTrue(isinstance(s, CachedSession))
def test_passing_backend_instance_support(self):
class MyCache(BaseCache):
pass
backend = MyCache()
requests_cache.install_cache(name=CACHE_NAME, backend=backend)
self.assertIs(requests.Session().cache, backend)
session = CachedSession(backend=backend)
self.assertIs(session.cache, backend)
if __name__ == '__main__':
unittest.main()
|
