tep 是一款测试工具，在 pytest 测试框架基础上集成了第三方包，提供项目脚手架，帮助以写 Python 代码方式，快速实现自动化项目落地。fixture 是 pytest 核心技术，本文聊聊如何使用 fixture 完成变量、接口、用例 3 个级别复用。

技术原理

fixture 是一种特殊函数，实现了依赖注入，pytest 规定，只要在conftest.py模块中定义了 fixture，作为参数传给测试函数test(fixture_name)就可以使用，无需import。tep 在conftest.py上封装了一层fixtures，类似于 Django 初始化数据加载的fixtures，定义了env_vars环境变量，预置了登录接口等。整体思路如下图所示：

• 实线：表示 fixture 流转，从tep.fixture预置 fixture，到fixtures，到conftest.py，到tests。
• 虚线：表示动态变量流转，参考 JMeter vars.put()和vars.get()，实现用例级别复用。
• 蓝色线：表示静态变量流转，参考 JMeter自定义用户变量，实现变量级别复用。

tep.fixture

tep.fixture第一块内容是预置 fixture 的实现代码，如url、faker_ch等，对使用者隐藏。第二块内容是类TepVar，预置了 1 个 Python 字典vars_和 2 个存取方法put()、get()，划分了一块 key-value 形式存储区域。

fixtures

fixtures是一个包，里面包括多个fixture_*.py格式的 Python 模块，定义的全部都是 fixture，只对外提供fixture，不提供function。

自动注册

conftest.py是 pytest 的特殊文件，文件名固定，tep 在其中实现了自动注册功能，它会自动查找fixtures目录下，所有以fixture_开头，以.py结尾的 Python 模块进行导入，无需手动添加。注册后，tests才能够找到fixtures中定义的fixture，否则报错fixture not found。

tests

tests存放所有测试用例，tep 建议每个 Python 模块中只定义 1 个test()函数，用例分离，独立运行。

参考了HttpRunner每个yaml文件 1 条用例的做法。复杂场景可以定义多个测试test()函数或Test类，pytest 都支持。

使用示例

变量级别复用

函数局部变量

import jmespath
from loguru import logger

from tep.client import request


def test_post(faker_ch, url, login):
    # description
    logger.info("test post")
    # data
    fake = faker_ch
    # request
    response = request(
        "post",
        url=url("/api/users"),
        headers=login.jwt_headers,
        json={
            "name": fake.name()
        }
    )
    # assert
    assert response.status_code < 400
    # extract
    user_id = jmespath.search("id", response.json())

函数内部变量，作用域是整条测试用例，可以在多个接口之间进行复用，比如参数化和关联：

全局环境变量

#!/usr/bin/python
# encoding=utf-8

""" Can only be modified by the administrator. Only fixtures are provided.
"""

from tep.dao import mysql_engine
from tep.fixture import *


@pytest.fixture(scope="session")
def env_vars(config):
    class Clazz(TepVars):
        env = config["env"]

        """Variables define start"""
        # Environment and variables
        mapping = {
            "qa": {
                "domain": "https://qa.com",
                "mysql_engine": mysql_engine("127.0.0.1",  # host
                                             "2306",  # port
                                             "root",  # username
                                             "123456",  # password
                                             "qa"),  # db_name
            },
            "release": {
                "domain": "https://release.com",
                "mysql_engine": mysql_engine("127.0.0.1",
                                             "2306",
                                             "root",
                                             "123456",
                                             "release"),
            }
            # Add your environment and variables
        }
        # Define properties for auto display
        domain = mapping[env]["domain"]
        mysql_engine = mapping[env]["mysql_engine"]
        """Variables define end"""

    return Clazz()


@pytest.fixture
def project_level():
    pass

使用：

def test(env_vars, env_vars_your_name):
    print(env_vars.common_var)
    print(env_vars_your_name.my_var)

全局环境变量由fixtures提供：

核心数据是env_vars.Clazz类里面的mapping、domain、mysql_engine等属性：

这么做的目的是为了在 PyCharm 中敲代码，输入.后能自动带出来，提高编码效率。

fixture_admin.py建议由管理员维护项目级别的公共变量，fixture_your_name.py由团队成员定义自己的环境变量，避免冲突。它们还有一个区别是fixture_admin.py的env_vars继承了tep.fixture.TepVar，支持env_vars.put()和env_vars.get()。

变量都由fixture提供。

接口级别复用

tep 预置登录接口fixture_login.py已经实现了接口级别复用，其他接口可以参照实现：

from tep.client import request
from tep.fixture import *


def _jwt_headers(token):
    return {"Content-Type": "application/json", "authorization": f"Bearer {token}"}


@pytest.fixture(scope="session")
def login(url):
    # Code your login
    logger.info("Administrator login")
    response = request(
        "post",
        url=url("/api/users/login"),
        headers={"Content-Type": "application/json"},
        json={
            "username": "admin",
            "password": "123456",
        }
    )
    assert response.status_code < 400
    response_token = jmespath.search("token", response.json())

    class Clazz:
        token = response_token
        jwt_headers = _jwt_headers(response_token)

    return Clazz

使用：

from loguru import logger


def test_login(login):
    logger.info(login.token)

用例级别复用

a_test.py，定义 1 条测试用例test_a()，使用faker_ch生成随机name，通过env_vars.put()方法写入TepVar.vars_中：

def test_a(faker_ch, env_vars):
    name = faker_ch.name()
    env_vars.put("name", name)

reuse_a_test.py，定义另外 1 条测试用例test()，import test_a后手动调用执行，从TepVar.vars_中读取name：

from tests.sample.case_reuse.a_test import test_a


def test(faker_ch, env_vars):
    test_a(faker_ch, env_vars)
    print(env_vars.get("name"))

注意test_a()函数参数引用的所有 fixture，需要在test()函数参数中重新申明一遍，比如示例中的(faker_ch, env_vars)。

tep 是借助conftest.py和fixture，用类变量来做的，参考了 JMeter BeanShell 的vars。

也可以外接redis。

tep0.6.0 更新

tep 用户手册是根据0.5.3版本写的，现在更新了0.6.0版本，一是支持本文提到的 3 个级别复用，微调了fixtures，二是把项目脚手架代码尽量封装到tep中，减少后续升级成本。涉及到 4 个文件：conftest.py、fixture_admin.py、fixture_login.py、fixture_your_name.py。

更新 tep

pip install tep==0.6.0

• 如果还没有用过 tep，使用这条命令安装后，执行tep startproject project_name就可以创建0.6.0版本的项目脚手架，包含了预置代码结构和 sample 测试用例。
• 如果已经安装了 tep，也需要执行这条命令升级0.6.0版本。

老项目升级（可选）

老项目升级也很简单，先在项目根目录执行命令：

tep -U

它会自动替换conftest.py，考虑到老项目fixtures已经有真实数据了，这里给出了version0.6.0文件包，命令执行后在项目根目录生成：

把老项目的真实数据提取出来，手动替换一下就可以了：

小结

本文介绍了如何使用pytest测试框架实现变量、接口、用例 3 个级别复用，给出了在tep中的使用示例，然后提供了安装升级tep0.6.0的方法。这套思路借鉴了JMeter和HttpRunner的部分设计思想，以及我使用过的接口测试平台经验，在 pytest 上实践了一下。tep 只是测试工具，本质上还是写 pytest，感兴趣可以试一下哦。

参考资料：

源码 https://github.com/dongfanger/tep