Python—Locust 设置断言，巧遇请求都是200的情况，使用断言来操作。

马虎的程序猿

已于 2022-12-08 16:57:42 修改

阅读量599

点赞数

分类专栏：压力测试 python 文章标签： python 压力测试

于 2022-12-08 15:29:06 首次发布

本文链接：https://blog.csdn.net/weixin_45195493/article/details/128237608

版权

python 同时被 2 个专栏收录

17 篇文章 2 订阅

订阅专栏

压力测试

2 篇文章 0 订阅

订阅专栏

找了很多代码，测试断言效果。（会出现报错问题，非常难看哦😯）

   with self.client.post(url,headers=headers,json=query,catch_response=True) as response:
        if json.loads(response.text)['code'] == '0':
            response.success()
        else:
            response.failure('Failed!')

在这里插入图片描述

最后找到这个断言，非常好用，完美解决了问题

from json import JSONDecodeError
with self.client.post(url,headers=headers,json=query,catch_response=True) as response:
    if response.status_code != 200:
        response.failure("Did not get expected status code")
    try:
        if json.loads(response.text)['code'] != '0':
            response.failure("Did not get expected value in status")
    except JSONDecodeError:
        response.failure("Response could not be decoded as JSON")
    except KeyError:
        response.failure("Response did not contain expected key 'status'")