SeleniumPython多线程

Posted 2020-10-22 素人渔芙2017

 # all_tests_pro.py

 

import unittest,time,os,multiprocessing
import HTMLTestRunner

#查找多有含有thread的文件，文件夹
def EEEcreatsuit():
    casedir = []
    listaa = os.listdir(r\'C:\\Users\\wangxue1\\PycharmProjects\\selenium2TestOne\')
    for xx in listaa:
        if \'thread\' in listaa:
            casedir.append(xx)
    suite = []
    for n in casedir:
        testunit = unittest.TestSuite()
        discover = unittest.defaultTestLoader.discover(n,pattern=\'test*.py\',top_level_dir=n)
        print(discover)
        for test_suite in discover:
            for test_case in test_suite:
                testunit.addTests(test_case)
        suite.append(testunit)
    print(\'================casedir:==========\',casedir)
    print(\'+++++++++++++++++++++++++++++++++++\')
    print(\'!!!!!suite:!!!!!\',suite)
    return suite,casedir

#多进程运行测试套件，将结果写入HTMLTestRunner报告
def EEEEmultiRunCase(suite,casedir):
    now = time.strftime(\'%Y-%m-%d-%H_%M_%S\',time.localtime(time.time()))
    testreport = \'C:\\\\Users\\\\wangxue1\\\\PycharmProjects\\\\selenium2TestOne\\\\\' + now + \'\\\\\'
    filename = testreport + \'report.html\'
    if not os.path.exists(testreport):
        os.makedirs(testreport)
    else:
        pass
    proclist = []
    s = 0
    with open(filename,\'wb\') as report:
        for i in suite:
            runner = HTMLTestRunner.HTMLTestRunner(stream=report,title=u\'测试报告\',description=u\'用例执行情况：\')
            proc = multiprocessing.Process(target=runner.run(i),args=(i,))
            proclist.append(proc)
            s = s + 1
        for proc in proclist: proc.start()
        for proc in proclist: proc.join()

runtmp = EEEcreatsuit()
EEEEmultiRunCase(runtmp[0],runtmp[1])