Fix test for python 3

porn · porn · commit 9c653add4096 · 2016-03-18T09:12:16.000+01:00
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -255,7 +255,10 @@ def text(self):
 class MockPdfResponse(object):
     @property
     def status_code(self):
-        import httplib
+        try:
+            import httplib  # python 2
+        except ImportError:
+            import http.client as httplib  # python 3
         return httplib.OK
 
     @property
