Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def assertRedirects(self, response, location, message=None):
- """
- Checks if response is an HTTP redirect to the
- given location.
- :param response: Flask response
- :param location: relative URL path to SERVER_NAME or an absolute URL
- """
- parts_location = urlparse(location)
- valid_status_codes = (301, 302, 303, 305, 307)
- valid_status_code_str = ', '.join(str(code) for code in valid_status_codes)
- not_redirect = "HTTP Status %s expected but got %d" % (valid_status_code_str, response.status_code)
- self.assertTrue(response.status_code in valid_status_codes, message or not_redirect)
- if parts_location.netloc:
- expected_location = location
- else:
- server_name = self.app.config.get('SERVER_NAME') or 'localhost'
- expected_location = urljoin("http://%s" % server_name, location)
- #expected_location = location
- parts_response = urlparse(response.location)
- if parts_response.netloc:
- self.assertEqual(response.location, expected_location, message)
- else:
- server_name = self.app.config.get('SERVER_NAME') or 'localhost'
- response_url = urljoin("http://%s" % server_name, location)
- self.assertEqual(response_url, expected_location, message)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement