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"完成所有测试用例执行任务"