Only grant permissions to new extensions from sync if they have the expected version
[chromium-blink-merge.git] / tools / telemetry / third_party / gsutilz / gslib / tests / test_tracker_file.py
blob76f980a4c95485f5c9866d7d9046f0f2c22979a9
1 # -*- coding: utf-8 -*-
2 # Copyright 2015 Google Inc. All Rights Reserved.
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 """Unit tests for tracker_file."""
17 from gslib.tests.testcase.unit_testcase import GsUtilUnitTestCase
18 from gslib.third_party.storage_apitools import storage_v1_messages as apitools_messages
19 from gslib.tracker_file import _HashFilename
20 from gslib.tracker_file import DeleteTrackerFile
21 from gslib.tracker_file import GetRewriteTrackerFilePath
22 from gslib.tracker_file import HashRewriteParameters
23 from gslib.tracker_file import ReadRewriteTrackerFile
24 from gslib.tracker_file import WriteRewriteTrackerFile
27 class TestTrackerFile(GsUtilUnitTestCase):
28 """Unit tests for parallel upload functions in cp command."""
30 def test_HashFilename(self):
31 # Tests that _HashFilename function works for both string and unicode
32 # filenames (without raising any Unicode encode/decode errors).
33 _HashFilename('file1')
34 _HashFilename(u'file1')
36 def test_RewriteTrackerFile(self):
37 """Tests Rewrite tracker file functions."""
38 tracker_file_name = GetRewriteTrackerFilePath('bk1', 'obj1', 'bk2', 'obj2',
39 self.test_api)
40 # Should succeed regardless of whether it exists.
41 DeleteTrackerFile(tracker_file_name)
42 src_obj_metadata = apitools_messages.Object(
43 bucket='bk1', name='obj1', etag='etag1', md5Hash='12345')
44 src_obj2_metadata = apitools_messages.Object(
45 bucket='bk1', name='obj1', etag='etag2', md5Hash='67890')
46 dst_obj_metadata = apitools_messages.Object(
47 bucket='bk2', name='obj2')
48 rewrite_token = 'token1'
49 self.assertIsNone(ReadRewriteTrackerFile(tracker_file_name,
50 src_obj_metadata))
51 rewrite_params_hash = HashRewriteParameters(
52 src_obj_metadata, dst_obj_metadata, 'full')
53 WriteRewriteTrackerFile(tracker_file_name, rewrite_params_hash,
54 rewrite_token)
55 self.assertEqual(
56 ReadRewriteTrackerFile(tracker_file_name, rewrite_params_hash),
57 rewrite_token)
59 # Tracker file for an updated source object (with non-matching etag/md5)
60 # should return None.
61 rewrite_params_hash2 = HashRewriteParameters(
62 src_obj2_metadata, dst_obj_metadata, 'full')
64 self.assertIsNone(ReadRewriteTrackerFile(tracker_file_name,
65 rewrite_params_hash2))
66 DeleteTrackerFile(tracker_file_name)