diff --git a/tests/test_factory.py b/tests/test_factory.py index d858a1d..134ce9e 100644 --- a/tests/test_factory.py +++ b/tests/test_factory.py @@ -59,5 +59,5 @@ def test_hsts_header(app_factory): client = app_factory.app.test_client() r = client.get('/', base_url='https://example.com') assert r.status_code == 200 - assert r.data == 'home' + assert r.data.decode('utf-8') == 'home' assert r.headers['Strict-Transport-Security'] == 'max-age=31536000' diff --git a/tests/test_simple.py b/tests/test_simple.py index 04a87bd..58fd1f0 100644 --- a/tests/test_simple.py +++ b/tests/test_simple.py @@ -35,5 +35,5 @@ def test_hsts_header(sslify): client = sslify.app.test_client() r = client.get('/', base_url='https://example.com') assert r.status_code == 200 - assert r.data == 'home' + assert r.data.decode('utf-8') == 'home' assert r.headers['Strict-Transport-Security'] == 'max-age=31536000'