def testDelegateUploadFileToObjectNormal(self):
mock_stream = mock.Mock()
mock_stream.close = mock.Mock()
def DelegateUpload():
return 'a', 'b'
elapsed_time, uploaded_object = _DelegateUploadFileToObject(
DelegateUpload, 'url', mock_stream, False, False, False, None)
# Ensure results are passed through.
self.assertEqual(elapsed_time, 'a')
self.assertEqual(uploaded_object, 'b')
# Ensure close was called.
self.assertTrue(mock_stream.close.called)
def testDelegateUploadFileToObjectGzipEncodedComposite(self, mock_lock):
mock_stream = mock.Mock()
mock_stream.close = mock.Mock()
def DelegateUpload():
# Ensure the lock was not aquired before the delegate was called.
self.assertFalse(mock_lock.__enter__.called)
return 'a', 'b'
elapsed_time, uploaded_object = _DelegateUploadFileToObject(
DelegateUpload, 'url', mock_stream, False, True, True, None)
# Ensure results are passed through.
self.assertEqual(elapsed_time, 'a')
self.assertEqual(uploaded_object, 'b')
# Ensure close was called.
self.assertTrue(mock_stream.close.called)
# Ensure the lock was released.
self.assertFalse(mock_lock.__exit__.called)
def testDelegateUploadFileToObjectZipped(self, mock_unlink):
mock_stream = mock.Mock()
mock_stream.close = mock.Mock()
mock_upload_url = mock.Mock()
mock_upload_url.object_name = 'Sample'
def DelegateUpload():
return 'a', 'b'
elapsed_time, uploaded_object = _DelegateUploadFileToObject(
DelegateUpload, mock_upload_url, mock_stream, True, False, False, None)
# Ensure results are passed through.
self.assertEqual(elapsed_time, 'a')
self.assertEqual(uploaded_object, 'b')
# Ensure the file was unlinked.
self.assertTrue(mock_unlink.called)
# Ensure close was called.
self.assertTrue(mock_stream.close.called)
