diff --git a/test_flask_kerberos.py b/test_flask_kerberos.py index 3cd3ee6..75f3017 100644 --- a/test_flask_kerberos.py +++ b/test_flask_kerberos.py @@ -50,7 +50,7 @@ def test_authorized(self, clean, name, response, step, init): c = self.app.test_client() r = c.get('/', headers={'Authorization': 'Negotiate CTOKEN'}) self.assertEqual(r.status_code, 200) - self.assertEqual(r.data, 'user@EXAMPLE.ORG') + self.assertEqual(r.data, b'user@EXAMPLE.ORG') self.assertEqual(r.headers.get('WWW-Authenticate'), 'negotiate STOKEN') self.assertEqual(init.mock_calls, [mock.call('HTTP@example.org')]) self.assertEqual(step.mock_calls, [mock.call(state, 'CTOKEN')]) @@ -77,7 +77,7 @@ def test_authorized_no_mutual_auth(self, clean, name, response, step, init): c = self.app.test_client() r = c.get('/', headers={'Authorization': 'Negotiate CTOKEN'}) self.assertEqual(r.status_code, 200) - self.assertEqual(r.data, 'user@EXAMPLE.ORG') + self.assertEqual(r.data, b'user@EXAMPLE.ORG') self.assertEqual(r.headers.get('WWW-Authenticate'), None) self.assertEqual(init.mock_calls, [mock.call('HTTP@example.org')]) self.assertEqual(step.mock_calls, [mock.call(state, 'CTOKEN')])