您的当前位置:首页正文

unittest小结

来源:花图问答

unittest单元测试框架, 是十分常用的自动化测试框架之一, 当然它并非只能做单元测试

unitttest基本架构

# 设置编码, utf-8支持中英文
# coding: utf-8 

# 导入使用unittest必备的包
import unittest

# 创建unittest 测试类
# unittest测试类, 均继承自unittest.TestCase
class Test(unittest.TestCase): 

    # 定义 `setUp()`方法, 进行初始化操作
    def setUp(self):
      print('Test start')

    # 定义'tearDown()'方法, 用于测试用例执行后的善后工作
    def tearDown(self):
      print('Test over')

    # 编写测试用例
    # 测试用例必须以'test'开头
    def test_1(self):
      print('test case 1')
      # 断言测试是否通过
      self.assertEqual(a, b)

  def test_2(self):
        print('test case 2')
        # 断言测试是否通过
        self.assertNotEqual(c, d)

# 运行测试用例
if __name__ == '__main__':
  unittest.main()

unittest 运行用例的三种方式

unittest为使用者提供了三种运行方式, 基本满足了大部分的运行需求, 但依然存在无法解决的问题
如: 使用unittest同时运行两个用例文件, 仅启动一次浏览器来跑完两个文件的全部case, 是无法实现的,想要实现这个功能就要使用pytest
  1. unittest.main()
    使用它可以将一个单元测试模块变为可直接运行的测试脚本,main()方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法,并自动执行。执行方法的默认顺序是:根据ASCII码的顺序加载测试用例,数字与字母的顺序为:0-9,A-Z,a-z。
  2. 构造测试集
# 实例化测试套件
suite=unittest.TestSuite()
# 将测试用例加载到测试套件中。
# 执行顺序是按添加顺序:先执行test_2,再执行test_1
suite.addTest(Test('test_2'))
suite.addTest(Test('test_1'))
# 执行测试用例
# 实例化TextTestRunner类
runner=unittest.TextTestRunner()
# 使用run()方法运行测试套件(即运行测试套件中的所有用例)
    runner.run(suite)
  1. 使用discover()执行
# 构造测试集
test_dir = './'   # 测试用例所在目录
discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')  
# 执行测试用例
# 实例化TextTestRunner类
runner=unittest.TextTestRunner()
# 使用run()方法运行测试套件(即运行测试套件中的所有用例)
runner.run(discover)  

unittest中主要使用的断言方法:

assertEqual(a,b,[msg='测试失败时打印的信息']): 断言a和b是否相等,相等则测试用例通过。

assertNotEqual(a,b,[msg='测试失败时打印的信息']): 断言a和b是否相等,不相等则测试用例通过。

assertTrue(x,[msg='测试失败时打印的信息']): 断言x是否True,是True则测试用例通过。

assertFalse(x,[msg='测试失败时打印的信息']): 断言x是否False,是False则测试用例通过。

assertIs(a,b,[msg='测试失败时打印的信息']): 断言a是否是b,是则测试用例通过。

assertNotIs(a,b,[msg='测试失败时打印的信息']): 断言a是否是b,不是则测试用例通过。

assertIsNone(x,[msg='测试失败时打印的信息']): 断言x是否None,是None则测试用例通过。

assertIsNotNone(x,[msg='测试失败时打印的信息']): 断言x是否None,不是None则测试用例通过。

assertIn(a,b,[msg='测试失败时打印的信息']): 断言a是否在b中,在b中则测试用例通过。

assertNotIn(a,b,[msg='测试失败时打印的信息']): 断言a是否在b中,不在b中则测试用例通过。

assertIsInstance(a,b,[msg='测试失败时打印的信息']): 断言a是否是b的一个实例,是则测试用例通过。

assertNotIsInstance(a,b,[msg='测试失败时打印的信息']): 断言a是否是b的一个实例,不是则测试用例通过。


unittest的常用属性

['BaseTestSuite', 'FunctionTestCase', 'SkipTest', 'TestCase', 'TestLoader', 'TestProgram', 'TestResult', 'TestSuite', 'TextTestResult', 'TextTestRunner', 'TextTestResult', '_ all__', '__ builtins__', '__ doc__', '__ file__', '__ name__', '__ package__', '__ path__', '__unittest', 'case', 'defaultTestLoader', 'expectedFailure', 'findTestCases', 'getTestCaseNames', 'installHandler', 'loader', 'main', 'makeSuite', 'registerResult', 'removeHandler', 'removeResult', 'result', 'runner', 'signals', 'skip', 'skipIf', 'skipUnless', 'suite', 'util']

  • unittest.TestCase:TestCase类,所有测试用例类继承的基本类。

  • unittest.main():使用她可以方便的将一个单元测试模块变为可直接运行的测试脚本,main()方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法,并自动执行他们。执行方法的默认顺序是:根据ASCII码的顺序加载测试用例,数字与字母的顺序为:0-9,A-Z,a-z。所以以A开头的测试用例方法会优先执行,以a开头会后执行。

  • unittest.TestSuite():unittest框架的TestSuite()类是用来创建测试套件的。

  • unittest.TextTextRunner():unittest框架的TextTextRunner()类,通过该类下面的run()方法来运行suite所组装的测试用例,入参为suite测试套件。

  • unittest.defaultTestLoader(): defaultTestLoader()类,通过该类下面的discover()方法可自动更具测试目录start_dir匹配查找测试用例文件(test*.py),并将查找到的测试用例组装到测试套件,因此可以直接通过run()方法执行discover。用法如下:

discover=unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')

  • unittest.skip():装饰器,当运行用例时,有些用例可能不想执行等,可用装饰器暂时屏蔽该条测试用例。一种常见的用法就是比如说想调试某一个测试用例,想先屏蔽其他用例就可以用装饰器屏蔽。

    • @unittest.skip(reason): skip(reason)装饰器:无条件跳过装饰的测试,并说明跳过测试的原因。

    • @unittest.skipIf(reason): skipIf(condition,reason)装饰器:条件为真时,跳过装饰的测试,并说明跳过测试的原因。

    • @unittest.skipUnless(reason): skipUnless(condition,reason)装饰器:条件为假时,跳过装饰的测试,并说明跳过测试的原因。

    • @unittest.expectedFailure(): expectedFailure()测试标记为失败。