unittest 运行slenium（五）---运行代码并生成HTMLTestRunner报告

整体代码如下：

import os
import sys
import time
import datetime
import unittest
import HTMLTestRunner
 
# git config --global user.name “用户名”
 
# git config --global user.email “邮箱”
 
# 获取项目路径下的目录
path_project = os.path.split(os.path.abspath(__file__))[0]
os.chdir(path_project)
 
# 将项目路径保存
sys.path.append(path_project)
 
# 获取当前文件所在目录
CUR_PATH = os.path.dirname(os.path.realpath(__file__))
 
# 设置时间格式
day_now = time.strftime("%Y-%m-%d")
time_now = time.strftime("%H-%M-%S")
 
def add_case(case_name='Case', rule='test*.py'):
    # 加载所有的用例
    case_path = case_name
    # case_path = os.path.join(CUR_PATH,case_name)
    # 定义方法的参数
    discover = unittest.defaultTestLoader.discover(case_path,
                                                   pattern=rule,
                                                   top_level_dir=CUR_PATH)
    print("用例所在位置 %s  执行的文件 %s " % (case_path, rule))
    return discover
 
def run_case(all_case, reportName='report', filename="filename"):
    # 1.1报告存放路径
    report_path = os.path.join(os.getcwd(), reportName)
 
    # 1.2报告根据时间来分类
    report_path = os.path.join(report_path, day_now)
    if not os.path.exists(report_path): os.mkdir(report_path)
 
    filename = "%s-%s.html" % (time_now, filename)
    report_abspath = os.path.join(report_path, filename)
 
    print("新创建报告文件所在位置 %s " % report_abspath)
 
    fp = open(report_abspath, "wb")
    runner = HTMLTestRunner.HTMLTestRunner(stream=fp,
                                           title=u'自动化测试报告测试结果如下：',
                                           description=u'用例执行情况：')
 
    # 调用add_case函数返回值
    runner.run(all_case)
    fp.close()
 
def get_report_file(report_path):
    # 获取最新的测试报告
    lists = os.listdir(report_path)
    lists.sort(key=lambda fn: os.path.getmtime(os.path.join(report_path, fn)))
    print("最新报告名 %s" % lists[-1])
 
    # 找到最新生成的报告文件
    report_file = os.path.join(report_path, lists[-1])
    return report_file
 
def run_main(file_name, rule='test*.py'):
    # 加载用例
    all_case = add_case(file_name, rule)
 
    # 运行文件所在目录为文件名
    file_name = str.split(file_name, '\\')[-1]
 
    # 生成测试报告的路径
    run_case(all_case, filename=file_name)
 
    # 获取最新测试报告所在的文件路径
 
    report_path = os.path.join("report", day_now)  # 最新报告文件夹在项目中的路径
    report_path = os.path.join(CUR_PATH, report_path)  # 拼接项目
 
    # 获取最新的测试报告
    report_file = get_report_file(report_path)
    print("最新报告路径为{0}".format(report_file))
 
def list_dir(file_dir):
    '''
        通过 listdir 得到的是仅当前路径下的文件名，不包括子目录中的文件，如果需要得到所有文件需要递归
    '''
    # print('\n\n<><><><><><> listdir <><><><><><>')
    # print("current dir : {0}".format(file_dir))
    file_dir = os.path.join(CUR_PATH, file_dir)
    dir_list = os.listdir(file_dir)
    file_name = []
    for cur_file in dir_list:
        # 获取文件的绝对路径
        path = os.path.join(file_dir, cur_file)
        if os.path.isfile(path):  # 判断是否是文件还是目录需要用绝对路径
            # print("{0} : is file!".format(cur_file))
            continue
        if os.path.isdir(path):
            # print("{0} : is dir!".format(cur_file))
            list_dir(path)  # 递归子目录
            file_name.append(cur_file)
    return file_name
 
def start_loading_case():
    file_name = r'case'
    all_dir = list_dir(file_name)
 
    for dir in range(len(all_dir)):
        if "__pycache__" in all_dir[dir]:
            continue
        else:
            dir = os.path.join(file_name, all_dir[dir])
            print("需要运行的文件: %s" % dir)
            run_main(dir)
 
def start_single_loading():
    run_main('demo', "test_*.py")
 
if __name__ == '__main__':
    # 记录运行时间
    startTime = datetime.datetime.now()
    print("\nStart Run Time: %s\n" % startTime)
    # 指定case所在位置并自动运行位置下需要运行的test文件夹,并且每个文件夹都自动生成一个相应的html报告
    start_loading_case()
    # 指定需要运行的test文件夹，运行完毕之后只生成一个相应的html报告
    # start_single_loading()
    # 记录结束时间
    stopTime = datetime.datetime.now()
    print("\nFinish Run Time: %s\n" % stopTime)
    # 打印一共运行的时间
    print('\nRunning time: %s\n' % (stopTime - startTime), file=sys.stderr)

项目所在位置：

https://github.com/namexiaohuihui/demotest

个人记录而已，不理解的可以留言。请勿直接喷