Fix a bug in download streaming.

craigcitro · craigcitro · commit a6feb8a17c6d · 2015-04-08T16:01:35.000-07:00
Currently apitools has a bug where downloads with (1) no total size and (2)
auto_transfer off will only stream one chunk and quit.

(Test added internally, but internal tests still aren't sync'd.)

There's still more to be done around normalizing use of chunksize in
downloads.
diff --git a/apitools/base/py/transfer.py b/apitools/base/py/transfer.py
@@ -404,6 +404,8 @@ def StreamInChunks(self, callback=None, finish_callback=None,
       else:
         response = self.__GetChunk(self.progress,
                                    additional_headers=additional_headers)
+      if self.total_size is None:
+        self.__SetTotal(response.info)
       response = self.__ProcessResponse(response)
       self._ExecuteCallback(callback, response)
       if (response.status_code == http_client.OK or
