『德不孤』Pytest框架 — 12.Fixture装饰器的使用

4、yield执行后置函数

前面的示例，是在用例前加前置条件，相当于执行了setup()前置函数，既然有setup()前置函数那就会有teardown()后置函数，Fixture里面的teardown()后置函数用yield来唤醒。

示例：

import pytest


# 步骤1
@pytest.fixture()
def login():  # 步骤2
    print("打开APP")
    print("输入账号，密码，进行登录")
    yield  # 当用例执行完成后,执行yield后的代码
    print("关闭APP")


def test_add_cart(login):  # 步骤3
    print("添加购物车--需要登录")


def test_add_address(login):  # 步骤3
    print("添加收货地址--需要登录")


if __name__ == '__main__':
    pytest.main()

"""
执行结果：

打开APP
输入账号，密码，进行登录
添加购物车--需要登录
关闭APP
PASSED打开APP
输入账号，密码，进行登录
添加收货地址--需要登录
关闭APP
PASSED
"""

5、addfinalizer关键字

yield是当用例执行完之后，会执行yield后面的代码，但用例不能有return返回值。

addfinalizer实现功能跟yield一样，但是用例可以return参数，传给后面用例。

示例1：

import pytest


@pytest.fixture()
def login(request): # request是Pytest中的一个关键字，固定写法。
    # 步骤1：正常编写前置用例
    print("打开APP")
    print("输入账号，密码，进行登录")
    username = '小明'

    # 步骤3：定义用例执行后要执行的代码，封装到一个函数中
    def closeApp(): # 这个closeApp函数命是自定义的
        print("关闭APP")
        
    # 步骤4：执行上面封装的代码。
    # 通过request关键字，结束上面的函数。
    request.addfinalizer(closeApp)# 终结函数
    
    # 步骤2：返回参数给测试用例。
    # 返回参数可以是变量，对象，表达式，常量值。
    return username


def test_add_cart(login):  # 步骤3
    print("添加购物车--需要登录")
    # 这里需要注意： 如果要使用fixture方法返回的数据，
    # 就直接填写fixture方法名称。
    # 返回参数的名称和可以和fixture方法的名称相同。
    print(f"登陆者：{login}") 


def test_add_address(login):  # 步骤3
    print("添加收货地址--需要登录")
    print(f"登陆者：{login}")

if __name__ == '__main__':
    pytest.main()

"""
执行结果：

test_01.py::test_add_cart 打开APP
输入账号，密码，进行登录
添加购物车--需要登录
登陆者：小明
PASSED关闭APP

test_01.py::test_add_address 打开APP
输入账号，密码，进行登录
添加收货地址--需要登录
登陆者：小明
PASSED关闭APP
"""

示例2：

# 示例一个selenium自动化关闭开启关闭浏览器的操作。
import time
import pytest
from selenium import webdriver


@pytest.fixture()
def driver(request):  # request是Pytest中的一个关键字，固定写法。
    # 步骤1：创建浏览器驱动对象
    driver = webdriver.Firefox()

    # 步骤3：定义用例执行后要执行的代码，封装到一个函数中
    def end():  # 这个end函数命是自定义的
        driver.quit()

    # 步骤4：执行上面封装的代码。
    # 通过request关键字，结束上面的函数。
    request.addfinalizer(end)  # 终结函数

    # 步骤2：返回浏览器驱动对象，给测试用例
    # 返回参数可以是变量，对象，表达式，常量值。
    # 返回参数的名称和可以和fixture方法的名称相同。
    return driver


def test_baidu(driver):
    """打开百度"""
    driver.get("http://www.baidu.com")
    time.sleep(3)


def test_163(driver):
    """打开网易"""
    driver.get("http://www.163.com")
    time.sleep(3)


if __name__ == '__main__':
    pytest.main()

示例3：使用yield也可以返回数据。（这种方式好神奇）

import pytest


@pytest.fixture()
def login(request):
    print("打开APP")
    print("输入账号，密码，进行登录")
    username = '小明'
    yield username
    print("关闭APP")


def test_add_cart(login):  # 步骤3
    print("添加购物车--需要登录")
    print(f"登陆者：{login}")


def test_add_address(login):  # 步骤3
    print("添加收货地址--需要登录")
    print(f"登陆者：{login}")


if __name__ == '__main__':
    pytest.main()

"""
执行结果：
test_01.py::test_add_cart 打开APP
输入账号，密码，进行登录
添加购物车--需要登录
登陆者：小明
PASSED关闭APP

test_01.py::test_add_address 打开APP
输入账号，密码，进行登录
添加收货地址--需要登录
登陆者：小明
PASSED关闭APP
"""

总结：return和yield都表示返回的意思，但是return的后面不能有代码，yield的后面可以接代码。