（四十）Unittest单元测试框架之认识unittest-discover更多测试用例

随笔记录方便自己和同路人查阅。

#------------------------------------------------我是可耻的分割线-------------------------------------------

　　学习selenium自动化之前，最好先学习HTML、CSS、JavaScript等知识，有助于理解定位及操作元素的原理。关于python和selenium安装请自行搜索别的资料，这里就不多做介绍了，所有例子均使用python3.6+selenium执行的。

#------------------------------------------------我是可耻的分割线-------------------------------------------

discover更多测试用例

随着软件功能的不断增加，对应的测试用例也会呈指数级增长。一个实现几十个功能的项目，对应的单元测试用例可能达到上百个。如果把所有的测试用例都写在一个test.py文件中，那么这个文件会越来越臃肿，后期委会起来也比较麻烦。需要将这些用例按照所测试的功能进行拆分，分散到不同的测试文件中。

对上例中test.py文件的测试用例进行差分，拆分后的目录结构如下：

文件拆分后的实现代码如下：

Testadd.py

import unittest
from calculator import Count

class TestAdd(unittest.TestCase):
    def setUp(self):
        print('test case start')

    def tearDown(self):
        print('test case end')

    def test_add(self):
        j = Count(2,3)
        self.assertEqual(j.add(),5)

    def test_add2(self):
        j = Count(10,5)
        self.assertEqual(j.add(),15)

#执行测试类的测试方法
if __name__ == '__main__':
    unittest.main()

Testsub.py文件

import unittest
from calculator import Count

class TestSub(unittest.TestCase):
    def setUp(self):
        print('test sub start')

    def test_sub(self):
        j = Count(3, 2)
        self.assertEqual(j.sub(), 1)

    def test_sub2(self):
        j = Count(10,5)
        self.assertEqual(j.sub(), 5)

    def tearDown(self):
        print('test sub end')
        
#执行测试类的测试方法
if __name__ == '__main__':
    unittest.main()

接着创建用于执行所有用例的runtest.py文件

__author__ = 'Mr.Li'

import unittest
#加载测试文件
import testadd
import testsub

#构建测试集
suite = unittest.TestSuite()

suite.addTest(testadd.TestAdd("test_add"))
suite.addTest(testadd.TestAdd("test_add2"))

suite.addTest(testsub.TestSub("test_sub"))
suite.addTest(testsub.TestSub("test_sub2"))

#执行测试类的测试方法
if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite)

这样的拆分带来了好处，可以根据不同的功能创建不同的测试文件，甚至是不同的测试目录，测试文件中还可以将不同的小功能划分为不同的测试类，在类下编写测试用例，整体结构更加清晰。

这样的设计看上去很完美，但依然没有解决添加用例的问题，当用例达到成百上千条时，在runtest.py文件中通过addTest()添加/删除测试用例就变的非常麻烦，那么有没有方法让unittest单元测试框架自动识别测试用例呢？答案是可定的，TestLoader类中提供的discover()方法可以解决这个问题。

TestLoader

该类负责根据各种标准加载测试用例，并将它们返回给测试套件。正常情况下，不需要创建这个类的实例。Unittest提供了可以共享的defaultTestLoader类，可以使用其子类和方法创建实例，discover()方法就是其中之一。

discover(start_dir,pattern=’test*.py’,top_level_dir=None)

找到指定目录下所有测试模块，并可递归查到子目录下的测试模块，只有匹配到文件名才能被加载。如果启动的不是顶层目录，那么顶层目录必须单独指定。

start_dir: 要测试的模块名或测试用例目录。

pattern=’test*.py’：表示用例文件名的匹配原则。此处匹配文件名以“test”开头的”.py”类型文件，星号“*”表示任意多个字符。

top_level_dir=None：测试模块的顶层目录，如果没有顶层目录，默认为None。

现在通过discover()方法重新实现runtest.py文件的功能。

import unittest

#定义测试用例的目录为当前目录
test_dir = './'
dicscover = unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')
#执行测试类的测试方法
if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(dicscover)

discover()方法会自动根据测试目录(test_dir)匹配查找测试用例文件（test*.py），并将查到的测试用例组装到测试套件中，因此，可以直接通过run()方法执行discover，大大简化了测试用例的查找与执行。