[refs sigmavirus24#112] Wrapping StringIO with an adaptor to deal with change to the method signature expected by requests

carlio · carlio · commit c7fde72ebadd · 2013-06-12T11:12:44.000+02:00
diff --git a/github3/decorators.py b/github3/decorators.py
@@ -18,6 +18,11 @@
     from io import BytesIO as StringIO
 
 
+class RequestsStringIO(StringIO):
+    def read(self, chunk_size, *args, **kwargs):
+        return super(RequestsStringIO, self).read(chunk_size)
+
+
 def requires_auth(func):
     """Decorator to note which object methods require authorization."""
     @wraps(func)
@@ -61,7 +66,7 @@ def generate_fake_error_response(msg, status_code=401, encoding='utf-8'):
     r = Response()
     r.status_code = status_code
     r.encoding = encoding
-    r.raw = StringIO(msg)
+    r.raw = RequestsStringIO(msg)
     r._content_consumed = True
     r._content = r.raw.read()
     return r
