05-Locust-设置断言

1. # -*- coding: utf-8 -*-
2. # @Time    : 2021/4/10
3. # @Author  : 大海
4. 
5. import os
6. from locust import HttpUser, task, TaskSet, between
7. 
8. 
9. class MyUser(TaskSet):
10. 
11.     @task
12. def my_task(self):
13. # catch_response=True, 允许该请求被标记为失败
14. with self.client.get('/', catch_response=True) as response:
15. # 断言，如果响应状态码为200,通过；否则失败
16. if response.status_code == 200:
17.                 response.success()
18. else:
19.                 response.failure('Failed!')
20. 
21. 
22. class User(HttpUser):
23.     tasks = [MyUser]
24.     wait_time = between(3, 25)
25.     host = "https://www.baidu.com"
26. 
27. 
28. if __name__ == '__main__':
29.     file_path = os.path.abspath(__file__)
30.     os.system(f'locust -f {file_path} --web-host=127.0.0.1')