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

Unified Diff: third_party/gsutil/gslib/tests/test_acl.py

Issue 1377933002: [catapult] - Copy Telemetry's gsutilz over to third_party. (Closed) Base URL: https://github.com/catapult-project/catapult.git@master
Patch Set: Rename to gsutil. Created 5 years, 3 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
« no previous file with comments | « third_party/gsutil/gslib/tests/test_Doption.py ('k') | third_party/gsutil/gslib/tests/test_bucketconfig.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: third_party/gsutil/gslib/tests/test_acl.py
diff --git a/third_party/gsutil/gslib/tests/test_acl.py b/third_party/gsutil/gslib/tests/test_acl.py
new file mode 100644
index 0000000000000000000000000000000000000000..79c2c5b982e0ad1b07abbf71729d25b06ef14a4f
--- /dev/null
+++ b/third_party/gsutil/gslib/tests/test_acl.py
@@ -0,0 +1,681 @@
+# -*- coding: utf-8 -*-
+# Copyright 2013 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.
+"""Integration tests for the acl command."""
+
+from __future__ import absolute_import
+
+import re
+
+from gslib import aclhelpers
+from gslib.command import CreateGsutilLogger
+from gslib.cs_api_map import ApiSelector
+from gslib.project_id import PopulateProjectId
+from gslib.storage_url import StorageUrlFromString
+import gslib.tests.testcase as testcase
+from gslib.tests.testcase.integration_testcase import SkipForGS
+from gslib.tests.testcase.integration_testcase import SkipForS3
+from gslib.tests.util import ObjectToURI as suri
+from gslib.translation_helper import AclTranslation
+from gslib.util import Retry
+
+PUBLIC_READ_JSON_ACL_TEXT = '"entity":"allUsers","role":"READER"'
+
+
+class TestAclBase(testcase.GsUtilIntegrationTestCase):
+ """Integration test case base class for acl command."""
+
+ _set_acl_prefix = ['acl', 'set']
+ _get_acl_prefix = ['acl', 'get']
+ _set_defacl_prefix = ['defacl', 'set']
+ _ch_acl_prefix = ['acl', 'ch']
+
+ _project_team = 'owners'
+ _project_test_acl = '%s-%s' % (_project_team, PopulateProjectId())
+
+
+@SkipForS3('Tests use GS ACL model.')
+class TestAcl(TestAclBase):
+ """Integration tests for acl command."""
+
+ def setUp(self):
+ super(TestAcl, self).setUp()
+ self.sample_uri = self.CreateBucket()
+ self.sample_url = StorageUrlFromString(str(self.sample_uri))
+ self.logger = CreateGsutilLogger('acl')
+
+ def test_set_invalid_acl_object(self):
+ """Ensures that invalid content returns a bad request error."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ inpath = self.CreateTempFile(contents='badAcl')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, obj_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('ArgumentException', stderr)
+
+ def test_set_invalid_acl_bucket(self):
+ """Ensures that invalid content returns a bad request error."""
+ bucket_uri = suri(self.CreateBucket())
+ inpath = self.CreateTempFile(contents='badAcl')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, bucket_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('ArgumentException', stderr)
+
+ def test_set_xml_acl_json_api_object(self):
+ """Ensures XML content returns a bad request error and migration warning."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ inpath = self.CreateTempFile(contents='<ValidXml></ValidXml>')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, obj_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('ArgumentException', stderr)
+ self.assertIn('XML ACL data provided', stderr)
+
+ def test_set_xml_acl_json_api_bucket(self):
+ """Ensures XML content returns a bad request error and migration warning."""
+ bucket_uri = suri(self.CreateBucket())
+ inpath = self.CreateTempFile(contents='<ValidXml></ValidXml>')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, bucket_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('ArgumentException', stderr)
+ self.assertIn('XML ACL data provided', stderr)
+
+ def test_set_valid_acl_object(self):
+ """Tests setting a valid ACL on an object."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ acl_string = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+ inpath = self.CreateTempFile(contents=acl_string)
+ self.RunGsUtil(self._set_acl_prefix + ['public-read', obj_uri])
+ acl_string2 = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+ self.RunGsUtil(self._set_acl_prefix + [inpath, obj_uri])
+ acl_string3 = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+
+ self.assertNotEqual(acl_string, acl_string2)
+ self.assertEqual(acl_string, acl_string3)
+
+ def test_set_valid_permission_whitespace_object(self):
+ """Ensures that whitespace is allowed in role and entity elements."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ acl_string = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+ acl_string = re.sub(r'"role"', r'"role" \n', acl_string)
+ acl_string = re.sub(r'"entity"', r'\n "entity"', acl_string)
+ inpath = self.CreateTempFile(contents=acl_string)
+
+ self.RunGsUtil(self._set_acl_prefix + [inpath, obj_uri])
+
+ def test_set_valid_acl_bucket(self):
+ """Ensures that valid canned and XML ACLs work with get/set."""
+ bucket_uri = suri(self.CreateBucket())
+ acl_string = self.RunGsUtil(self._get_acl_prefix + [bucket_uri],
+ return_stdout=True)
+ inpath = self.CreateTempFile(contents=acl_string)
+ self.RunGsUtil(self._set_acl_prefix + ['public-read', bucket_uri])
+ acl_string2 = self.RunGsUtil(self._get_acl_prefix + [bucket_uri],
+ return_stdout=True)
+ self.RunGsUtil(self._set_acl_prefix + [inpath, bucket_uri])
+ acl_string3 = self.RunGsUtil(self._get_acl_prefix + [bucket_uri],
+ return_stdout=True)
+
+ self.assertNotEqual(acl_string, acl_string2)
+ self.assertEqual(acl_string, acl_string3)
+
+ def test_invalid_canned_acl_object(self):
+ """Ensures that an invalid canned ACL returns a CommandException."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ stderr = self.RunGsUtil(
+ self._set_acl_prefix + ['not-a-canned-acl', obj_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('CommandException', stderr)
+ self.assertIn('Invalid canned ACL', stderr)
+
+ def test_set_valid_def_acl_bucket(self):
+ """Ensures that valid default canned and XML ACLs works with get/set."""
+ bucket_uri = self.CreateBucket()
+
+ # Default ACL is project private.
+ obj_uri1 = suri(self.CreateObject(bucket_uri=bucket_uri, contents='foo'))
+ acl_string = self.RunGsUtil(self._get_acl_prefix + [obj_uri1],
+ return_stdout=True)
+
+ # Change it to authenticated-read.
+ self.RunGsUtil(
+ self._set_defacl_prefix + ['authenticated-read', suri(bucket_uri)])
+ obj_uri2 = suri(self.CreateObject(bucket_uri=bucket_uri, contents='foo2'))
+ acl_string2 = self.RunGsUtil(self._get_acl_prefix + [obj_uri2],
+ return_stdout=True)
+
+ # Now change it back to the default via XML.
+ inpath = self.CreateTempFile(contents=acl_string)
+ self.RunGsUtil(self._set_defacl_prefix + [inpath, suri(bucket_uri)])
+ obj_uri3 = suri(self.CreateObject(bucket_uri=bucket_uri, contents='foo3'))
+ acl_string3 = self.RunGsUtil(self._get_acl_prefix + [obj_uri3],
+ return_stdout=True)
+
+ self.assertNotEqual(acl_string, acl_string2)
+ self.assertIn('allAuthenticatedUsers', acl_string2)
+ self.assertEqual(acl_string, acl_string3)
+
+ def test_acl_set_version_specific_uri(self):
+ """Tests setting an ACL on a specific version of an object."""
+ bucket_uri = self.CreateVersionedBucket()
+ # Create initial object version.
+ uri = self.CreateObject(bucket_uri=bucket_uri, contents='data')
+ # Create a second object version.
+ inpath = self.CreateTempFile(contents='def')
+ self.RunGsUtil(['cp', inpath, uri.uri])
+
+ # Find out the two object version IDs.
+ lines = self.AssertNObjectsInBucket(bucket_uri, 2, versioned=True)
+ v0_uri_str, v1_uri_str = lines[0], lines[1]
+
+ # Check that neither version currently has public-read permission
+ # (default ACL is project-private).
+ orig_acls = []
+ for uri_str in (v0_uri_str, v1_uri_str):
+ acl = self.RunGsUtil(self._get_acl_prefix + [uri_str],
+ return_stdout=True)
+ self.assertNotIn(PUBLIC_READ_JSON_ACL_TEXT,
+ self._strip_json_whitespace(acl))
+ orig_acls.append(acl)
+
+ # Set the ACL for the older version of the object to public-read.
+ self.RunGsUtil(self._set_acl_prefix + ['public-read', v0_uri_str])
+ # Check that the older version's ACL is public-read, but newer version
+ # is not.
+ acl = self.RunGsUtil(self._get_acl_prefix + [v0_uri_str],
+ return_stdout=True)
+ self.assertIn(PUBLIC_READ_JSON_ACL_TEXT, self._strip_json_whitespace(acl))
+ acl = self.RunGsUtil(self._get_acl_prefix + [v1_uri_str],
+ return_stdout=True)
+ self.assertNotIn(PUBLIC_READ_JSON_ACL_TEXT,
+ self._strip_json_whitespace(acl))
+
+ # Check that reading the ACL with the version-less URI returns the
+ # original ACL (since the version-less URI means the current version).
+ acl = self.RunGsUtil(self._get_acl_prefix + [uri.uri], return_stdout=True)
+ self.assertEqual(acl, orig_acls[0])
+
+ def _strip_json_whitespace(self, json_text):
+ return re.sub(r'\s*', '', json_text)
+
+ def testAclChangeWithUserId(self):
+ change = aclhelpers.AclChange(self.USER_TEST_ID + ':r',
+ scope_type=aclhelpers.ChangeType.USER)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'UserById', self.USER_TEST_ID)
+
+ def testAclChangeWithGroupId(self):
+ change = aclhelpers.AclChange(self.GROUP_TEST_ID + ':r',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'GroupById', self.GROUP_TEST_ID)
+
+ def testAclChangeWithUserEmail(self):
+ change = aclhelpers.AclChange(self.USER_TEST_ADDRESS + ':r',
+ scope_type=aclhelpers.ChangeType.USER)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'UserByEmail', self.USER_TEST_ADDRESS)
+
+ def testAclChangeWithGroupEmail(self):
+ change = aclhelpers.AclChange(self.GROUP_TEST_ADDRESS + ':fc',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'OWNER', 'GroupByEmail', self.GROUP_TEST_ADDRESS)
+
+ def testAclChangeWithDomain(self):
+ change = aclhelpers.AclChange(self.DOMAIN_TEST + ':READ',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'GroupByDomain', self.DOMAIN_TEST)
+
+ def testAclChangeWithProjectOwners(self):
+ change = aclhelpers.AclChange(self._project_test_acl + ':READ',
+ scope_type=aclhelpers.ChangeType.PROJECT)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'Project', self._project_test_acl)
+
+ def testAclChangeWithAllUsers(self):
+ change = aclhelpers.AclChange('AllUsers:WRITE',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'WRITER', 'AllUsers')
+
+ def testAclChangeWithAllAuthUsers(self):
+ change = aclhelpers.AclChange('AllAuthenticatedUsers:READ',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ change.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'AllAuthenticatedUsers')
+ remove = aclhelpers.AclDel('AllAuthenticatedUsers')
+ remove.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHasNo(acl, 'READER', 'AllAuthenticatedUsers')
+
+ def testAclDelWithUser(self):
+ add = aclhelpers.AclChange(self.USER_TEST_ADDRESS + ':READ',
+ scope_type=aclhelpers.ChangeType.USER)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ add.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'UserByEmail', self.USER_TEST_ADDRESS)
+
+ remove = aclhelpers.AclDel(self.USER_TEST_ADDRESS)
+ remove.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHasNo(acl, 'READ', 'UserByEmail', self.USER_TEST_ADDRESS)
+
+ def testAclDelWithProjectOwners(self):
+ add = aclhelpers.AclChange(self._project_test_acl + ':READ',
+ scope_type=aclhelpers.ChangeType.PROJECT)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ add.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'Project', self._project_test_acl)
+
+ remove = aclhelpers.AclDel(self._project_test_acl)
+ remove.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHasNo(acl, 'READ', 'Project', self._project_test_acl)
+
+ def testAclDelWithGroup(self):
+ add = aclhelpers.AclChange(self.USER_TEST_ADDRESS + ':READ',
+ scope_type=aclhelpers.ChangeType.GROUP)
+ acl = list(AclTranslation.BotoBucketAclToMessage(self.sample_uri.get_acl()))
+ add.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHas(acl, 'READER', 'GroupByEmail', self.USER_TEST_ADDRESS)
+
+ remove = aclhelpers.AclDel(self.USER_TEST_ADDRESS)
+ remove.Execute(self.sample_url, acl, 'acl', self.logger)
+ self._AssertHasNo(acl, 'READER', 'GroupByEmail', self.GROUP_TEST_ADDRESS)
+
+ #
+ # Here are a whole lot of verbose asserts
+ #
+
+ def _AssertHas(self, current_acl, perm, scope, value=None):
+ matches = list(self._YieldMatchingEntriesJson(current_acl, perm, scope,
+ value))
+ self.assertEqual(1, len(matches))
+
+ def _AssertHasNo(self, current_acl, perm, scope, value=None):
+ matches = list(self._YieldMatchingEntriesJson(current_acl, perm, scope,
+ value))
+ self.assertEqual(0, len(matches))
+
+ def _YieldMatchingEntriesJson(self, current_acl, perm, scope, value=None):
+ """Generator that yields entries that match the change descriptor.
+
+ Args:
+ current_acl: A list of apitools_messages.BucketAccessControls or
+ ObjectAccessControls which will be searched for matching
+ entries.
+ perm: Role (permission) to match.
+ scope: Scope type to match.
+ value: Value to match (against the scope type).
+
+ Yields:
+ An apitools_messages.BucketAccessControl or ObjectAccessControl.
+ """
+ for entry in current_acl:
+ if (scope in ['UserById', 'GroupById'] and
+ entry.entityId and value == entry.entityId and
+ entry.role == perm):
+ yield entry
+ elif (scope in ['UserByEmail', 'GroupByEmail'] and
+ entry.email and value == entry.email and
+ entry.role == perm):
+ yield entry
+ elif (scope == 'GroupByDomain' and
+ entry.domain and value == entry.domain and
+ entry.role == perm):
+ yield entry
+ elif (scope == 'Project' and entry.role == perm and
+ value == entry.entityId):
+ yield entry
+ elif (scope in ['AllUsers', 'AllAuthenticatedUsers'] and
+ entry.entity.lower() == scope.lower() and
+ entry.role == perm):
+ yield entry
+
+ def _MakeScopeRegex(self, role, entity_type, email_address):
+ template_regex = (r'\{.*"entity":\s*"%s-%s".*"role":\s*"%s".*\}' %
+ (entity_type, email_address, role))
+ return re.compile(template_regex, flags=re.DOTALL)
+
+ def _MakeProjectScopeRegex(self, role, project_team):
+ template_regex = (r'\{.*"entity":\s*"project-%s-\d+",\s*"projectTeam":\s*'
+ r'\{\s*"projectNumber":\s*"(\d+)",\s*"team":\s*"%s"\s*\},'
+ r'\s*"role":\s*"%s".*\}') % (project_team, project_team,
+ role)
+
+ return re.compile(template_regex, flags=re.DOTALL)
+
+ def testBucketAclChange(self):
+ """Tests acl change on a bucket."""
+ test_regex = self._MakeScopeRegex(
+ 'OWNER', 'user', self.USER_TEST_ADDRESS)
+ json_text = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-u', self.USER_TEST_ADDRESS+':fc', suri(self.sample_uri)])
+ json_text = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+ self.assertRegexpMatches(json_text, test_regex)
+
+ test_regex2 = self._MakeScopeRegex(
+ 'WRITER', 'user', self.USER_TEST_ADDRESS)
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-u', self.USER_TEST_ADDRESS+':w', suri(self.sample_uri)])
+ json_text2 = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+ self.assertRegexpMatches(json_text2, test_regex2)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-d', self.USER_TEST_ADDRESS, suri(self.sample_uri)])
+
+ json_text3 = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+ self.assertNotRegexpMatches(json_text3, test_regex)
+
+ def testProjectAclChangesOnBucket(self):
+ """Tests project entity acl changes on a bucket."""
+
+ if self.test_api == ApiSelector.XML:
+ stderr = self.RunGsUtil(self._ch_acl_prefix +
+ ['-p', self._project_test_acl +':w',
+ suri(self.sample_uri)],
+ expected_status=1,
+ return_stderr=True)
+ self.assertIn(('CommandException: XML API does not support project'
+ ' scopes, cannot translate ACL.'), stderr)
+ else:
+ test_regex = self._MakeProjectScopeRegex(
+ 'WRITER', self._project_team)
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-p', self._project_test_acl +':w',
+ suri(self.sample_uri)])
+ json_text = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+
+ self.assertRegexpMatches(json_text, test_regex)
+
+ # The api will accept string project ids, but stores the numeric project
+ # ids internally, this extracts the numeric id from the returned acls.
+ proj_num_id = test_regex.search(json_text).group(1)
+ acl_to_remove = '%s-%s' % (self._project_team, proj_num_id)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-d', acl_to_remove, suri(self.sample_uri)])
+
+ json_text2 = self.RunGsUtil(
+ self._get_acl_prefix + [suri(self.sample_uri)], return_stdout=True)
+ self.assertNotRegexpMatches(json_text2, test_regex)
+
+ def testObjectAclChange(self):
+ """Tests acl change on an object."""
+ obj = self.CreateObject(bucket_uri=self.sample_uri, contents='something')
+ self.AssertNObjectsInBucket(self.sample_uri, 1)
+
+ test_regex = self._MakeScopeRegex(
+ 'READER', 'group', self.GROUP_TEST_ADDRESS)
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-g', self.GROUP_TEST_ADDRESS+':READ', suri(obj)])
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text, test_regex)
+
+ test_regex2 = self._MakeScopeRegex(
+ 'OWNER', 'group', self.GROUP_TEST_ADDRESS)
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-g', self.GROUP_TEST_ADDRESS+':OWNER', suri(obj)])
+ json_text2 = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text2, test_regex2)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-d', self.GROUP_TEST_ADDRESS, suri(obj)])
+ json_text3 = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text3, test_regex2)
+
+ all_auth_regex = re.compile(
+ r'\{.*"entity":\s*"allAuthenticatedUsers".*"role":\s*"OWNER".*\}',
+ flags=re.DOTALL)
+
+ self.RunGsUtil(self._ch_acl_prefix + ['-g', 'AllAuth:O', suri(obj)])
+ json_text4 = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text4, all_auth_regex)
+
+ def testObjectAclChangeAllUsers(self):
+ """Tests acl ch AllUsers:R on an object."""
+ obj = self.CreateObject(bucket_uri=self.sample_uri, contents='something')
+ self.AssertNObjectsInBucket(self.sample_uri, 1)
+
+ all_users_regex = re.compile(
+ r'\{.*"entity":\s*"allUsers".*"role":\s*"READER".*\}', flags=re.DOTALL)
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, all_users_regex)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-g', 'AllUsers:R', suri(obj)])
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text, all_users_regex)
+
+ def testMultithreadedAclChange(self, count=10):
+ """Tests multi-threaded acl changing on several objects."""
+ objects = []
+ for i in range(count):
+ objects.append(self.CreateObject(
+ bucket_uri=self.sample_uri,
+ contents='something {0}'.format(i)))
+
+ self.AssertNObjectsInBucket(self.sample_uri, count)
+
+ test_regex = self._MakeScopeRegex(
+ 'READER', 'group', self.GROUP_TEST_ADDRESS)
+ json_texts = []
+ for obj in objects:
+ json_texts.append(self.RunGsUtil(
+ self._get_acl_prefix + [suri(obj)], return_stdout=True))
+ for json_text in json_texts:
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ uris = [suri(obj) for obj in objects]
+ self.RunGsUtil(['-m', '-DD'] + self._ch_acl_prefix +
+ ['-g', self.GROUP_TEST_ADDRESS+':READ'] + uris)
+
+ json_texts = []
+ for obj in objects:
+ json_texts.append(self.RunGsUtil(
+ self._get_acl_prefix + [suri(obj)], return_stdout=True))
+ for json_text in json_texts:
+ self.assertRegexpMatches(json_text, test_regex)
+
+ def testRecursiveChangeAcl(self):
+ """Tests recursively changing ACLs on nested objects."""
+ obj = self.CreateObject(bucket_uri=self.sample_uri, object_name='foo/bar',
+ contents='something')
+ self.AssertNObjectsInBucket(self.sample_uri, 1)
+
+ test_regex = self._MakeScopeRegex(
+ 'READER', 'group', self.GROUP_TEST_ADDRESS)
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ @Retry(AssertionError, tries=5, timeout_secs=1)
+ def _AddAcl():
+ self.RunGsUtil(
+ self._ch_acl_prefix +
+ ['-R', '-g', self.GROUP_TEST_ADDRESS+':READ', suri(obj)[:-3]])
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text, test_regex)
+ _AddAcl()
+
+ @Retry(AssertionError, tries=5, timeout_secs=1)
+ def _DeleteAcl():
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-d', self.GROUP_TEST_ADDRESS, suri(obj)])
+ json_text = self.RunGsUtil(self._get_acl_prefix + [suri(obj)],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+ _DeleteAcl()
+
+ def testMultiVersionSupport(self):
+ """Tests changing ACLs on multiple object versions."""
+ bucket = self.CreateVersionedBucket()
+ object_name = self.MakeTempName('obj')
+ self.CreateObject(
+ bucket_uri=bucket, object_name=object_name, contents='One thing')
+ # Create another on the same URI, giving us a second version.
+ self.CreateObject(
+ bucket_uri=bucket, object_name=object_name, contents='Another thing')
+
+ lines = self.AssertNObjectsInBucket(bucket, 2, versioned=True)
+
+ obj_v1, obj_v2 = lines[0], lines[1]
+
+ test_regex = self._MakeScopeRegex(
+ 'READER', 'group', self.GROUP_TEST_ADDRESS)
+ json_text = self.RunGsUtil(self._get_acl_prefix + [obj_v1],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ self.RunGsUtil(self._ch_acl_prefix +
+ ['-g', self.GROUP_TEST_ADDRESS+':READ', obj_v1])
+ json_text = self.RunGsUtil(self._get_acl_prefix + [obj_v1],
+ return_stdout=True)
+ self.assertRegexpMatches(json_text, test_regex)
+
+ json_text = self.RunGsUtil(self._get_acl_prefix + [obj_v2],
+ return_stdout=True)
+ self.assertNotRegexpMatches(json_text, test_regex)
+
+ def testBadRequestAclChange(self):
+ stdout, stderr = self.RunGsUtil(
+ self._ch_acl_prefix +
+ ['-u', 'invalid_$$@hello.com:R', suri(self.sample_uri)],
+ return_stdout=True, return_stderr=True, expected_status=1)
+ self.assertIn('BadRequestException', stderr)
+ self.assertNotIn('Retrying', stdout)
+ self.assertNotIn('Retrying', stderr)
+
+ def testAclGetWithoutFullControl(self):
+ object_uri = self.CreateObject(contents='foo')
+ with self.SetAnonymousBotoCreds():
+ stderr = self.RunGsUtil(self._get_acl_prefix + [suri(object_uri)],
+ return_stderr=True, expected_status=1)
+ self.assertIn('AccessDeniedException', stderr)
+
+ def testTooFewArgumentsFails(self):
+ """Tests calling ACL commands with insufficient number of arguments."""
+ # No arguments for get, but valid subcommand.
+ stderr = self.RunGsUtil(self._get_acl_prefix, return_stderr=True,
+ expected_status=1)
+ self.assertIn('command requires at least', stderr)
+
+ # No arguments for set, but valid subcommand.
+ stderr = self.RunGsUtil(self._set_acl_prefix, return_stderr=True,
+ expected_status=1)
+ self.assertIn('command requires at least', stderr)
+
+ # No arguments for ch, but valid subcommand.
+ stderr = self.RunGsUtil(self._ch_acl_prefix, return_stderr=True,
+ expected_status=1)
+ self.assertIn('command requires at least', stderr)
+
+ # Neither arguments nor subcommand.
+ stderr = self.RunGsUtil(['acl'], return_stderr=True, expected_status=1)
+ self.assertIn('command requires at least', stderr)
+
+ def testMinusF(self):
+ """Tests -f option to continue after failure."""
+ bucket_uri = self.CreateBucket()
+ obj_uri = suri(self.CreateObject(bucket_uri=bucket_uri, object_name='foo',
+ contents='foo'))
+ acl_string = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+ self.RunGsUtil(self._set_acl_prefix +
+ ['-f', 'public-read', suri(bucket_uri) + 'foo2', obj_uri],
+ expected_status=1)
+ acl_string2 = self.RunGsUtil(self._get_acl_prefix + [obj_uri],
+ return_stdout=True)
+
+ self.assertNotEqual(acl_string, acl_string2)
+
+
+class TestS3CompatibleAcl(TestAclBase):
+ """ACL integration tests that work for s3 and gs URLs."""
+
+ def testAclObjectGetSet(self):
+ bucket_uri = self.CreateBucket()
+ obj_uri = self.CreateObject(bucket_uri=bucket_uri, contents='foo')
+ self.AssertNObjectsInBucket(bucket_uri, 1)
+
+ stdout = self.RunGsUtil(self._get_acl_prefix + [suri(obj_uri)],
+ return_stdout=True)
+ set_contents = self.CreateTempFile(contents=stdout)
+ self.RunGsUtil(self._set_acl_prefix + [set_contents, suri(obj_uri)])
+
+ def testAclBucketGetSet(self):
+ bucket_uri = self.CreateBucket()
+ stdout = self.RunGsUtil(self._get_acl_prefix + [suri(bucket_uri)],
+ return_stdout=True)
+ set_contents = self.CreateTempFile(contents=stdout)
+ self.RunGsUtil(self._set_acl_prefix + [set_contents, suri(bucket_uri)])
+
+
+@SkipForGS('S3 ACLs accept XML and should not cause an XML warning.')
+class TestS3OnlyAcl(TestAclBase):
+ """ACL integration tests that work only for s3 URLs."""
+
+ # TODO: Format all test case names consistently.
+ def test_set_xml_acl(self):
+ """Ensures XML content does not return an XML warning for S3."""
+ obj_uri = suri(self.CreateObject(contents='foo'))
+ inpath = self.CreateTempFile(contents='<ValidXml></ValidXml>')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, obj_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('BadRequestException', stderr)
+ self.assertNotIn('XML ACL data provided', stderr)
+
+ def test_set_xml_acl_bucket(self):
+ """Ensures XML content does not return an XML warning for S3."""
+ bucket_uri = suri(self.CreateBucket())
+ inpath = self.CreateTempFile(contents='<ValidXml></ValidXml>')
+ stderr = self.RunGsUtil(self._set_acl_prefix + [inpath, bucket_uri],
+ return_stderr=True, expected_status=1)
+ self.assertIn('BadRequestException', stderr)
+ self.assertNotIn('XML ACL data provided', stderr)
+
+
+class TestAclOldAlias(TestAcl):
+ _set_acl_prefix = ['setacl']
+ _get_acl_prefix = ['getacl']
+ _set_defacl_prefix = ['setdefacl']
+ _ch_acl_prefix = ['chacl']
« no previous file with comments | « third_party/gsutil/gslib/tests/test_Doption.py ('k') | third_party/gsutil/gslib/tests/test_bucketconfig.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698