Python:教你用几十行代码实现系统的性能自动化测试

前言

在开发中，我们有时会碰到，相同的代码名明明测试环境没有问题，上了生产就出现各种“幺蛾子”，不是连接超时，就是数据有问题，去检查代码也不觉得有问题。然后查看系统监控发现，出问题的时间段，往往系统的cpu使用率、内存占有率都比较高。如果我们预估上线后, 网站或者接口的访问量, 提前对接口或者网站做压力测试，提前评估下可能会出现的情况，结果就不一样了。

关注，转发，私信小编“01”即可免费领取Python入门学习资料！

性能测试是通过自动化的测试工具模拟多种正常、峰值以及异常负载条件来对系统的各项性能指标进行测试。负载测试和压力测试都属于性能测试，两者可以结合进行。通过负载测试，确定在各种工作负载下系统的性能，目标是测试当负载逐渐增加时，系统各项性能指标的变化情况。压力测试是通过确定一个系统的瓶颈或者不能接受的性能点，来获得系统能提供的最大服务级别的测试。

一、Locust介绍和安装

Locust 是一个用 Python 编写的开源的负载测试工具。它允许您针对模拟用户行为的 Web 应用程序编写测试，然后按规模运行测试以帮助查找瓶颈或其他性能问题。

安装:pip install locust

检查是否安装成功:locust -help

二、Locust的源码

创建unctionTest.py，编写源码

1、创建一个方法WebsiteTasks(TaskSet)并继承TaskSet

2、编写启动执行项on_start(self), 改启动项只会启动的时候被执行一次。

3、添加执行的任务@task(5), 方法的参数用于指定该行为的执行权重，参数越大每次被虚拟用户执行的概率越高，默认为1。

4、创建 WebsiteUser(HttpLocust)继承HttpLocust, 配置Locust执行的一些配置参数。

源码如下：

三、开发自动化性能测试

1、执行源码:

locust -f D:\javacode\hk03\webPageTest\src\性能自动化测试\functionTest.py

2、浏览器访问locust:

http://localhost:8089/

Number of users to simulate 设置并发的总用户数

Hatch rate (users spawned/second) 设置每秒产生的用户数

点击start swmarming开始运行

3、运行结果

统计页面如下:

图表页面:

每秒请求总数统计

请求响应的时间统计:

请求用户数的统计

故障统计

还可以下载统计的结果

完美!!!!!!!!!!!!!!!!

四、一些扩展

no-web模式运行启动

终端中–>进入代码目录：>> locust -f xxoo.py –no-web -c10 -r2 -t 1m

启动参数：–no-web 表示不使用web界面运行测试。 -c 设置虚拟用户数 。 -r 设置每秒启动虚拟用户数 。 -t 设置运行时间.。

no-web模式运行将测试结果保存到当前.py目录中：locust -f xxoo.py –csv=起一个名字

例如：locust -f test3.py –csv=foobar –no-web -c2 -t10s

分布式压测：

主从机中必须运行相同的测试代码（把主机中代码复制一份到多个从机中），主机负责收集测试数据，从机进行施压测试；

在主机终端中–>进入代码目录：>> locust -f xxxoo.py –master

从机中终端中–>进入代码目录：>> locust -f xxxoo.py –slave –master-host=主机ip

分布式压测no-web模式保存结果到主机中当前运行.py的目录中：>>locust -f test2.py –csv=foobartt –no-web -c2 -t10s –master

WebsiteTasks类中如何去调用 WebsiteUser(HttpLocust)类中定义的字段和方法呢？

通过在WebsiteTasks类中self.locust.xxoo xxoo就是我们在WebsiteUser类中定义的字段或方法；

代码如下

self.client 的理解

在Locust类中，静态字段client即客户端的请求方法，这里的client字段没有绑定客户端请求方法，因此在使用Locust时，需要先继承Locust类class HttpLocust（Locust），然后在self.client = HttpSession(base_url=self.host)绑定客户端请求方法；

对于常见的HTTP(s)协议，Locust已经实现了HttpLocust类，其self.client=HttpSession(base_url=self.host)，而HttpSession继承自requests.Session。因此在测试HTTP（s）的Locust脚本中，可以通过client属性来使用Python requests库的所 有方法，调用方式与 reqeusts完全一致。另外，由于requests.Session的使用，client的方法调用之间就自动具有了状态记忆功能。常见的场景就是，在登录系统后可以维持登录状态的Session，从而后续HTTP请求操作都能带上登录状态。