一、Locust 简介

        Locust又名蝗虫，寓意需要较小的硬件资源，提供大量级的并发访问量，是一个基于事件的负载工具，使用协程的方式，Locust 0.14.6以后的版本必须要使用Python 3.6及以上的版本，Locust 0.* 小于等于0.14.6，Locust 1.* 大于等于1.0，且2个版本类型的用法不兼容，执行命令差异，同一个物理环境不能同时存在Locust 0.和Locust 1.，Locust相关资源的下载地址。

        Python官网：https://www.locust.io
        Microsoft Visual C++ 14.0：https://www.python.org/downloads/release/python-380/

二、Python国内镜像源

        阿里云：https://mirrors.aliyun.com/pypi/simple
        清华大学：https://pypi.tuna.tsinghua.edu.cn/simple
        中国科技大学：<u>https://pypi.mirrors.ustc.edu.cn/simple</u>

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com

(1) Linux环境

        修改 ~/.pip/pip.conf (没有就创建一个文件夹及文件。文件夹要加“.”，表示是隐藏文件夹)，文件内容如上所示。

(2) Windows环境

        直接在 %userprofile% 目录中创建一个 pip目录，再新建文件 pip.ini，文件内容如上所示。

三、Locust 环境

#!/usr/bin/python3

import os
from locust import HttpLocust,between,seq_task,task
from locust import TaskSequence #locust 1.*没有这个方法。

#性能测试用例管理类，TaskSequence父类的作用是管理执行顺序。
#locust 1.*版本父类就变成了SequentialTaskSet
class TestSuite(TaskSequence):
    #初始化方法，相当于__init__函数，在用例之前执行。
    def on_start(self):
        pass

    #最后执行的方法，相当于teardown，在用例之后执行。
    def on_stop(self):
        pass
    
    #测试用例模版
    @task    #表示这个方法是一个任务，@task(1)表示执行的次数。
    @seq_task(2) #表示执行的顺序，locust 1.*的顺序是按照代码位置排序，不需要这个装饰器。
    def case_demo(self):
        url = "/downloads/release/python-380"
        header = {"user-agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"}
        data = {"user":"muge","pwd":"123456"}

        #self.count相当于requests,catch_response表示允许失败，name设置任务标签。
       #注意json=data与data=json.dumps(data)是表达的意思相同。
        response = self.count.post(url,headers=header,json=data,catch_response=True,name="case_demo")
        if response.status_code == 200:
            response.success()
        else:
            response.failure("request interface failure!")

        """
        #locust 1.*版本的请求尽量按照以下方式写，不然会存在兼容性问题。
        with self.count.post(url,headers=header,json=data,catch_response=True,name="case_demo") as response:
            if response.status_code == 200:
                response.success()
            else:
                response.failure("request interface failure!")
        """


#性能测试用例执行类
class RunCase(HttpLocust):
    #task_set在locust 1.*版本已被移除换成了tasks且不需要导入，必须使用列表或字典。
    task_set = TestSuite #指向测试类，也可以是[TestSuite1，TestSuite2]
    wait_time = between(0.1,3) #定义用例执行过程中的随机等等事件单位是秒。
    host = "https://www.python.org:8080" #定义被测服务器，也可以在执行命令中用-H ip:port表现。




if __name__ == '__main__':
    pass
    # os.system("locust -f locustmode.py")