def test_disable_default_redirect_cache(self, mock_request): """ Test disable default redirect cache (by setting default redirect cache to None) """ response0 = Response() response0.url = 'http://www.test.com/neverseemeagain' response0.status_code = 301 response0.headers = { 'Location': 'http://www.test.com/redirect_here', } response1 = Response() response1.url = 'http://www.test.com/redirect_here' response1.status_code = 200 response1._content = 'Mocked response content' response1.headers = { 'Vary': 'Accept', } response1.history = [response0] mock_request.return_value = response1 get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/redirect_here', allow_redirects=True) set_default_redirect_cache(None) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True)
def test_set_default_redirect_cache(self, mock_request): response0 = Response() response0.url = 'http://www.test.com/neverseemeagain' response0.status_code = 301 response0.headers = { 'Location': 'http://www.test.com/redirect_here', } response1 = Response() response1.url = 'http://www.test.com/redirect_here' response1.status_code = 200 response1._content = 'Mocked response content' response1.headers = { 'Vary': 'Accept', } response1.history = [response0] mock_request.return_value = response1 C0 = self.redirect_cache C1 = Cache() C2 = Cache() get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/redirect_here', allow_redirects=True) set_default_redirect_cache(C1) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/redirect_here', allow_redirects=True) set_default_redirect_cache(C2) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/neverseemeagain', allow_redirects=True) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/redirect_here', allow_redirects=True) set_default_redirect_cache(C0) get('http://www.test.com/neverseemeagain') mock_request.assert_called_with('GET', 'http://www.test.com/redirect_here', 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()