#coding:utf-8
from locust import HttpLocust,TaskSet,task
class BlogDemo(TaskSet):
'''用户行为：打开我的博客首页demo'''
@task(1)
def open_blog(self):
#定义requests的请求头
header= {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36"}
r=self.client.get("/yoyoketang", headers=header, verify=False)
print(r.status_code)
assert r.status_code == 200
class websitUser(HttpLocust):
task_set=BlogDemo
min_wait=3000
max_wait=6000
if __name__=='__main__':
import os
os.system("locust -f locustdemo.py --host=https://www.cnblogs.com")#-f 参数是指定运行的脚本
#—host是指定运行项目的host地址，这里用的https://www.cnblogs.com，代码里面get访问的是"/yoyoketang"，拼接起来就是完整地址了
'''代码注解：
新建一个类BlogDemo(TaskSet)，继承TaskSet，该类下面写一些准备请求的行为（访问的接口）
里面的self.client调用get和post方法，跟requests是一样的
@task装饰该方法表示为用户行为。括号里面参数表示该行为挑选执行的权重，数值越大，执行频率越高，不设置默认是1
WebsiteUser()类用于设置性能测试。
task_set ：指向一个定义了的用户行为类。
min_wait ：用户执行任务之间等待时间的下界，单位：毫秒。
max_wait ：用户执行任务之间等待时间的上界，单位：毫秒。'''
