本文的文字及图片来源于网络,仅供学习、交流使用,不具有任何商业用途,,版权归原作者所有，如有问题请及时联系我们以作处理
作者：dongfanger 来源：掘金
链接：https://juejin.cn/post/6910414324712996872
想要获取更多Python学习资料，了解更多关于Python的知识，可以加Q群630390733踊跃发言，大家一起来学习讨论吧！
tep的设计理念是让人人都可以用Python写自动化，本文就来介绍如何用tep完成增删改查接口自动化。
环境变量
编辑fixtures/fixture_admin.py：
"qa": {
"domain": "https://qa.com",
},
复制代码
修改qa环境的domain。
登录
因为非登录接口需要从登录接口拿token，放在请求参数中，所以先在fixtures/fixture_admin.py中实现登录，修改url和json：
response = request(
"post",
url=url("/api/users/login"),
headers={"Content-Type": "application/json"},
json={
"username": "admin",
"password": "123456",
}
)
复制代码
根据实际响应数据结构，修改response_token赋值：
response_token = jmespath.search("token", response.json())
class Clazz:
token = response_token
jwt_headers = _jwt_headers(response_token)
return Clazz
复制代码
参考_jwt_headers()实现自定义headers：
def _jwt_headers(token):
return {"Content-Type": "application/json", "authorization": f"Bearer {token}"}
复制代码
_表示内部函数，外部无法访问，遵循conftest.py只对外提供fixture的原则。
完整代码
def _jwt_headers(token):
return {"Content-Type": "application/json", "authorization": f"Bearer {token}"}
@pytest.fixture(scope="session")
def login():
# 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
复制代码
说明：
可以复制login为多个fixture，如login_admin、login_some_user，灵活运用。
scope="session"，表示只登录一次，所有测试用例使用同一个token。可以改为function，让每条用例使用不同token。
写用例
新建测试
新建tests\crud_test.py：
from loguru import logger
from tep.client import request
def test(faker_ch, login, url):
复制代码
pytest的fixture作为参数传入test()函数来使用，faker_ch、login、url是fixture_admin.py中定义好的fixture。
from loguru import logger用于在测试用例中打印日志
from tep.client import requesttep封装了请求日志功能，也可以用原生from requests import request
新增
请求参数为nickname和phone，使用faker_ch造1条测试数据：
fake = faker_ch
nickname = fake.name()
phone = fake.phone_number()
复制代码
请求方法为post，headers取登录返回值login.jwt_headers：
response = request(
"post",
url=url("/api/users"),
headers=login.jwt_headers,
json={
"nickname": nickname, "phone": phone
}
)
复制代码
添加断言，简单判断下响应状态码<400：
assert response.status_code < 400
复制代码
也可以查数据库来断言。
提取修改接口需要的数据：
user_id = jmespath.search("id", response.json())
created_at = jmespath.search("createdAt", response.json())
updated_at = jmespath.search("updatedAt", response.json())
复制代码
推荐用jmespath来提取json。
查询
请求参数传入刚才定义的局部变量nickname，使用get请求并断言，headers取登录返回值login.jwt_headers：
response = request(
"get",
url=url("/api/users"),
headers=login.jwt_headers,
params={
"page": 1,
"perPage": 10,
"keyword": nickname
}
)
assert response.status_code < 400
复制代码
get请求需要把json关键字改为params。
修改
使用faker再造1条新数据：
nickname_new = fake.name()
phone_new = fake.phone_number()
复制代码
请求方法为put，headers取登录返回值login.jwt_headers：
response = request(
"put",
url=url(f"/api/users/{user_id}"),
headers=login.jwt_headers,
json={
"id": user_id, "createdAt": created_at, "updatedAt": updated_at,
"phone": phone_new, "nickname": nickname_new
}
)
assert response.status_code < 400
复制代码
请求参数中用到了新增接口提取的数据user_id、created_at、updated_at。
删除
请求方法为delete，url中传入user_id，headers取登录返回值login.jwt_headers：
response = request(
"delete",
url=url(f"/api/users/{user_id}"),
headers=login.jwt_headers
)
assert response.status_code < 400
复制代码
删除接口没有json和params。
完整用例
"""
@Author : Don
@Date : 12/25/2020 1:02 PM
@Desc : 增删改查
"""
import jmespath
from loguru import logger
from tep.client import request
def test(faker_ch, login, url):
fake = faker_ch
logger.info("新增")
nickname = fake.name()
phone = fake.phone_number()
response = request(
"post",
url=url("/api/users"),
headers=login.jwt_headers,
json={
"nickname": nickname, "phone": phone
}
)
assert response.status_code < 400
user_id = jmespath.search("id", response.json())
created_at = jmespath.search("createdAt", response.json())
updated_at = jmespath.search("updatedAt", response.json())
logger.info("查询")
response = request(
"get",
url=url("/api/users"),
headers=login.jwt_headers,
params={
"page": 1,
"perPage": 10,
"keyword": nickname
}
)
assert response.status_code < 400
logger.info("修改")
nickname_new = fake.name()
phone_new = fake.phone_number()
response = request(
"put",
url=url(f"/api/users/{user_id}"),
headers=login.jwt_headers,
json={
"id": user_id, "createdAt": created_at, "updatedAt": updated_at,
"phone": phone_new, "nickname": nickname_new
}
)
assert response.status_code < 400
logger.info(f"用户姓名手机 {nickname} {phone} 修改后 {nickname_new} {phone_new}")
logger.info("删除")
response = request(
"delete",
url=url(f"/api/users/{user_id}"),
headers=login.jwt_headers
)
assert response.status_code < 400
复制代码
小结
本文介绍了tep的基本使用，先配置环境变量，再修改登录代码，然后新增测试，最后编写增删改查的接口请求。每条用例放在一个函数中，通过函数参数引用fixture来使用全局环境变量，函数内部可以定义测试需要的局部变量。每个接口是一个代码块，由接口描述、测试数据、请求、断言和数据提取5部分组成。接口之间通过变量实现参数化和关联。