Pytest使用fixture实现token共享

美团程序员

于 2024-09-18 22:08:01 发布

阅读量194

点赞数 3

分类专栏：面试软件测试软件测试面试文章标签： pytest 软件测试软件测试面试职场和发展

本文链接：https://blog.csdn.net/qq_73332379/article/details/142345100

版权

软件测试同时被 3 个专栏收录

1052 篇文章 8 订阅

订阅专栏

软件测试面试

324 篇文章 3 订阅

订阅专栏

面试

290 篇文章 3 订阅

订阅专栏

同学们在做pytest接口自动化时，会遇到一个场景就是不同的测试用例需要有一个登录的前置步骤，登录完成后会获取到token，用于之后的代码中。首先我先演示一个常规的做法。

首先在conftest定义一个login的方法，方法返回toke
@pytest.fixture(scope="function")
def login():
header = {
"Content-Type": "application/json"
}
data = {
"username": "test",
"password": "test"
}
login_info = requests.post(url='http://.../login',json=data).json()
return login_info['token']

2.在测试用例方法中引入这个方法，所有用到token，都需要使用login这个方法。

def test_user(login):
token = login # 通过login获取token
......
def test_address(login):
token = login # 通过login获取token
......

通过上面的方法，可以解决接口使用token的需求，不过这么做每次都会先执行login接口，有没有一种方法可以只执行一次login接口，后面的接口就不调用login接口了呢。经过老武测试，其实是可以做到的。

思路：login接口使用缓存，当有token的时候就使用token，就不调用login接口了，如果没有则调用login接口，再把token缓存起来。

那需要对上面login的方法进行改造，增加save_token方法，将token存起来。引入pickle模块，用来写数据。

import pickle
@pytest.fixture(scope="function")
def login():
......
# 增加缓存方法
save_token(login_info['token'])
return login_info['token']
def save_token(token):
# login_path为目录中一个存放token的文件路径，自己定义。
with open(login_path, 'wb') as f:
pickle.dump(login_info, f)
f.close()

上面代码完成了token的写入，当没有token的时候token写入到文件中。如果有token，我们就不需要执行登录操作，直接使用文件中的token，这个时候代码需要继续改造。方法开始时判断当前是否存在保存token的文件，如果有则获取token，不再执行登录的操作。再增加一个读取文件内容的方法get_login_info