Python-unittest测试框架

妖狐艹你老母 2022-06-13 05:29 324阅读 0赞

# 一、unittest是一个Python的单元测试框架，可以帮我们完成接口自动化测试、UI自动化测试 #

## 一个使用unittest框架的自动化测试用例代码如下： ##

test_baidu.py

# coding=utf-8
    ''' Created on 2017-6-14 @author: 灵枢 Project:百度测试用例 '''
    from selenium import webdriver
    import unittest, time
    
    class BaiduTest(unittest.TestCase):
        def setUp(self):
            self.driver = webdriver.Chrome()
            self.driver.implicitly_wait(30) #隐性等待时间为30秒
            self.base_url = "https://www.baidu.com"
    
        def test_baidu_search(self):
            u"""百度搜索"""
            driver = self.driver
            driver.get(self.base_url + "/")
            driver.find_element_by_id("kw").clear()
            driver.find_element_by_id("kw").send_keys("unittest")
            driver.find_element_by_id("su").click()
            time.sleep(3)
            title=driver.title
            self.assertEqual(title, u"unittest_百度搜索")
    
        def test_baidu_set(self):
            u"""百度设置"""
            driver=self.driver
            driver.get(self.base_url+"/gaoji/preferences.html")
            m=driver.find_element_by_name("NR")
            time.sleep(1)
            m.find_element_by_xpath("//option[@value='50']").click()
            time.sleep(1)
    
        def tearDown(self):
            self.driver.quit()
    
    if __name__ == "__main__":
        # unittest.main()
        # 构建测试集
        suite = unittest.TestSuite()
        suite.addTest(BaiduTest("test_baidu_search"))
        suite.addTest(BaiduTest("test_baidu_set"))
        #执行测试
        runner = unittest.TextTestRunner()
        runner.run(suite)

# 二、代码解释 #

###  首先，通过import导入unittest测试框架。创建BaiduTest 类继承unittest.TestCase 类，从 TestCase 类继承是告诉 unittest 这是一个测试用例。 ###

###  setUp ()方法用于设置初始化的部分，如初始化变量、打开浏览器等，在测试用例执行前，这个方法中的函数将先被调用。 ###

###  tearDown()方法用例测试用例执行之后的善后工作，如清除数据、关闭文件、关闭浏览器等。 ###

###  unittest要求测试用例必须以“test”开头。如test\_baidu\_search、test\_baidu\_set。 ###

###  接下来，调用unittest.TestSuite()类的addTest()方法向测试套件中添加测试用例 ###

###  最后，通过unittest.TextTestRunner()类的run()方法运行测试套件中的测试用例 ###

###  如果想直接运行当前测试文件中的所有测试用例，则可以直接使用unittest提供的main()方法。只需在if **name** == “**main**“:下面写unittest.main()就行了。 ###

# 三、批量执行测试用例 #

## 这里还有一个搜狗搜索的测试用例，如下： ##

test_sogou.py

# coding=utf-8
    ''' Created on 2017-6-14 @author: 灵枢 Project:搜狗搜索测试用例 '''
    from selenium import webdriver
    import unittest, time
    
    class SogouTest(unittest.TestCase):
        def setUp(self):
            self.driver = webdriver.Chrome()
            self.driver.implicitly_wait(30) #隐性等待时间为30秒
            self.base_url = "http://www.sogou.com"
    
        def test_sogou(self):
            u"""搜狗搜索"""
            driver = self.driver
            driver.get(self.base_url + "/")
            driver.find_element_by_id("query").clear()
            driver.find_element_by_id("query").send_keys(u"python3")
            driver.find_element_by_id("query").submit()
            time.sleep(3)
            title = driver.title
            self.assertEqual(title, u"python2 - 搜狗搜索")
    
        def tearDown(self):
            self.driver.quit()
    
    if __name__ == "__main__":
        unittest.main()

## 如果想同时执行test\_baidu.py和test\_sogou.py，可以新建一个类用于批量执行测试用例： ##

all_search_tests.py

#coding=utf-8
    ''' Created on 2017-6-14 @author: 灵枢 Project:批量执行测试用例 '''
    import HTMLTestRunner
    import unittest
    import os,time
    
    
    test_case_dir = "E:\\selenium_use_case\\test_case"  #test_baidu.py和test_sogou.py都放在该目录下
    def get_test_suite():
        testsunit=unittest.TestSuite()
        #在listaa目录下查找以test开头的python文件
        discover=unittest.defaultTestLoader.discover(test_case_dir,pattern='test_*.py',top_level_dir=None)
        #循环添加测试用例
        for test_suite in discover:
            for test_case in test_suite:
                testsunit.addTests(test_case)
                print(testsunit)
        return testsunit
    
    if __name__ == '__main__':
        runner = unittest.TextTestRunner()
        runner.run(get_test_suite())