Merge branch '85-after-login-url'

level12 · Nov 5, 2019 · 76a1222 · 76a1222
2 parents 1d67548 + c40b10c
commit 76a1222
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/keg_auth/testing.py b/keg_auth/testing.py
@@ -59,7 +59,9 @@ def test_login_field_success(self):
         resp = resp.form.submit()
 
         assert resp.status_code == 302, resp.html
-        assert resp.headers['Location'] == 'http://keg.example.com/'
+        assert resp.headers['Location'] == flask.url_for(
+            flask.current_app.auth_manager.endpoints['after-login']
+        )
         assert resp.flashes == [('success', 'Login successful.')]
 
     def test_login_field_success_next_parameter(self):
@@ -110,7 +112,9 @@ def test_next_parameter_not_open_redirect(self):
 
         assert resp.status_code == 302, resp.html
         # verify the 'next' parameter was ignored
-        assert resp.headers['Location'] == 'http://keg.example.com/'
+        assert resp.headers['Location'] == flask.url_for(
+            flask.current_app.auth_manager.endpoints['after-login']
+        )
         assert resp.flashes == [('success', 'Login successful.')]
 
         # quoted next parameter
@@ -123,7 +127,9 @@ def test_next_parameter_not_open_redirect(self):
 
         assert resp.status_code == 302, resp.html
         # verify the 'next' parameter was ignored
-        assert resp.headers['Location'] == 'http://keg.example.com/'
+        assert resp.headers['Location'] == flask.url_for(
+            flask.current_app.auth_manager.endpoints['after-login']
+        )
         assert resp.flashes == [('success', 'Login successful.')]
 
     def test_login_invalid_password(self):