Fix grantip on Python 3

1 files changed, 11 insertions, 8 deletions

diff --git a/tests/test_grantip.py b/tests/test_grantip.py
index 8d74280..2ddf7f1 100644
--- a/tests/test_grantip.py
+++ b/tests/test_grantip.py
@@ -4,11 +4,14 @@ from paste.fixture import *
 def test_make_app():
     def application(environ, start_response):
         start_response('200 OK', [('content-type', 'text/plain')])
-        return [
+        lines = [
             str(environ.get('REMOTE_USER')),
             ':',
             str(environ.get('REMOTE_USER_TOKENS')),
             ]
+        if six.PY3:
+            lines = [line.encode('utf8') for line in lines]
+        return lines
     ip_map = {
         '127.0.0.1': (None, 'system'),
         '192.168.0.0/16': (None, 'worker'),
@@ -24,11 +27,11 @@ def test_req():
     def doit(remote_addr):
         res = app.get('/', extra_environ={'REMOTE_ADDR': remote_addr})
         return res.body
-    assert doit('127.0.0.1') == 'None:system'
-    assert doit('192.168.15.12') == 'None:worker'
-    assert doit('192.168.0.4') == 'None:worker'
+    assert doit('127.0.0.1') == b'None:system'
+    assert doit('192.168.15.12') == b'None:worker'
+    assert doit('192.168.0.4') == b'None:worker'
     result = doit('192.168.0.5')
-    assert result.startswith('bob:')
-    assert 'editor' in result and 'worker' in result
-    assert result.count(',') == 1
-    assert doit('192.168.0.8') == 'None:editor'
+    assert result.startswith(b'bob:')
+    assert b'editor' in result and b'worker' in result
+    assert result.count(b',') == 1
+    assert doit('192.168.0.8') == b'None:editor'