mirror of
https://github.com/apache/impala.git
synced 2025-12-31 06:02:51 -05:00
99c37aac37
their parent's permissions This patch adds --insert_inherit_permissions. If true, all new partition directories created by INSERT will inherit their permissions from their parent. When false, the directories are created with the default permissions. Change-Id: Ib2b4c251e51ea5048387169678e8dde34ecfe5f6 Reviewed-on: http://gerrit.ent.cloudera.com:8080/1917 Tested-by: jenkins Reviewed-by: Henry Robinson <henry@cloudera.com>
106 lines
5.2 KiB
Python
Executable File
106 lines
5.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright (c) 2012 Cloudera, 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.
|
|
#
|
|
|
|
import pytest
|
|
from tests.common.custom_cluster_test_suite import CustomClusterTestSuite
|
|
from tests.util.hdfs_util import HdfsConfig, get_hdfs_client, get_hdfs_client_from_conf
|
|
|
|
class TestInsertBehaviourCustomCluster(CustomClusterTestSuite):
|
|
def check_partition_perms(self, part, perms):
|
|
ls = self.hdfs_client.get_file_dir_status(
|
|
"test-warehouse/functional.db/insert_inherit_permission/%s" % part)
|
|
assert ls['FileStatus']['permission'] == perms
|
|
|
|
@classmethod
|
|
def setup_class(cls):
|
|
if pytest.config.option.namenode_http_address is None:
|
|
hdfs_conf = HdfsConfig(pytest.config.option.minicluster_xml_conf)
|
|
cls.hdfs_client = get_hdfs_client_from_conf(hdfs_conf)
|
|
else:
|
|
host, port = pytest.config.option.namenode_http_address.split(":")
|
|
cls.hdfs_client = get_hdfs_client()
|
|
|
|
@pytest.mark.execute_serially
|
|
@CustomClusterTestSuite.with_args("--insert_inherit_permissions=true")
|
|
def test_insert_inherit_permission(self):
|
|
"""Create a table with three partition columns to test permission inheritance"""
|
|
impalad = self.cluster.get_any_impalad()
|
|
client = impalad.service.create_beeswax_client()
|
|
|
|
self.execute_query_expect_success(client, "DROP TABLE IF EXISTS"
|
|
" functional.insert_inherit_permission")
|
|
self.execute_query_expect_success(client, "CREATE TABLE "
|
|
"functional.insert_inherit_permission (col int)"
|
|
" PARTITIONED BY (p1 int, p2 int, p3 int)")
|
|
|
|
self.execute_query_expect_success(client, "ALTER TABLE "
|
|
"functional.insert_inherit_permission ADD PARTITION"
|
|
"(p1=1, p2=1, p3=1)")
|
|
self.hdfs_client.chmod(
|
|
"test-warehouse/functional.db/insert_inherit_permission/p1=1/", "777")
|
|
|
|
# 1. INSERT that creates two new directories gets permissions from parent
|
|
self.execute_query_expect_success(client, "INSERT INTO "
|
|
"functional.insert_inherit_permission "
|
|
"PARTITION(p1=1, p2=2, p3=2) VALUES(1)")
|
|
self.check_partition_perms("p1=1/p2=2/", "777")
|
|
self.check_partition_perms("p1=1/p2=2/p3=2/", "777")
|
|
|
|
# 2. INSERT that creates one new directory gets permissions from parent
|
|
self.execute_query_expect_success(client, "INSERT INTO "
|
|
"functional.insert_inherit_permission "
|
|
"PARTITION(p1=1, p2=2, p3=3) VALUES(1)")
|
|
self.check_partition_perms("p1=1/p2=2/p3=3/", "777")
|
|
|
|
# 3. INSERT that creates no new directories keeps standard permissions
|
|
self.hdfs_client.chmod(
|
|
"test-warehouse/functional.db/insert_inherit_permission/p1=1/p2=2", "644")
|
|
self.execute_query_expect_success(client, "INSERT INTO "
|
|
"functional.insert_inherit_permission "
|
|
"PARTITION(p1=1, p2=2, p3=3) VALUES(1)")
|
|
self.check_partition_perms("p1=1/p2=2/", "644")
|
|
self.check_partition_perms("p1=1/p2=2/p3=3/", "777")
|
|
|
|
|
|
@pytest.mark.execute_serially
|
|
@CustomClusterTestSuite.with_args("--insert_inherit_permissions=false")
|
|
def test_insert_inherit_permission_disabled(self):
|
|
"""Check that turning off insert permission inheritance works correctly."""
|
|
impalad = self.cluster.get_any_impalad()
|
|
client = impalad.service.create_beeswax_client()
|
|
|
|
self.execute_query_expect_success(client, "DROP TABLE IF EXISTS"
|
|
" functional.insert_inherit_permission")
|
|
self.execute_query_expect_success(client, "CREATE TABLE "
|
|
"functional.insert_inherit_permission (col int)"
|
|
" PARTITIONED BY (p1 int, p2 int, p3 int)")
|
|
|
|
self.execute_query_expect_success(client, "ALTER TABLE "
|
|
"functional.insert_inherit_permission ADD PARTITION"
|
|
"(p1=1, p2=1, p3=1)")
|
|
ls = self.hdfs_client.get_file_dir_status(
|
|
"test-warehouse/functional.db/insert_inherit_permission/p1=1/")
|
|
default_perms = ls['FileStatus']['permission']
|
|
|
|
self.hdfs_client.chmod(
|
|
"test-warehouse/functional.db/insert_inherit_permission/p1=1/", "777")
|
|
|
|
self.execute_query_expect_success(client, "INSERT INTO "
|
|
"functional.insert_inherit_permission "
|
|
"PARTITION(p1=1, p2=3, p3=4) VALUES(1)")
|
|
# Would be 777 if inheritance was enabled
|
|
self.check_partition_perms("p1=1/p2=3/", default_perms)
|