分析：如何多线程运行测试用例

这是时常被问到的问题，尤其是UI自动化的运行，过程非常耗时，所以，所以多线程不失为一种首先想到的解决方案。

多线程是针对的测试用例，所以和selenium没有直接关系，我们要关心的是单元测试框架。

unittest

首先，应该说明的是unittest本身是不支持多线程的。当然，如果你学过Python的threading模块，也未必不行。不过我在stackoverflow 找了半天，大多是介绍unittest 测试多线程模块，并非是unittest本身如何多线程运行用例。

“我如何学习葵花宝典” 和 “我如何验证 张三 学会了葵花宝典”是两回事，而我显然要解决的问题是前者。

又重新百度，结果就找了答案。核心借助 tomorrow3 的测试库，再配合HTMLTestRunner 生成测试报告。

GitHub - SeldomQA/XTestRunner: Modern style test report based on unittest framework.
GitHub - dflupu/tomorrow3: An implementation of the tomorrow package for python >= 3.0

测试用例

测试用例如下，你可以复制多份测试。

# test_a.py
import time
import unittest
from selenium import webdriver


class Test1(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()

    def test_01(self):
        self.driver.get("https://www.bing.com/?mkt=zh-CN")
        elem = self.driver.find_element_by_id("sb_form_q")
        elem.send_keys("多线程")
        elem.submit()
        time.sleep(2)
        self.assertIn("多线程", self.driver.title)

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()


if __name__ == "__main__":
    unittest.main()

运行文件

核心在tomorrow3提供的threads装饰器，用于装饰测试运行方法。

import unittest
import os
from TestRunner import HTMLTestRunner
from tomorrow3 import threads

# 定义目录
BASE_DIR = os.path.dirname(os.path.realpath(__file__))
TEST_DIR = os.path.join(BASE_DIR, "test_dir")
REPORT_DIR = os.path.join(BASE_DIR, "test_report")


def test_suits():
    """
    加载所有的测试用例
    """
    discover = unittest.defaultTestLoader.discover(
        TEST_DIR,
        pattern="test_*.py"
    )
    return discover


@threads(2)  # !!!核心!!!! 设置线程数
def run_case(all_case, nth=0):
    """
    执行所有的用例, 并把结果写入测试报告
    """
    report_abspath = os.path.join(REPORT_DIR, f"result{nth}.html")
    with open(report_abspath, "wb+") as file:
        runner = HTMLTestRunner(stream=file, title='多线程测试报告')
        # 调用test_suits函数返回值
        runner.run(all_case)


if __name__ == "__main__":
    cases = test_suits()
    # 循环启动线程
    for i, j in zip(cases, range(len(list(cases)))):
        run_case(i, nth=j)  # 执行用例，生成报告

总结：

确实可以（根据设置的线程数）同时开启两个浏览器运行，同时带来了两个问题。

程序会根据线程数，生成多份测试报告，每个测试报告统计当前线程所运行的测试结果。
如果去掉nth参数，设置为一个报告，那么第二个线程运行的结果不能正确展示的一张报告上。
至少每个线程中的用例要保证有浏览器的启动/关闭。

第3点，你可能不理解，我举个例子，小时候见过妈妈/奶奶缝衣服。假设一件衣服拿一根针来缝制，缝衣服最麻烦就是穿针引线，针眼很小，每次都要费半天功夫，小孩子眼神好，我时常被叫去穿针引线。那为了节约时间怎么办，我当然是把线弄的长长的，最好是一根针线可以缝制多件衣服。

在Selenium 中，定义的浏览器驱动driver，每一条用例都定义一次驱动，伴随而来的就每个用例都开启/关闭一次浏览器，这当然是非常耗时的，为了缩短这个时间，我们最好是启动一次浏览器把所有用例都跑完才关闭。

那么问题来了，当我们使用多线程之后，相当于多个人同时缝制衣服，共用一根针线肯定不行啊。至少需要每人一根针线吧！而且缝制衣服的人是变化的，为了赶时间就多两个人（多开两个线程），用例比较少就少几个人（少启两个线程），为了适应这种变化，那么最好一件衣配置一根针线，每件衣服是最小单位，一个人必须要独立的把一件衣服缝好。

为了使用多线程，我们就必须每条用例开启/关闭一次浏览器，这其实是另一种时间的浪费，为了弥补这里的浪费，你必须把线程开得足够多才行。

最后，你必须再配置一位统计员，去统计每个人缝制衣服的数量，合计到一个报告中。

pytest

多线程运行用例在 pytest 中就相对容易太多了，pytest-xdist 插件就可以完成这件事情。

GitHub - pytest-dev/pytest-xdist: pytest plugin for distributed testing and loop-on-failures testing modes.

测试用例

测试用例如下，你可以复制多份测试。

# test_a.py
import time
from selenium import webdriver


def test_01():
    driver = webdriver.Chrome()
    driver.get("http://www.baidu.com")
    elem = driver.find_element_by_id("kw")
    elem.send_keys("unittest")
    elem.submit()
    time.sleep(2)
    assert driver.title == "unittest_百度搜索"
    driver.quit()


def test_02():
    driver = webdriver.Chrome()
    driver.get("http://www.baidu.com")
    elem = driver.find_element_by_id("kw")
    elem.send_keys("python")
    elem.submit()
    time.sleep(2)
    assert driver.title == "python_百度搜索"
    driver.quit()

运行文件

核心就是安装 pytest-xdist 插件，通过-n参数设置线程数即可。

import os
import pytest

# 定义目录
BASE_DIR = os.path.dirname(os.path.realpath(__file__))
TEST_DIR = os.path.join(BASE_DIR, "test_dir")
REPORT_DIR = os.path.join(BASE_DIR, "test_report")


if __name__ == '__main__':
    report_file = os.path.join(REPORT_DIR, "result.html")
    pytest.main([
        "-n", "2",  # ！！核心！！！设置2个线程
        "-v", "-s", TEST_DIR,
        "--html=" + report_file,
        ])

总结：

相对于unittest，在pytest实现多线程非常简单，最终线线程的测试结果也可以很好在一个测试报告中展示。