Pytest使用fixture实现token共享
同学们在做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
-
import pickle -
@pytest.fixture(scope="function") -
def login(): -
if os.path.exists(login_info): -
return get_login_info(login_info) -
else: -
...... -
# 增加缓存方法 -
save_token(login_info['token']) -
return login_info['token'] -
# 增加一个读取文件的方法 -
def get_login_info(): -
with open(login_path, 'rb') as f: -
data = pickle.load(f) -
f.close() -
return data
以上工作已经完成了90%，再来个收尾就完美了。每次执行用例结束后把存token文件删除，不然生成了一个token，代码判断token文件存在，就不会去调用登录接口。token一但过期，用例就会执行失败。
在conftest定义一个方法，每次执行用例都清理一下token文件。要注意级别是session级别哦。
-
@pytest.fixture(scope='session', autouse=True) -
def clear_login_file(): -
if os.path.exists(login_path): -
logger.info("清理token文件") -
os.remove(login_path)
总结：
感谢每一个认真阅读我文章的人！！！
作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。
软件测试面试文档
我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。
视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。