【功能自动化】unittest 自动识别并执行测试用例方式

 discover（）方法介绍

自动化测试执行的用例有很多，python测试用例文件，都是以“test”开头的。

TestLoader（defaultTestLoader）是unittest的测试用例加载器，它包括多个加载测试用例的方法。它的结果是返回一个测试套件。本文介绍discover（）用法与功能

discover()函数是Python中unittest模块中的一个方法，用于自动发现和执行测试用例。它会自动查找并运行指定目录下的所有测试用例。

使用discover()方法可以省去手动编写测试用例集合的步骤，而是自动识别和执行所有符合要求的测试用例。

结构

discover（start_dir, pattern='test*.py', top_level_dir=None）

作用：找到指定目录下所有测试用例模块，并递归查询子目录下的测试模块，找到匹配的文件进行加载。

解释：

start_dir：需要测试的用例文件目录或是模块

pattern：用例匹配原则

top_level_dir：测试模块的顶层目录，没有就默认None。

 1.创建unitWebtours.py

将unitWebtours.py放在test文件夹下

废话不多说了，直接上代码

编写测试用例

unitWebtours.py

from selenium import webdriver
from selenium.webdriver.support.select import Select
from time import sleep
import unittest

driver = None

class Webtours(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        global driver
        driver = webdriver.Chrome()
        
    @classmethod
    def tearDownClass(cls):
        driver.quit()
    
    def test1OpenWeb(self):
        u'''打开网站'''
        driver.get("http://127.0.0.1:1080/WebTours/")
        # 等待3秒
        sleep(3)

        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")
        actual = driver.find_element_by_tag_name("body").text
        # expect = 'Welcome to the Web Tours site.'
        # # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("打开网站：成功")
        # else:
        #     print("打开网站：失败") 
    
    def test2Loginin(self):
        u'''登录'''
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")

        # 登录
        yhm = 'admin'
        mm = '123456'
        # 输入用户名
        driver.find_element_by_name("username").send_keys(yhm)
        # 输入密码
        driver.find_element_by_name("password").send_keys(mm)
        # 点击登录
        driver.find_element_by_name("login").click()
        sleep(3)
        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")

        expect = "Welcome, "+yhm
        actual = driver.find_element_by_tag_name("body").text
        # # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("登录：成功")
        # else:
        #     print("登录：失败")
    
    def test3Search(self):
        u'''搜票'''
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")
        driver.find_element_by_xpath('//img[@alt="Search Flights Button"]').click()
        sleep(3)
        
    def test4Book(self):
        u'''订票'''
        # 预定票
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")
        Select(driver.find_element_by_name("depart")).select_by_visible_text="London"
        # select_by_value 值是看不见的
        Select(driver.find_element_by_name("arrive")).select_by_visible_text="Paris"
        driver.find_element_by_name("findFlights").click()
        sleep(1)
        
        # 查看预定的航班信息
        driver.find_elements_by_name("outboundFlight")[0].click()
        driver.find_element_by_name("reserveFlights").click()
        sleep(1)
        
        # 支付信息
        driver.find_element_by_name("buyFlights").click()
        sleep(1)


        # 切换框架

        expect = "Flight 240 leaves London for Paris"
        actual = driver.find_element_by_tag_name("body").text
        # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("订票：成功")
        # else:
        #     print("订票：失败")
            
        #预定下一个
        driver.find_element_by_name("Book Another").click()
        sleep(1)
    
    def test5Logoff(self):
        u'''注销'''
        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")
        
        driver.find_element_by_xpath('//img[@alt="SignOff Button"]').click()
        sleep(1)


        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")

        expect = "sign up now"
        actual = driver.find_element_by_tag_name("body").text
        # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("注销：成功")
        # else:
        #     print("注销：失败")

if __name__=="__main__":
    # verbosity=2 显示详细信息，可以省略，测试信息不详细
    unittest.main(verbosity=2)

使用测试套件运行测试函数

下面是使用discover()方法自动识别和执行测试用例的示例代码：

import unittest
# 创建一个测试套件
suite = unittest.TestSuite()
tests = unittest.defaultTestLoader.discover('./test',pattern='unit*.py')
suite.addTests(tests)
# 创建一个测试运行器并执行测试
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)

在上述代码中，参数"./test"表示测试用例所在的目录，"pattern"参数表示测试用例文件的匹配模式，可以使用通配符进行匹配。例如，"unit*.py"表示所有以"unit"开头的.py文件。

通过调用unittest.defaultTestLoader.discover()方法可以自动识别指定目录下的所有测试用例，并返回一个测试套件。然后，通过创建一个测试运行器并调用其run()方法来执行测试。

需要注意的是，测试用例文件必须以"unit_"开头，并且必须继承自unittest.TestCase类，以便被discover()方法正确识别为测试用例。