Fixed test encoding for Python 3.

JNRowe · JNRowe · commit 129a612f1815 · 2011-06-11T06:24:59.000+01:00
diff --git a/tests/utils.py b/tests/utils.py
@@ -1,6 +1,7 @@
 import _setup
 
 import os
+import sys
 import unittest
 
 from email import message_from_file
@@ -11,6 +12,10 @@
 from github2.request import charset_from_headers
 
 
+if sys.version_info[0] == 2:
+    bytes = lambda x, enc: x
+
+
 HTTP_DATA_DIR = "tests/data/"
 
 ORIG_HTTP_OBJECT = httplib2.Http
@@ -37,7 +42,7 @@ def request(self, uri, method='GET', body=None, headers=None,
         if os.path.exists(file):
             response = message_from_file(open(file))
             headers = httplib2.Response(response)
-            body = response.get_payload().encode(charset_from_headers(headers))
+            body = bytes(response.get_payload(), charset_from_headers(headers))
             return (headers, body)
         else:
             return (httplib2.Response({"status": "404"}),
