使用Locust快速完成高并发测试

一纸荒年

简介

Locust是一款易于使用的分布式用户负载测试工具。它用于对网站（或其他系统）进行负载测试，并确定系统可以处理多少并发用户。

特点

• 使用python编写测试脚本
• 支持模拟数十万用户
• 基于web的操作界面
• 可以测试任何网站
  

安装

系统centos 7

1. pip install locustio

复制代码

示例代码

1. from locust import HttpLocust, TaskSet, task
   
2. 
   
3. class UserBehavior(TaskSet):
   
4.     def on_start(self):
   
5.         """ on_start is called when a Locust start before any task is scheduled """
   
6.         self.login()
   
7. 
   
8.     def on_stop(self):
   
9.         """ on_stop is called when the TaskSet is stopping """
   
10.         self.logout()
    
11. 
    
12.     def login(self):
    
13.         self.client.post("/login", {"username":"ellen_key", "password":"education"})
    
14. 
    
15.     def logout(self):
    
16.         self.client.post("/logout", {"username":"ellen_key", "password":"education"})
    
17. 
    
18.     @task(2)
    
19.     def index(self):
    
20.         self.client.get("/")
    
21. 
    
22.     @task(1)
    
23.     def profile(self):
    
24.         self.client.get("/profile")
    
25. 
    
26. class WebsiteUser(HttpLocust):
    
27.     weight = 1
    
28.     task_set = UserBehavior
    
29.     min_wait = 5000
    
30.     max_wait = 9000
    
31.     host=https://google.com
    
32.    
    
33. class MobileUserLocust(HttpLocust):
    
34.     weight = 3
    
35.     task_set = UserBehavior
    
36.     min_wait = 5000
    
37.     max_wait = 9000
    
38.     host=https://google.com
    

复制代码

Locust类参数(MobileUserLocust)

task_set：指定定义用户行为的类,包含一组任务
min_wait：最小的等待时间，毫秒
max_wait：最大的等待时间，毫秒 两者为声明则默认为1秒
host：加载的主机的URL前缀
weight：运行的次数比例
TaskSet类参数

@task(1)：任务装饰器，参数为运行次数的比例

TaskSequence类

TaskSequence类是一个TaskSet，但它的任务将按顺序执行.

1. from locust import TaskSequence, task
   
2. class MyTaskSequence(TaskSequence):
   
3.     @seq_task(1)
   
4.     def first_task(self):
   
5.         pass
   
6. 
   
7.     @seq_task(2)
   
8.     def second_task(self):
   
9.         pass
   
10. 
    
11.     @seq_task(3)
    
12.     @task(10)
    
13.     def third_task(self):
    
14.         pass
    

复制代码

在上面的示例中，顺序被定义为执行first_task，然后执行second_task，最后执行third_task 10次

启动测试

1. //启动指定文件
   
2. locust -f ./locust_test.py
   
3. 
   
4. //启用2个locust
   
5. locust -f locust_file.py WebUserLocust MobileUserLocust

复制代码

Miss_love

感谢分享

qqq911

感谢分享