Index: tests/upload_to_google_storage_unittests.py |
diff --git a/tests/upload_to_google_storage_unittests.py b/tests/upload_to_google_storage_unittests.py |
index 24ac6b89c29cacd32bc2730de8e7acc2a9f448b6..3bac038d7402185c3a3a78fff7a4e3f0c4d8159b 100755 |
--- a/tests/upload_to_google_storage_unittests.py |
+++ b/tests/upload_to_google_storage_unittests.py |
@@ -48,14 +48,14 @@ class UploadTests(unittest.TestCase): |
filenames = [self.lorem_ipsum] |
output_filename = '%s.sha1' % self.lorem_ipsum |
code = upload_to_google_storage.upload_to_google_storage( |
- filenames, self.base_url, self.gsutil, True, False, 1, False) |
+ filenames, self.base_url, self.gsutil, True, False, 1, False, 'txt') |
self.assertEqual( |
self.gsutil.history, |
[('check_call', |
('ls', '%s/%s' % (self.base_url, self.lorem_ipsum_sha1))), |
('check_call', |
- ('cp', '-q', filenames[0], '%s/%s' % (self.base_url, |
- self.lorem_ipsum_sha1)))]) |
+ ('cp', '-z', 'txt', filenames[0], |
+ '%s/%s' % (self.base_url, self.lorem_ipsum_sha1)))]) |
self.assertTrue(os.path.exists(output_filename)) |
self.assertEqual( |
open(output_filename, 'rb').read(), |
@@ -70,7 +70,7 @@ class UploadTests(unittest.TestCase): |
self.gsutil.add_expected(0, '', '') |
self.gsutil.add_expected(0, etag_string, '') |
code = upload_to_google_storage.upload_to_google_storage( |
- filenames, self.base_url, self.gsutil, False, False, 1, False) |
+ filenames, self.base_url, self.gsutil, False, False, 1, False, None) |
self.assertEqual( |
self.gsutil.history, |
[('check_call', |
@@ -100,7 +100,8 @@ class UploadTests(unittest.TestCase): |
False, |
False, |
self.stdout_queue, |
- self.ret_codes) |
+ self.ret_codes, |
+ None) |
expected_ret_codes = [ |
(20, |
'Encountered error on uploading %s to %s/%s\nExpected error message' % |
@@ -114,7 +115,7 @@ class UploadTests(unittest.TestCase): |
with open(output_filename, 'wb') as f: |
f.write(fake_hash) # Fake hash. |
code = upload_to_google_storage.upload_to_google_storage( |
- filenames, self.base_url, self.gsutil, False, False, 1, True) |
+ filenames, self.base_url, self.gsutil, False, False, 1, True, None) |
self.assertEqual( |
self.gsutil.history, |
[('check_call', |
@@ -122,7 +123,7 @@ class UploadTests(unittest.TestCase): |
('check_call', |
('ls', '-L', '%s/%s' % (self.base_url, fake_hash))), |
('check_call', |
- ('cp', '-q', filenames[0], '%s/%s' % (self.base_url, fake_hash)))]) |
+ ('cp', filenames[0], '%s/%s' % (self.base_url, fake_hash)))]) |
self.assertEqual( |
open(output_filename, 'rb').read(), fake_hash) |
os.remove(output_filename) |
@@ -162,4 +163,4 @@ class UploadTests(unittest.TestCase): |
if __name__ == '__main__': |
- unittest.main() |
+ unittest.main() |