基于Python代码构造的压力测试工具

测试界霄霄

于 2024-08-30 20:25:18 发布

阅读量587

点赞数 8

分类专栏：软件测试文章标签： python 测试工具开发语言软件测试功能测试自动化测试程序人生

本文链接：https://blog.csdn.net/2301_78276982/article/details/141726552

版权

软件测试专栏收录该内容

1000 篇文章 28 订阅

订阅专栏

Locust 是一个开源负载测试工具，通过 Python 形式的代码构造来定义用户行为，而不是与混乱的 UI 和臃肿的 XML 配置纠缠在一起。

先决条件

Python（+ pip 用于安装 locust 包）

任何模拟负载测试的 HTTP 服务。（我们将创建一个小型 Spring Boot HTTP 服务作为本博客演示的一部分

演示 Spring Boot HTTP 服务

我们通过 Spring Boot 创建了一个简单的 REST API，如下所示，它有一个简单的端点/greet，它不执行 CPU 或 I/O 密集型工作，而只是返回一个字符串。该服务在端口 8080 上本地运行，我们将在该端口上执行负载测试。

示例 Spring Boot HTTP 代码


import org.springframework.boot.SpringApplication;

import org.springframework.boot.autoconfigure.SpringBootApplication;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.RestController;


@SpringBootApplication

public class SpringBootLoadServiceApplication {


public static void main(String[] args) {

SpringApplication.run(SpringBootLoadServiceApplication.class, args);

}


}


@RestController

class RandomResource {


@GetMapping(value = "/greet")

public String greet() {

return "Locust Load Test.";

}

}

Locust设置

像往常一样，我们将创建并激活虚拟环境以避免将来使用时出现包冲突。

python -m venv virtualenv

激活虚拟环境：

source virtualenv/Scripts/activate.bat

设置虚拟环境后，我们将使用 pip 仅安装 locust 所需的软件包，如下所示：

pip install locust

安装locust后，验证以下是否显示版本以证明安装成功：

locust -V

如果如下所示，我们就可以开始使用 Locust 进行负载测试了。

Locust版本验证

设置 Locust 脚本文件

现在我们将编写一个 Python 代码（springboot_ locust.py）来模拟负载测试，该测试将到达我们的本地端点，如下所示。

代码片段


from locust import HttpUser, task, between #this is self-explanatory we importing required classes, function from locust module.


class SpringBootUser(HttpUser): #Simulates a single Real-world User. HttpUser class is extended so we provide what task the user will do.


'''this won't take effect in our scenario, as this attribute is optional and specify wait time randomly picked between parameters passed (1-3 seconds) between multiple @task executed. We have only one @task executed'''

wait_time = between(1, 3)


'''This is main core functionality of locust. @task are treated as a singel http request made.

We can multiple have @task. In below we are going to hit /greet endpoint on our localhost'''

@task

def hello_world(self):

self.client.get("/greet")

添加注释以确保代码是不言自明的。我们只是定义一个任务，一个任务是一个微线程的单次执行。我们也可以有多个任务，以模拟流程和权重来确定它们的执行顺序，否则它们将被随机选取。

现在我们将开始 Locust 测试，通过执行以下命令来启动 Locust Web UI：