def test_redirects(self):
app = Flask(__name__)
def add_redirect(old_url, new_url):
app.add_url_rule(
old_url, 'redirect_' + old_url, lambda: redirect(new_url))
add_redirect('/a', '/b')
add_redirect('/b', '/c')
add_redirect('/c', '/d')
app.add_url_rule('/d', 'd', lambda: 'Ok')
add_redirect('/1', '/2')
add_redirect('/2', '/3')
add_redirect('/3', '/1') # redirect loop
with app.test_request_context():
fetcher = make_url_fetcher()
result = fetcher('http://localhost/a')
assert result['string'] == 'Ok'
assert result['redirected_url'] == 'http://localhost/d'
self.assertRaises(ClientRedirectError, fetcher, 'http://localhost/1')
self.assertRaises(ValueError, fetcher, 'http://localhost/nonexistent')
def test_redirects(self):
app = Flask(__name__)
def add_redirect(old_url, new_url):
app.add_url_rule(old_url, 'redirect_' + old_url,
lambda: redirect(new_url))
add_redirect('/a', '/b')
add_redirect('/b', '/c')
add_redirect('/c', '/d')
app.add_url_rule('/d', 'd', lambda: 'Ok')
add_redirect('/1', '/2')
add_redirect('/2', '/3')
add_redirect('/3', '/1') # redirect loop
with app.test_request_context():
fetcher = make_url_fetcher()
result = fetcher('http://localhost/a')
assert result['string'] == b'Ok'
assert result['redirected_url'] == 'http://localhost/d'
self.assertRaises(ClientRedirectError, fetcher, 'http://localhost/1')
self.assertRaises(ValueError, fetcher, 'http://localhost/nonexistent')
def add_redirect(old_url, new_url):
app.add_url_rule(
old_url, 'redirect_' + old_url, lambda: redirect(new_url))
def add_redirect(old_url, new_url):
app.add_url_rule(old_url, 'redirect_' + old_url,
lambda: redirect(new_url))