本程序主要实现hive权限测试。系统中有管理员用户single和测试用户test。在路径/home/test/下,将用例和预期结果写在xml文件中。
执
行过程:kinit single用户,beeline -u -e登录并执行对test用户对应角色回收和赋予权限的语句;kinit
test用户,beeline -u
-e去执行测试语句并保存执行结果到tmp文件中;在tmp文件中查找预期关键字,得出该测试pass还是fail的结果,统计测试结果。
#!/usr/bin/python
#coding:utf-8
#by cvv54
import sys
import os
import re
try:
import xml.etree.cElementTree as ET
except ImportError:
import xml.etree.ElementTree as ET
try:
tree = ET.parse("/home/test/case/usecase.xml")
#root = ET.fromstring(country_string)
root = tree.getroot()
except Exception,e:
print "Error:cannot parse file:usecase.xml."
sys.exit(1)
print root.tag,"---",root.attrib
for child in root:
print child.tag,"---",child.attrib
passed=0
failed=0
for case in root.findall('case'):
pre = case.find('pre').text
perform = case.find('perform').text
expect = case.find('expect').text.strip('\n')
id = case.get("id")
print "key words are :"
print expect
for each in pre.split(';'):
# print each
if not each.strip()=='':
command = each.strip('\n')
os.environ['command']=str(command)
print command
os.system("kdestroy")
os.system("kinit -kt /etc/security/keytabs/single.keytab single")
os.system('beeline -u "jdbc:hive2://gateway.xxx.xxx:10000/;principal=single" -e "$command;" &>>log')
for each in perform.split(';'):
# print each
if not each.strip()=='':
command = each.strip('\n')
os.environ['command']=str(command)
print command
os.system("kdestroy")
os.system("kinit -kt /etc/security/keytabs/test.keytab test")
os.system('beeline -u "jdbc:hive2://gateway.xxx.xxx:10000/;principal=single" -e "$command;" &>tmp')
f=open('tmp')
flag=0
for line in f:
# print "line is :"
# print line
match=re.findall(expect,line)
if match != []:
passed+=1
flag=1
if flag == 0:
failed+=1
print(id)
os.system("cat tmp>>log")
os.system("rm -f tmp")
print "passed:"
print passed
print "failed:"
print failed
os.system('mv log `date "+%Y-%m-%d~%H-%M-%S"`')
我的xml是这样写的:<?xml version="1.0" encoding="utf-8"?>
REVOKE ALL ON SERVER server1 FROM ROLE test_role; GRANT ALL ON URI TO ROLE test_role;
SHOW TABLES;
tab_name
REVOKE ALL ON SERVER server1 FROM ROLE test_role;
SHOW TABLES;
FAILED: SemanticException No valid privileges
CREATE TABLE IF NOT EXISTS test_table1 (id INT,name STRING,salary FLOAT,street STRING,city STRING,state STRING,zip INT) PARTITIONED BY (address STRING) row format delimited fields terminated by ' ';
REVOKE ALL ON SERVER server1 FROM ROLE test_role;GRANT SELECT ON TABLE test_table1 TO ROLE test_role;
SHOW CREATE TABLE test_table1;
createtab_stmt
REVOKE ALL ON SERVER server1 FROM ROLE test_role;GRANT INSERT ON TABLE test_table1 TO ROLE test_role;
SHOW CREATE TABLE test_table1;
createtab_stmt
REVOKE ALL ON SERVER server1 FROM ROLE test_role;
SHOW CREATE TABLE test_table1;
FAILED: SemanticException No valid privileges