unittestreport之unittest 用例失败重运行机制

前言

很多小伙伴一直在诟病 unittest，说 unittest 相对 pytest 来说太鸡肋了，pytest 中提供了很多高级功能 unittest 中都没有。

在这里还是想为 unittest 打抱不平一下，unittest 是由 python 官方维护的官方库，官方库都是比较轻量级的，本身只实现核心的一些功能，肯定不会像 pytest 这些第三库一样根据需求，去开发一些定制的功能。

其实 unittest 也有自己的优势，学习成本低，使用起来简单也易上手。目前还是有很多测试的小伙伴在使用的，于是前段时间小编想了想，决定在 unittest 的基础上扩展开发第三方库 unittestreport，为 unittest 开发一些常用的扩展功能，来弥补 unittest 的不足。

目前准备实现的功能有：多种风格 HTML 测试报告生成、多线程运行、用例失败重跑、数据驱动、命令行启动执行测试等功能。

unittestreport 的安装：pip install unittestreport

前段时间实现了第一个版本，unittest 生成 HTML 测试报告，以及 unittest 用例多线程运行机制，最近这几天又优化了 unittestreport 的一个版本（1.0.9 版本），实现了 unittest 用例失败重运行的机制，那么接下来就给大家介绍一下如何使用 unittest 用例失败重运行机制。

一、unittest 用例重运行

关于 unittest 重运行机制，unittestreport 中提供了两种方式，第一种选择单个测试用例实现重运行，可以自己去标记失败需要重运行的测试用例，第二种是针对运行的所有的测试用例，只要运行失败的用例就会重运行改用例。那么接下来分别给大家介绍这两个功能的使用。

单个用例重运行

如果像标记单个测试用例失败重运行，可以直接使用 unittestreport 中的 rerun 来标记测试用例，rerun 接收两个参数 count，和 interval。

count：用来指定用例失败重运行的次数

interval：指定每次重运行的时间间隔

下面有三个测试用例，其中有一个 test_case_01 使用了 rerun 进行了标记，设置的失败重运行次数为 4 次，每次间隔的时间 2 秒。

import unittest

from unittestreport import rerun

class TestClass(unittest.TestCase):

  @rerun(count=4, interval=2)

  def test_case_01(self):

      a = 100

      b = 99

      assert a == b

  def test_case_02(self):

      a = 100

      b = 100

      assert a == b

  def test_case_03(self):

      a = 100

      b = 101

      assert a == b

运行上述用例

import unittest

from unittestreport import TestRunner

from testcase import TestClass  # 导入上面写的测试用例类

# 加载测试套件

suite = unittest.defaultTestLoader.loadTestsFromTestCase(TestClass)

# 执行测试用例

runner = TestRunner(suite=suite)

runner.run()

生成测试报告如下图，可以看到第一条测试用例运行了三次：

二、全部用例失败重跑机制

关于所有的测试用例失败重跑，unittestreport 中提供了一个更为简单的使用入口，直接使用 unittestreport 中 TestRunner 对象的 rerun_run 方法即可实现所有的用例失败重运行，rerun_run 同样有两个参数，count 和 interval。

count：用来指定用例失败重运行的次数

interval：指定每次重运行的时间间隔

测试用例如下：

import unittest

from unittestreport import rerun

class TestClass(unittest.TestCase):

  def test_case_01(self):

      a = 100

      b = 99

      assert a == b

  def test_case_02(self):

      a = 100

      b = 100

    assert a == b

使用 unittestreport 的重运行机制，运行上述用例

import unittest

from unittestreport import TestRunner

from testcase import TestClass  # 导入测试用例类

# 测试套件

suite = unittest.defaultTestLoader.loadTestsFromTestCase(TestClass)

# 创建执行对象

runner = TestRunner(suite=suite)

# 执行测试用例，失败重运行设置为3次，间隔时间为2秒

runner.rerun_run(count=3, interval=2)

执行测试报告如下图，可以看到两个失败的用例，都重复运行了三次：

关于 unittestreport 重运行用例，就给大家介绍到这里。

本文作者：柠檬班软件测试（lemonban）——专注于最新最前沿的软件测试技术，解决你的测试技术烦恼，对软件测试感兴趣的朋友赶快关注我们吧！