# -*- coding: utf-8 -*-
# @Author   : supassxu
# @time     : 2021/4/28 16:51
# @File     : locustfile.py.py
# @Software : interface_test_dataqin

from locust import Locust, TaskSet, task, HttpUser


class MyTasks(TaskSet):
    """
    创建测试任务类，需要继承TaskSet
    可以添加多个测试任务
    """

    # 每个测试任务，往往会以实例方法的形式来呈现
    # 同时需要使用task装饰器来装饰测试任务
    @task
    def baidu(self):
        url = '/huahuage/p/12917114.html'
        header = {
            "User-Agent": "Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36"}
        req = self.client.get(url, headers=header, verify=False)

        if req.status_code == 200:
            print("success")
        else:
            print("fails")


class websitUser(HttpUser):
    """
    创建运行测试类，需要继承Locust父类
    """
    tasks = [MyTasks]  # 指定测试任务类，使用task_set覆盖父类的类属性
    min_wait = 2000  # 指定启动任务间隔的时间范围（单位毫秒）：2~5秒之间
    max_wait = 5000  # 使用min_wait、max_wait覆盖父类的类属性


if __name__ == "__main__":
    import os

    os.system("locust -f locustfile.py --host=https://www.cnblogs.com")