def test_set_default_cookie_cache(self, mock_request): response = Response() response.headers = { 'Set-Cookie': 'name=value', } response.url = 'http://www.test.com/path' mock_request.return_value = response C0 = self.cookie_cache C1 = Cache() C2 = Cache() get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', cookies={'name': 'value'}, allow_redirects=True) set_default_cookie_cache(C1) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', cookies={'name': 'value'}, allow_redirects=True) set_default_cookie_cache(C2) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', cookies={'name': 'value'}, allow_redirects=True) set_default_cookie_cache(C0) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', cookies={'name': 'value'}, allow_redirects=True)
def test_disable_default_cookie_cache(self, mock_request): """ Test disable default cookie cache (by setting default cookie cache to None) """ response = Response() response.headers = { 'Set-Cookie': 'name=value', } response.url = 'http://www.test.com/path' mock_request.return_value = response get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', cookies={'name': 'value'}, allow_redirects=True) set_default_cookie_cache(None) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True) get('http://www.test.com/path') mock_request.assert_called_with('GET', 'http://www.test.com/path', allow_redirects=True)
def tearDown(self): set_default_redirect_cache(self._orig_default_redirect_cache) set_default_cookie_cache(self._orig_default_cookie_cache) set_default_cache(self._orig_default_cache) super(TestDefaults, self).tearDown()