python selenium执行所有测试用例并生成报告

直接上代码。

# -*- coding: utf-8 -*-
import time
import os
import os.path
import re
import unittest
import HTMLTestRunner
import shutil
shutil.copyfile("setting.ini","../setting.ini")
casepaths = []
def createsuite(casepath):
testunit = unittest.TestSuite()
#discover方法定义
discover = unittest.defaultTestLoader.discover(
casepath,
pattern = 'case*.py',
top_level_dir= casepath
)
for test_suite in discover:
for test_case in test_suite:
testunit.addTest(test_case)
print testunit
return testunit
for parent,dirnames,filenames in os.walk('.'):

for filename in filenames:
#print "parent is:" + parent
#print "filename is:" + filename
path=os.path.join(parent,filename)
#正则判断是否为测试用例
match = re.match('case', filename)
if match:
print u"获取测试用例目录:%s"%parent
casepaths.append(parent)
break

#定义报告存放目录,支持相对路径
now = time.strftime("%Y-%m-%M-%H-%M-%S",time.localtime(time.time()))
filename = now+'report.html'
fp = file(filename,'wb')
runner = HTMLTestRunner.HTMLTestRunner(
stream = fp,
title = u'自动化测试报告',
description = u'用例执行情况'
)

for casepath in casepaths:
print u"正在执行 %s目录下的测试用例"%casepath
alltestnames = createsuite(casepath)
runner.run(alltestnames)
print u"执行 %s目录下的测试用例完成"%casepath
print u"完成所有测试用例执行任务"