UnitTest是Python自带的一个单元测试框架

作用：

1. 批量执行用例
2. 提供丰富的断言知识
3. 可以生成报告

核心要素：

1. TestCase 测试用例
2. TestSuite 测试案件
3. TestRunner 以文本的形式运行测试用例
4. TestLoader 批量执行测试用例-搜索指定文件夹内指定字母开头的模块
5. Fixture 固定装置（两个固定的函数，一个初始化时使用，一个结束时使用）

TestCase 测试用例

案例： 定义一个实现加法操作的函数，并对该函数进行测试

1、定义测试用例

1. 导包 import unittest
2. 定义测试类：新建测试类必须继承 unittest.TestCase
3. 定义测试方法：必须以test开头命名

2、执行测试用例

法一：右键选择使用UnitTest运行
法二：调用uniitest.main()运行

import unittest


import unittest


def add(a, b):
    return a + b


# 定义测试类
class Test01(unittest.TestCase):
    # 定义测试方法
    def test_add(self):
        result = add(1, 1)
        print("结果为：", result)
        print("__name__的值：", __name__)   # __name__的值： test_unittest01

    def test_add2(self):
        result = add(1, 2)
        print("结果为：", result)


if __name__ == '__main__':  # 未执行的
    unittest.main()
    
""""
   __name__是python中内置变量
值:
    1、如果当前运行的模块为当前模块，那么__name__的值为：__main__
    2、如果当前运行的模块不是主模块，那么__name__的值为：模块名称
   
""" 

TestSuite 测试套件

多条测试用例集合在一起就算一个TestSuite

1. 实例化： suite = unittest.TestSuite()
2. 添加用例：suite.addTest(ClassName("MethodName"))
3. 添加扩展：suite.addTest(uniitest.maskSuite(ClassName))

TestSuite需要配合TestRunner才能被执行

TextTestRunner 执行测试套件

import unittest
from test_unittest01 import Test01
from test02 import Test02
# 实例化
suite = unittest.TestSuite()
# 调用添加方法
# 写法一: suite.addTest(类名(“方法名”)) 注意:方法名称使用双引号
# suite.addTest(Test01("test_add"))
# suite.addTest(Test01("test_add2"))

# 法二:suite.addTest(unittest.makeSuite(类名)) 添加指定类中所有已test开头的方法
suite.addTest(unittest.makeSuite(Test02))

# 执行测试套件
runner = unittest.TextTestRunner()
runner.run(suite)

TestLoader 套件

用来加载TestCase到TestSuite中，即加载满足条件的测试用例，并把测试用例封装成测试套件。
使用unittest.TestLoader，通过该类下面的discover()方法自动搜索指定目录下指定开头的.py文件，并将查找到的测试用例组装到测试套件。

用法：

suite = unittest.TestLoader().discover(test_dir, pattern='test*.py')
也可以使用unittest.defaultTestLoader代替unittest.TestLoader()

import unittest

suite = unittest.TestLoader().discover("../cases", "test*.py")
# 推荐使用
# suite = unittest.defaultTestLoader.discover("../cases")

unittest.TextTestRunner().run(suite)

TestLoader与TestSuite区别

1. TestSuite需要手动添加测试用例（可以添加测试类，也可以添加测试类中某个测试方法）
2. TestLoader搜索指定目录下指定开头.py文件，并添加测试类中的所有的测试方法，不能指定添加测试方法

Fixture装置函数

setUp 、teatDown

1. 函数级别
   初始化函数：def setup()
   结束函数：def teardown()
   几个测试函数，执行几次，每个测试函数执行之前都会执行setUp，执行之后都会执行tearDown

2. 类级别（需使用@classmethod修饰）
   def setUpClass() 、def tearDownClass()
   测试前运行一次setUpClass，运行之后运行一次tearDownClass

3. 模块级别
   def setUpModule() def tearDownModule()
   模块运行一次setUpModule，运行之后运行一次tearDownModule

断言assert

概念：让程序代替人为判断测试程序执行结果是否符合预期结果的过程

self.

断言方法	断言描述
assertTrue(expr,msg=None)	验证expr是true，如果为false，则fail
assertFlase(expr,msg=None)	验证expr是true，如果为false，则fall
assertEqual(expected,actual,msg=None)	验证expected==actual，不等则fall
assertNotEqual(first,second,msg=None)	验证first!=second，相等则fall
asserIn(member,container,msg=None)	验证是否member in container
asserNotIn(member,container,msg=None)	验证是否member not in container

断言案例

import unittest
import time
from selenium import webdriver


class TestLogin(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome('../chromedriver-win64/chromedriver.exe')
        self.driver.get("https://www.4399.com/")
        self.driver.maximize_window()
        # 隐式等待
        self.driver.implicitly_wait(30)

    def tearDown(self):
        time.sleep(2)
        self.driver.quit()

    def test_login_code(self):
        driver = self.driver
        driver.find_element_by_css_selector("#login_tologin").click()
        driver.switch_to.frame("popup_login_frame")
        driver.find_element_by_css_selector("#username").send_keys("1238624243")
        driver.find_element_by_css_selector(".ptlogin_btn").click()

        result = driver.find_element_by_css_selector("#Msg").text
        print("result:", result)

        driver.switch_to_default_content()

        # 断言
        # expect_result = "请输入密码"![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/4e9dcff870ca4f10822680bda805344f.png)

        expect_result = "请输入密码！"
        try:
            self.assertEqual(result, expect_result)
        except AssertionError:
            driver.get_screenshot_as_file("../images/{}.png".format(time.strftime("%Y_%m_%d_%H_%M_%S")))
            raise

参数化应用

pip install parameterized

参数化：解决冗余代码问题

import unittest
from parameterized import parameterized


class Test01(unittest.TestCase):
    # 单个参数
    @parameterized.expand(["1", "2", "3"])
    def test_add_one(self, num):
        print("num:", num)

    # 多个参数
    @parameterized.expand([(1, 2, 3), (3, 0, 3), (2, 1, 3)])
    def test_add_more(self, a, b, result):
        print("{}+{}={}".format(a, b, result))

    data = [(1, 2, 3), (3, 0, 3), (2, 1, 3)]
    @parameterized.expand(data)
    def test_add_more(self, a, b, result):
        print("{}+{}={}".format(a, b, result))

    # 推荐写法
    def get_data(self):
        return [(1, 2, 3), (3, 0, 3), (2, 1, 3)]

    @parameterized.expand(get_data)
    def test_add_more(self, a, b, result):
        print("{}+{}={}".format(a, b, result))

总结