python自动测试e_python实现hive自动化测试

本程序主要实现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