Selenium实战（四）——unittest单元测试框架1

Python中的单元测试框架包含：doctest、unittest、pyttest、nose等，使用unittest单元测试框架不需要自行定义断言失败的提示，并且当一个测试函数执行失败后，后面的测试函数还会继续执行下去。下面以一个简单计算器来编写自动化测试用例。被测试的代码如下：

 # 计算器类
 class Calculator:
     def __init__(self, a, b):
         self.a = int(a)
         self.b = int(b)

     def add(self):
         return self.a + self.b

     def sub(self):
         return self.a - self.b

     def mul(self):
         return self.a * self.b

     def div(self):
         return self.a / self.b

通过unittest单元测试框架编写如下测试用例：

 import unittest

 from files.unit_test.calculator import Calculator

 class TestCalculator(unittest.TestCase):
     def test_add(self):
         c = Calculator(3, 5)
         result = c.add()
         self.assertEqual(result, 8)

     def test_sub(self):
         c = Calculator(10, 5)
         result = c.sub()
         self.assertEqual(result, 5)

     def test_mul(self):
         c = Calculator(5, 7)
         result = c.mul()
         self.assertEqual(result, 34)

     def test_div(self):
         c = Calculator(12, 6)
         result = c.div()
         self.assertEqual(result, 2)

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

执行结果展示：

一、使用unittest单元测试框架的规则

1.创建一个测试类（TestCalculator）,该类必须继承unittest模块的TestCase类；

2.创建的测试方法必须以test开头。

通过unittest提供的assertEqual()方法来断言结果与预期是否一致，该方法由TestCase父类提供，由于继承了该类，可用self调用。关于测试结果：

（1）“.”表示一条运行通过的测试用例。

（2）“F”表示一条运行失败的测试用例。

（3）“E”表示一条运行错误的测试用例。

（4）“s”表示一条运行跳过的测试用例。

二、重要的概念

1.Test Case

　　最小的测试单元。用于检查特定输入集合的特定返回值。unittest提供了TestCase基类，脚本中创建的测试类需要继承该基类，用来创建新的测试用例。

2.Test Suit

　　测试套件是测试用例、测试套件或两者的集合，用于组装一组要运行的测试。

3.Test Runner

　　是一个组件，用于协调测试的执行并向用户提供结果。可以使用图形界面、文本界面或返回特殊值来展示执行测试的结果。unittest提供了TextTestRunner类运行测试用例，为了生成HTML格式的测试报告，会选择使用HTMLTestRunner运行类。

4.Test Fixture

　　代表执行一个或多个测试所需的环境准备，以及关联的清理动作。例如，创建临时或者代理数据库、目录，或启动服务器进程。unittest中提供了setUp()/tearDown()、setUpClass()/tearDownClass()等方法来完成这些操作。

基于这四个概念的理解，修改以上测试用例如下：

 import unittest

 from files.unit_test.calculator import Calculator

 class TestCalculator(unittest.TestCase):

     # 测试用例前置动作
     def setUp(self):
         print("test start")

     # 测试用例后置动作
     def tearDown(self):
         print("test end")

     def test_add(self):
         c = Calculator(3, 5)
         result = c.add()
         self.assertEqual(result, 8)

     def test_sub(self):
         c = Calculator(10, 5)
         result = c.sub()
         self.assertEqual(result, 5)

     def test_mul(self):
         c = Calculator(5, 7)
         result = c.mul()
         self.assertEqual(result, 34)

     def test_div(self):
         c = Calculator(12, 6)
         result = c.div()
         self.assertEqual(result, 2)

 if __name__ == '__main__':
     # 创建测试套件
     suit = unittest.TestSuite()
     suit.addTest(TestCalculator("test_add"))
     suit.addTest(TestCalculator("test_sub"))
     suit.addTest(TestCalculator("test_mul"))
     suit.addTest(TestCalculator("test_div"))
     # 创建测试运行器
     runner = unittest.TestRunner()
     runner.run(suit)

　　在测试用例的执行时，抛弃了unittest提供的main()方法，而是调用了TestSuite类下面的addTest()来添加测试用例。因为一次只能添加一条用例，所以需要指定测试类及测试方法。然后调用TextTestRunner类下面的run()运行测试套件。这样做比直接用main()麻烦，首先，测试用例的执行顺序可以由测试套件的添加顺序控制，而main()方法只能按照测试类、方法的名称来执行测试用例。例如，TestA类比TestB类限制性，test_add()用例比test_div()用例限制性；其次，当一个测试文件中有很多测试用例时，并不是每次都要执行所有的测试用例，尤其是比较耗时的UI自动化测试。因而通过测试套件和测试运行器可以灵活的控制要执行的测试用例。