Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(827)

Unified Diff: tools/telemetry/third_party/gsutil/gslib/tests/test_rsync_funcs.py

Issue 1258583006: Add gsutil 4.13 to telemetry/third_party (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Undo all other changes so this just add gsutil to third_party Created 5 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/telemetry/third_party/gsutil/gslib/tests/test_rsync_funcs.py
diff --git a/tools/telemetry/third_party/gsutil/gslib/tests/test_rsync_funcs.py b/tools/telemetry/third_party/gsutil/gslib/tests/test_rsync_funcs.py
new file mode 100644
index 0000000000000000000000000000000000000000..0afe4a3affb6b417e1b85d901a5cb84c4a5b6540
--- /dev/null
+++ b/tools/telemetry/third_party/gsutil/gslib/tests/test_rsync_funcs.py
@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# Copyright 2014 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Unit tests for functions in rsync command."""
+
+import logging
+import os
+
+from gslib.commands.rsync import _ComputeNeededFileChecksums
+from gslib.commands.rsync import _NA
+from gslib.hashing_helper import CalculateB64EncodedCrc32cFromContents
+from gslib.hashing_helper import CalculateB64EncodedMd5FromContents
+from gslib.tests.testcase.unit_testcase import GsUtilUnitTestCase
+
+
+class TestRsyncFuncs(GsUtilUnitTestCase):
+
+ def test_compute_needed_file_checksums(self):
+ """Tests that we compute all/only needed file checksums."""
+ size = 4
+ logger = logging.getLogger()
+ tmpdir = self.CreateTempDir()
+ file_url_str = 'file://%s' % os.path.join(tmpdir, 'obj1')
+ self.CreateTempFile(tmpdir=tmpdir, file_name='obj1', contents='obj1')
+ cloud_url_str = 'gs://whatever'
+ with open(os.path.join(tmpdir, 'obj1'), 'rb') as fp:
+ crc32c = CalculateB64EncodedCrc32cFromContents(fp)
+ fp.seek(0)
+ md5 = CalculateB64EncodedMd5FromContents(fp)
+
+ # Test case where source is a file and dest has CRC32C.
+ (src_crc32c, src_md5, dst_crc32c, dst_md5) = _ComputeNeededFileChecksums(
+ logger, file_url_str, size, _NA, _NA, cloud_url_str, size, crc32c, _NA)
+ self.assertEquals(crc32c, src_crc32c)
+ self.assertEquals(_NA, src_md5)
+ self.assertEquals(crc32c, dst_crc32c)
+ self.assertEquals(_NA, dst_md5)
+
+ # Test case where source is a file and dest has MD5 but not CRC32C.
+ (src_crc32c, src_md5, dst_crc32c, dst_md5) = _ComputeNeededFileChecksums(
+ logger, file_url_str, size, _NA, _NA, cloud_url_str, size, _NA, md5)
+ self.assertEquals(_NA, src_crc32c)
+ self.assertEquals(md5, src_md5)
+ self.assertEquals(_NA, dst_crc32c)
+ self.assertEquals(md5, dst_md5)
+
+ # Test case where dest is a file and src has CRC32C.
+ (src_crc32c, src_md5, dst_crc32c, dst_md5) = _ComputeNeededFileChecksums(
+ logger, cloud_url_str, size, crc32c, _NA, file_url_str, size, _NA, _NA)
+ self.assertEquals(crc32c, dst_crc32c)
+ self.assertEquals(_NA, src_md5)
+ self.assertEquals(crc32c, src_crc32c)
+ self.assertEquals(_NA, src_md5)
+
+ # Test case where dest is a file and src has MD5 but not CRC32C.
+ (src_crc32c, src_md5, dst_crc32c, dst_md5) = _ComputeNeededFileChecksums(
+ logger, cloud_url_str, size, _NA, md5, file_url_str, size, _NA, _NA)
+ self.assertEquals(_NA, dst_crc32c)
+ self.assertEquals(md5, src_md5)
+ self.assertEquals(_NA, src_crc32c)
+ self.assertEquals(md5, src_md5)

Powered by Google App Engine
This is Rietveld 408576698