setting the stream part of the request

core-api · guydou · Dec 10, 2017 · Dec 10, 2017 · Dec 10, 2017 · Dec 10, 2017
commit 552bce22c226ae94a57e5ced440512643bc4c39a
diff --git a/coreapi/transports/http.py b/coreapi/transports/http.py
@@ -200,7 +200,7 @@ def _get_upload_headers(file_obj):
     }
 
 
-def _build_http_request(session, url, method, headers=None, encoding=None, params=empty_params):
+def _build_http_request(session, url, method, headers=None, encoding=None, params=empty_params, stream=False):
     """
     Make an HTTP request and return an HTTP response.
     """
@@ -210,6 +210,7 @@ def _build_http_request(session, url, method, headers=None, encoding=None, param
 
     if params.query:
         opts['params'] = params.query
+    opts['stream'] = stream
 
     if params.data or params.files:
         if encoding == 'application/json':
@@ -376,8 +377,8 @@ def transition(self, link, decoders, params=None, link_ancestors=None, force_cod
         headers = _get_headers(url, decoders)
         headers.update(self.headers)
 
-        request = _build_http_request(session, url, method, headers, encoding, params)
-        with session.send(request, stream=stream) as response:
+        request = _build_http_request(session, url, method, headers, encoding, params, stream=stream)
+        with session.send(request) as response:
             result = _decode_result(response, decoders, force_codec)
 
         if isinstance(result, Document) and link_ancestors: