diff --git a/tests/test_configuration.py b/tests/test_configuration.py index 9a75ae0..0c9519d 100644 --- a/tests/test_configuration.py +++ b/tests/test_configuration.py @@ -23,11 +23,13 @@ def test_view_configuration(client): response = client.get('/custom_login') assert b"

Login

" in response.data - response = authenticate(client, endpoint='/custom_login', follow_redirects=True) - assert b'Post Login' in response.data + response = authenticate(client, endpoint='/custom_login') + assert b'location' in response.headers + assert response.headers['Location'] == 'http://localhost/post_login' - response = logout(client, endpoint='/custom_logout', follow_redirects=True) - assert b'Post Logout' in response.data + response = logout(client, endpoint='/custom_logout') + assert b'location' in response.headers + assert response.headers['Location'] == 'http://localhost/post_logout' response = client.get('/http', headers={ 'Authorization': 'Basic %s' % base64.b64encode(b"joe@lp.com:bogus")