Fix tests to pass python 3

pallets-eco · Jun 10, 2014 · 96f1b3e · 96f1b3e
1 parent 0facdaa
commit 96f1b3e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/flask_security/utils.py b/flask_security/utils.py
@@ -150,7 +150,7 @@ def encrypt_password(password):
 
 
 def md5(data):
-    return hashlib.md5(data.encode('ascii')).hexdigest()
+    return hashlib.md5(data.encode('utf-8')).hexdigest()
 
 
 def do_flash(message, category=None):

diff --git a/tests/test_configuration.py b/tests/test_configuration.py
@@ -24,11 +24,11 @@ def test_view_configuration(client):
     assert b"<h1>Login</h1>" in response.data
 
     response = authenticate(client, endpoint='/custom_login')
-    assert b'location' in response.headers
+    assert 'location' in response.headers
     assert response.headers['Location'] == 'http://localhost/post_login'
 
     response = logout(client, endpoint='/custom_logout')
-    assert b'location' in response.headers
+    assert 'location' in response.headers
     assert response.headers['Location'] == 'http://localhost/post_logout'
 
     response = client.get('/http', headers={