# coding:utf-8
from locust import task, HttpUser, between
import os
class WebUser(HttpUser): # 任务集 用户行为脚本
wait_time = between(0, 0) #(思考时间)
@task() # 权重
def index(self):
response = self.client.get("/", name='百度')
if response.status_code == 200:
return True
else:
print(response.text)
class User(WebUser): #(模拟从谷歌浏览器请求)
user_agent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.106 Safari/537.36'
class MobileUser(WebUser): #(模拟从手机端请求)
user_agent = 'Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_0 like Mac OS X; en-us) AppleWebKit/532.9 (KHTML, like Gecko) Version/4.0.5 Mobile/8A293 Safari/6531.22.7'
if __name__ == "__main__":
# 直接在cmd下面启动程序
os.system("locust -f locust_test.py --host=http://www.baidu.com")