1 # coding=utf-8
 2 import os
 3 
 4 import requests
 5 from locust import HttpUser, task, constant
 6 from requests.packages.urllib3.exceptions import InsecureRequestWarning
 7 
 8 # 禁用安全请求警告
 9 requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
10 
11 
12 class MyLocust(HttpUser):
13     # 设置每次请求间隔时间
14     # wait_time = between(1, 1)
15     wait_time = constant(1)
16 
17     # 前置处理
18     def on_start(self):
19         print("开始调用")
20 
21     # 后置处理
22     def on_stop(self):
23         print("停止调用")
24 
25     # 权重,执行比例
26     @task(1)
27     # 待测试的方法
28     def test(self):
29         # 请求的接口url
30         req = self.client.head("/xxxxx", verify=False)
31         if req.status_code == 200:
32             print("success")
33         else:
34             print("fails")
35 
36 
37 if __name__ == "__main__":
38     # 启动并设置域名地址,,也可以在启动后web端设置
39     # 启动后的web端页面 http://localhost:8089/
40     os.system("locust -f locust_demo.py --host=https://www.baidu.com/")