第三章 UnitTest框架

一、UnitTest框架

1.1 UnitTest框架介绍

1、什么是框架？

• 框架英文单词：framework
• 为解决一类事情的功能集合

2、什么是UnitTest框架？

是Python自带的一个单元测试框架，用它来做单元测试。

3、为什么使用UnitTest框架？

• 能够组织多个用例去执行
• 提供丰富的断言方法
• 能够生成测试报告

1.2 UnitTest核心要素

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

1.3 UnitCase

步骤：

1. 导包 import unittest
2. 新建测试类并继承uniitest.TestCase
3. 测试方法必须以test_开头

注意：__name__为python中内置变量

• 如果当前运行的模块为当前模块，那么__name__的值为__main__
• 如果当前运行的模块不是主模块，那么__name__的值为模块名称

1.4 TestSuite&TextTestRunner

"""
    目标：TestSuite使用
    操作：
        1.导包
        2.实例化对象获取
        3.调用addTest方法添加到执行的套件中
"""
# 1、导包
import unittest
from test_01_unittest import Test01
from test_02 import Test02

# 2、实例化suite
suite = unittest.TestSuite()

# 3、调用方法
# 写法1：类名('方法名')   注意：方法名使用双引号
suite.addTest(Test01("test_1"))
suite.addTest(Test01("test_2"))

# 写法2：添加测试类中所有test开头的方法
suite.addTests(unittest.makeSuite((Test02)))

# 4、实例化runner
runner = unittest.TextTestRunner()

# 5、执行
runner.run(suite)

需求：
将test01.py …test10.py共十条用例，将10条用例批量执行
问题：
使用suite.addTest(uniitest.makeSuite(ClassName))导入10条测试类
.addTest()需要添加10条
麻烦。。。

1.5 TestLoader

TestLoader：将符合条件的测试方法添加到测试套件中

# 1.导包
import unittest

# 2.调用TestLoader().discover()方法
suite = unittest.TestLoader().discover("./cases",pattern="tp*.py")

# 扩展
suite = unittest.defaultTestLoader.discover("./cases")`	# 推荐使用

# 3.执行套件
unittest.TextTestRunner().run(suite)

TestSuite和TestLoader：

• 共同点：都是测试套件
• 不同点：实现方式不同
  • TestSuite：需要手动添加测试用例
  • TestLoader：搜索指定目录下的.py文件

1.6 Fixture

Fixture控制级别：

1. 方法级别：
   def setUp() / def tearDown()
   特性：
   几个测试方法，执行几次，每个测试函数执行之前都会执行setUp，执行之后都会执行tearDown
2. 类级别：
   def setUpClass() / def tearDownClass()
   特性：
   测试类执行之前运行一次setUpClass，执行之后执行tearDownClass
3. 模块级别：
   def setUpModule() / def tearDownModule()
   特性：
   模块运行之前执行一次setUpModule，执行之后执行tearDownModule

提示：不论是有方法级别还是类级别，最常用场景为：

初始化：
1.获取浏览器实例对象
2.最大化浏览器
3.隐式等待
结束：关闭驱动对象

"""
    目标:unittest中的fixture用法
    fixture其实就是两个函数,可以一起使用,也可以单独使用
        1.初始化函数:def setUp()
        2.结束函数:def tearDown()
"""
import unittest


def setUpModule(self):
    print("setUpModule被执行")


def tearDownModule(self):
    print("tearDownModule被执行")


class Test03(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("setUpClass被执行")

    @classmethod
    def tearDownClass(cls):
        print("tearDownClass被执行")

    def test01(self):
        print("test01被执行")

    def setUp(self):
        print("setUp被执行")

    def test02(self):
        print("test02被执行")

    def tearDown(self):
        print("tearDown被执行")

二、UnitTest断言

2.1 断言介绍

1、什么是断言

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

2、为什么断言

自动化脚本都是无人值守，需要通过断言来判断自动化脚本是否通过
注意：自动化脚本不写断言，相当于没有执行测试。

2.2 常用断言方法

1. 判断是否相等：self.assertEqual(A,B)
2. 判断B是否包含A：self.assertIn(A,B)
3. 判断是否为True：self.assertTrue(flag)

2.3 案例

"""
    目标:断言练习
    实例:
        输入:正确用户名和密码,验证码为空
        断言:提示信息是否为空,验证码不能为空!
        要求:如果断言出错,截图保存
"""
# 导包
import time
import unittest
from selenium import webdriver

# 定义测试类
from selenium.webdriver.common.by import By


class TestTpshopLogin(unittest.TestCase):
    def setUp(self):
        # 获取浏览器驱动对象
        self.driver = webdriver.Chrome()
        # 打开url
        self.driver.get(r"http://demo6.tp-shop.cn/")
        # 最大化浏览器
        self.driver.maximize_window()
        # 隐式等待
        self.driver.implicitly_wait(30)
        pass

    def tearDown(self):
        # 关闭浏览器驱动
        time.sleep(2)
        self.driver.quit()
        pass

    def test_login_code_null(self):
        driver = self.driver
        # 点击登录链接
        driver.find_element(By.PARTIAL_LINK_TEXT, "登录").click()
        # 输入用户名
        driver.find_element(By.CSS_SELECTOR, "#username").send_keys("13812341234")
        # 输入密码
        driver.find_element(By.CSS_SELECTOR, "#password").send_keys("123456")
        # 输入验证码
        driver.find_element(By.CSS_SELECTOR, "#verify_code").send_keys("")
        # 点击登录
        driver.find_element(By.CSS_SELECTOR, ".J-login-submit").click()
        # 获取登录后提示信息
        result = driver.find_element(By.CSS_SELECTOR, ".layui-layer-content").text
        print(result)
        # 定义预期结果
        expect_result = "验证码不能为空! "
        try:
            # 断言
            self.assertEqual(result, expect_result)
        except AssertionError:
            # 截图
            driver.get_screenshot_as_file("./image/{}.png".format(time.strftime("%Y_%m_%d_%H_%M_%S")))
            # 抛出异常
            raise
        pass

2.4 assert扩展

断言两种实现方式：

1. 使用unittest框架断言
2. python自带断言
   assert str1 == str2 # 判断是否相等
   assert str1 in str2 # 判断str2是否包含str1
   assert True # 判断是否为True

2.5 参数化

1、为什么要参数化？

解决冗余代码问题

2、什么是参数化

根据需求动态获取参数并引用的过程

3、应用场景

解决相同业务逻辑、不同测试数据问题

4、参数化应用

1. 安装插件：pip install parameterized、
2. 验证：pip show parameterized
3. 导包 from parameterized import parameterized
4. 修饰测试函数 @parameterized.expand(数据)

参数格式：

• 单个参数：类型为列表
• 多个参数：类型为列表嵌套元组

在测试函数中的参数变量设置引用参数值时，注意：变量的数量必须和数据值的个数相同

三、跳过

1. 直接跳过：

• 语法：@unitest.skip(说明)
• 应用场景：一般适合功能未实现完成用例

2. 条件满足跳过：

• 语法：@unittest.skipIf(条件,原因)
• 场景：一般判断条件满足，就不执行；如：达到指定版本，此功能失效

提示：以上两种修饰，都可以修饰函数和类。

四、生成HTML测试报告

HMTL报告：根据TextTestRunner改编而来

4.1 操作

1. 导包：
   from HtmlTestRunner import HtmlTestRunner
2. 定义测试套件：
   suite = unittest.TestLoader().discover(‘文件路径’,pattern=‘指定文件名称’)
3. 实例化HTMLTestRunner类，并调用run方法执行测试套件：
   with open (报告存放路径,“wb”) as f :
   #实例化HtmlTestRunner类
   HtmlTestRunner(stream=f).run(suite)

"""
    目标:基于unittest框架执行生成html版报告
    操作:
        1.导包 from HtmlTestRunner import HtmlTestRunner
        2.获取报告存放文件流,并实例化HtmlTestRunner类,执行run方法
"""
import time
import unittest
from HTMLTestRunner import HTMLTestRunner
# 定义测试套件
suite = unittest.TestLoader().discover("./cases",pattern="test*.py")
# 定义报告存放路径及文件名称
report_dir = "./report/{}.html".format(time.strftime("%Y_%m_%d_%H_%M_%S"))
with open(report_dir,"wb") as f :
    HTMLTestRunner(
        stream=f,verbosity=2,title="XX项目自动化测试报告",description="操作系统win10"
    ).run(suite)

4.2 open与with open区别

"""
    open与with open的区别
    共同点:打开文件
    不同点:with open = 执行打开操作+关闭操作
"""
""" open """
f = None
try:
    f = open("./report/report.txt", "r", encoding="utf-8")
    print(f.read())
except:
    pass
finally:
    f.close()

"""with open 极力推荐"""
# 读取
with open("./report/report.txt","r",encoding="utf-8") as f:
    f.read()

    # 写入
with open("./report/report.txt","w",encoding="utf-8") as f:
    f.read()