细说unittest-2

• 2020 年 1 月 16 日
• 笔记

一、unittest模块官方文档：

https://docs.python.org/3/library/unittest.html

二、一张图看懂unittest：

三、Unittest主要方法属性：

1.unittest.TestCase:TestCase类，所有测试用例继承的基本类:

class myfunTest(unittest.TestCase):

   　　　 def setUp(self):

   　　　 print('每个用例执行前都会调用setUp准备环境')

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

if __name=="__main__()":

  　　　 unittest.main()

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

　tests=myfunTest("test_is_prime")

　　　　suite=unittest.TestSuite()

　　　　suite.addTest(tests)

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

runner=unittest.TextTestRunner()

　　　　runner.run(suite)

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

　discover=unittest.defaultTestLoader.discover(test_dir,pattern=’test*.py’):

　　　　test_dir = r'D:CollectionOverviewwebzidonghu'

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

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

@unittest.skip(reason)：无条件跳过装饰的测试，并说明跳过测试的原因。

@unittst.skipif(condition,reason)：条件为真时，跳过装饰的测试，并说明跳过测试的原因。

@unittest.skipUnless(condition,reason)：条件为假是，跳过装饰的测试，并说明跳过测试的原因。

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

@unittest.skipUnless(sys.platform.startswith('linux'),'requires Linux') #跳过该测试用例

def test_divide(self):

   　　　　　''' test method divide '''

　　　　　　print('divide')

7.setUp():用于测试用例执行前的初始化工作。如测试用例中需要访问数据库，可以在setUp中建立数据库连接并进行初始化。如测试用例需要登陆web，可以先实例化浏览器:

def setUp(self):

   　　　　　url='https://mail.yeah.net/'

　　　　　　self.browser=webdriver.Firefox()

　　　　　　self.browser.got(url)

　　　　　　time.sleep(5)

8.tearDown():用于测试用例执行之后的善后工作。如关闭数据库链接。关闭浏览器:

　def tearDown(self):

   　　　　　self.browser.closet()

9.assert*():一些断言方法，在执行测试用例的过程中，最终用例是否执行通过，是通过判断测试得到的实际结果和预期结果是否想等决定的:

self.assertTrue(is_prime(5))

　self.assertFalse(is_prime(-3))

10.addTest():将测试用例添加到测试套件中，如下方，是将test_baidu模块下的BaiduTest类下的test_baidu测试用例添加到测试套件：

　suite=unittest.TestSuite()

　　　　suite.addTest(test_baidu.BaiduTest(‘test_baidu’))

11.run():运行测试套件的测试用例，入参为suite测试套件：

runner=unittest.Text.Text.Runner()

　　　　runner.run(suite)

四、框架如何解决自动化需求的4个问题：

1.如何控制用例执行顺序？

在unittest中，用例是以test开头的方法定义的，默认执行顺序是根据用例名称的ASCII码升序进行如上面的用例，实际执行顺序为：test_add->test_divide->test_is_prime，而不是用例定义的先后顺序。在unittest中解决用例执行顺序的问题是使用TestSuite，代码如下：

#使用TestSuite控制用例顺序，用例的执行顺序是由添加到TestSuite的顺序决定的 tests=[myfunTest("test_is_prime"),myfunTest("test_add"),myfunTest("test_divide")] suite=unittest.TestSuite() #suite.addTest() suite.addTest(tests) #将测试用例实例增加到测试套件中 runner=unittest.TextTestRunner() runner.run(suite)

 2.如何让多个用例共用setUp,tearDown？

Unittest的setUp、tearDown会在每个用例执行前后执行一次，如上面测试用例类中有3个测试用例，那么每个用例执行前会执行setUp，执行后会执行tearDown，即setUp、tearDown总共会调用三次，但考虑实际自动化测试场景，多个用例只需执行一次setUp，全部用例执行完成后，执行一次tearDown，针对该种场景，unittest的处理方法是使用setupclass、teardownclass，注意@classmethod的使用，代码如下：

@classmethod def setUpClass(cls): print('所有用例执行前会调用一次setUp准备环境') @classmethod def tearDownClass(cls): print('所有用例执行后会调用一次tearDown进行环境清理')

 3.如何跳过用例：

在自动化测试中，经常会遇到挑选用例的清况，在unittest中的解决方法是使用skip装饰器，其中skip装饰器主要有3种：

unittest.skip(reason)，

unittst.skipif(condition,reason)，

unittest.skipUnless(condition,reason)，

即在满足conditiong条件下跳过该用例，reason用于描述跳过的原因，实例代码如下：

@unittest.skipUnless(sys.platform.startswith('linux'),'requires Linux') #跳过该测试用例 def test_divide(self): ''' test method divide ''' print('divide') self.assertEqual(2,divide(6,3)) self.assertNotEqual(2,divide(5,2))

 4.如何生成html格式的测试报告：

Unittest中默认生成的报告格式为txt,如果想生成html格式的报告，可以使用HtmlTestRunner模块，安装后导入该模块，使用HTMLTestRunner代替默认的TextTestRunner()执行测试用例即可。实例代码如下：

suite=unittest.TestSuite() suite.addTest(unittest.TestLoader().loadTestFromTest(myfunTest)) #使用TestLoader加载测试用例 runner=HTMLTestRunner(output="result") runner.run(suite)

五、完整代码如下：

 1 import unittest   2 #https://pypi.org/project/html-testRunner   3 from HtmlTestRunner import HTMLTestRunner    #导入第三方模块，执行结果生成html报告：pip install html-testRunner   4   5 class myfunTest(unittest.TestSuite):   6     ''' test myfunTest: test fixture only once '''   7   8   9     @classmethod  10     def setUpClass(cls):  11         print('所有用例执行前会调用一次setUp准备环境')  12  13  14     @classmethod  15     def tearDownClass(cls):  16         print('所有用例执行后会调用一次tearDown进行环境清理')  17  18  19     def test_add(self):  20         ''' test method add(a,b) '''  21         print('add')  22         self.assertEqual(3,add(1,2))  23         self.assertNotEqual(3,add(2,2))  24  25  26     @unittest.skipUnless(sys.platform.startswith('linux'),'requires Linux')    #跳过该测试用例  27     def test_mul(self):  28         ''' test method mul '''  29         print('mul')  30         self.assertEqual(2,mul(6,3))  31         self.assertNotEqual(13,mul(7,2))  32  33  34 if __name__=='__main__':  35     suite=unittest.TestSuite()  36     suite.addTest(unittest.TestLoader().loadTestFromTest(myfunTest))    #使用TestLoader加载测试用例  37  38     runner=HTMLTestRunner(output="result")  39     runner.run(suite)

 1 #coding=utf-8   2   3 import unittest   4   5   6 class myfunTest(unittest.TestCase):   7      '''Test myfunTes'''   8   9  10     def setUp(self):  11         print('每个用例执行前都会调用setUp准备环境')  12  13  14     def is_prime(number)：  15         if number<0 or number in (0,2):  16             return False  17         for element in range(2,number):  18             if number%element==0:  19                 return False  20         return True  21  22  23     def add(a,b):  24         return a+b  25  26  27     def mul(a,b):  28         return a*b  29  30  31     def test_is_prime(self):  32         '''  33         Test method is_prime(number)  34         '''  35         print('is_prime')  36         self.assertTrue(is_prime(5))  37         self.assertFalse(is_prime(8))  38         self.assertFalse(is_prime(-3))  39  40  41     def test_add(self):  42         '''  43         Test method add(a,b)  44         '''  45         print('add')  46         self.assertEqual(3,add(1,2))  47         self.assertNotEqual(3,add(2,2))  48  49  50     def test_mul(self):  51         '''  52         Test method mul  53         '''  54         print ('mul')  55         self.assertEqual(18,mul(6,3))  56         self.assertNotEqual(6,mul(3,2))  57  58  59     def tearDown(self):  60         print('每个用例执行完都会调用tearDown进行环境清理')  61  62  63 if __name__='__main__':  64     #使用TestSuite控制用例顺序，用例的执行顺序是由添加到TestSuite的顺序决定的  65     tests=[myfunTest("test_is_prime"),myfunTest("test_add"),myfunTest("test_mul")]  66  67     suite=unittest.TestSuite()  68     #suite.addTest()  69     suite.addTest(tests)    #将测试用例实例增加到测试套件中  70  71     runner=unittest.TextTestRunner()  72     runner.run(suite)