def testAutoGzipObject(self):
# TODO(craigcitro): Move this to a new object once we have a more
# permanent one, see: http://b/12250275
request = storage.StorageObjectsGetRequest(bucket='ottenl-gzip',
object='50K.txt')
# First, try without auto-transfer.
self.__GetFile(request)
self.assertEqual(0, self.__buffer.tell())
self.__download.StreamInChunks()
self.assertEqual(50000, self.__buffer.tell())
# Next, try with auto-transfer.
self.__ResetDownload(auto_transfer=True)
self.__GetFile(request)
self.assertEqual(50000, self.__buffer.tell())
def testSerializedDownload(self):
def _ProgressCallback(unused_response, download_object):
print 'Progress %s' % download_object.progress
file_contents = self.__GetTestdataFileContents('fifteen_byte_file')
object_name = os.path.join(self._TESTDATA_PREFIX, 'fifteen_byte_file')
request = storage.StorageObjectsGetRequest(bucket=self._DEFAULT_BUCKET,
object=object_name)
response = self.__client.objects.Get(request)
self.__buffer = six.StringIO()
download_data = json.dumps({
'auto_transfer': False,
'progress': 0,
'total_size': response.size,
'url': response.mediaLink,
})
self.__download = storage.Download.FromData(self.__buffer,
download_data,
http=self.__client.http)
self.__download.StreamInChunks(callback=_ProgressCallback)
self.assertEqual(15, self.__buffer.tell())
self.__buffer.seek(0)
self.assertEqual(file_contents, self.__buffer.read(15))
def __GetRequest(self, filename):
object_name = os.path.join(self._TESTDATA_PREFIX, filename)
return storage.StorageObjectsGetRequest(object=object_name,
bucket=self._DEFAULT_BUCKET)
def __GetRequest(cls, filename):
object_name = os.path.join(cls._TESTDATA_PREFIX, filename)
return storage.StorageObjectsGetRequest(bucket=cls._DEFAULT_BUCKET,
object=object_name)
