unittest 的用法

一、discover方法

discover方法可以根据标准加载用例，并将结果返回给测试套件（suite），start_dir:待测试的目录，pattern:测试用例文件名的匹配规。

如:

start_dir = './test_case'suites = unittest.defaultTestLoader.discover(start_dir ,pattern='test.*py')

当执行多个目录的测试用例时，如何查找test_case下的其他目录吃里的测试用例文件呢？就是在test_case下每个目录里加__init__.py文件，这样会把这个目录当做标记成一个标准的Python模块。

二、测试用例的执行顺序是根据ASCLL码的顺序执行的。如：test_aaa.py优先于test_bbb.py。

三、装饰器 。这些装饰器不仅用在方法上，同样适用于类。四#跳过测试和预期失败

class MyTest(unittest.TestCase):
    @unittest.skip('直接跳过测试')
    def test_skip(self):
        print('test aaa')
 
    @unittest.skipIf(3>2,'条件为真时跳过装饰的测试')
    def test_skip_if(self):
        print('test bbb')
 
    @unittest.skipUnless(3>2,'条件为真时执行装饰的测试')
    def test_skip_unless(self):
        print('test ccc')
 
    @unittest.expectedFailure
    def test_expected_failure(self):
        self.assertEqual(3,3)
 
if __name__ == '__main__':
    unittest.main()
 
四、Fixture 执行一个或者多个测试所需要的环境。如创建临时或者代理 数据库、目录、或者启动服务器进程。

import unittest
 
def setUpModule():
    print('test module start >>>>>>>>>>>>>>>>')
def tearDownModule():
    print('test module end >>>>>>>>>>>>>>>>>>')
 
class MyTest2(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print('test class start >>>>>>>>>>>>>>')
 
    @classmethod
    def tearDownClass(cls):
        print('test class end >>>>>>>>>>>>>>>>')
 
    def setUp(self):
        print('test case start>>>>>>>>>>>>>>')
 
    def tearDown(self):
        print('test case end>>>>>>>>>>>>>>>')
 
    def test_case1(self):
        print('test case1')
 
    def test_case2(self):
        print("test case2")
 
if __name__ == '__main__':
    unittest.main()

五、断言方法。

class TestAssert(unittest.TestCase):
 
    def test_equal(self):
        self.assertEqual(2+2,4)
        self.assertEqual('python','python')
        self.assertNotEqual('python','pythno')
 
    def test_in(self):
        self.assertIn('hello','hello world')
        self.assertNotIn('hi','hello')
 
    def test_true(self):
        self.assertTrue(1) # or Ture
        self.assertFalse(0) # or False

六、数据驱动应用。
常用的csv读取数据执行用例，会导致所有数据被当做一条用例来执行，只要有一条用例失败，那么整个用例就执行失败了。一种解决方法是将数据读取到列表中，在用例中按照下标来调用。
但是这样有两个问题，一是每次都要读取文件，耗费空间，二是数据在列表中，如果插入一条数据，或者删除一条数据，那么用例在调用数据时就会非常容易出错，不利于维护。

下面介绍2个模块来解决数据驱动问题。
Parameterized是Python的一个库。支持unittest、pytest、Nose框架。

import unittest
from  time import sleep
from selenium import webdriver
from parameterized import parameterized
 
class TestBaidu(unittest.TestCase):
    '''百度搜索测试'''
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = 'https://www.baidu.com'
 
    def baidu_search(self,search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id('kw').send_keys(search_key)
        self.driver.find_element_by_id('su').click()
        sleep(2)
 
    @parameterized.expand([
        ('case1','selenium'),
        ('case2', 'unittest'),
        ('case3', 'parameterized'),
    ])
    def test_search(self,name,search_key):
        ''''''
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + '_百度搜索')
 
    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()
 
if __name__ == '__main__':
    unittest.main()
 
在@parameterized.expand()中每个元组都是一条测试用例。name对应的是元组中第一列数据，search_key对应的是第二列数据。参数化会自动加上0,1,2来区分每条用例。元组中的数据作为用例名的后缀出现在测试报告中。

DDT是uittest的一个扩展库。
测试类需要通过@ddt装饰器进行装饰。DDT提供不同形式的参数化，比如列表，元组，字典等。需要注意字典的KEY与测试方法的参数要保持一致。
DDT也支持数据文件(json)的参数化。

import unittest
from  time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack
 
@ddt
class TestBaidu(unittest.TestCase):
    '''百度搜索测试'''
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = 'https://www.baidu.com'
 
    def baidu_search(self,search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id('kw').send_keys(search_key)
        self.driver.find_element_by_id('su').click()
        sleep(2)
 
    @data(['case1','selenium'],['case2','unittest'],['case3', 'parameterized'])
    @unpack
    def test_search1(self,name,search_key):
        print('第一组测试用例：', name)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + '_百度搜索')
 
    @data(('case1','selenium'),
        ('case2', 'unittest'),
        ('case3', 'parameterized'),)
    @unpack
    def test_search2(self, name, search_key):
        print('第二组测试用例：',name)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + '_百度搜索')
 
    @data({"search_key":"python"},{"search_key":"ddt"},{"search_key":"Selenium"})
    @unpack
    def test_search2(self, search_key):
        print('第三组测试用例：',search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + '_百度搜索')
 
    # @file_data('ddt_data_file.json')
    # def test_search4(self, search_key):
    #     print('第四组测试用例：',search_key)
    #     self.baidu_search(search_key)
    #     self.assertEqual(self.driver.title, search_key + '_百度搜索')
 
    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()
 
if __name__ == '__main__':
    unittest.main()

ddt_data_file.json

{
  "case1":{"search_key":"python"},
  "case2":{"search_key":"ddt"},
  "case3":{"search_key":"Selenium"}
 
}
 
七、HTML测试报告。HTMLTestRunner支持Python3的下载地址：https://github.com/defnngj/HTMLTestRunner

#获取当前的时间
now_time = time.strftime('%Y-%m-%d %H_%M_%S')
HTML_report = './test_report/'+now_time+'result.html'
fp = open(HTML_report,'wb')
runner = HTMLTestRunner(stream=fp,title='Fixture', description='运行环境：Windows 10,Chrome浏览器')
runner.run(suites)
#关闭
fp.close()