开源测试工具｜使用rust开发的python压测工具

电玩女神 2024-05-23 18:38 57阅读 0赞

![5ed80d7da6904639a76a02864c8beec2.png][]

[2024软件测试面试刷题，这个小程序（永久刷题），靠它快速找到工作了！（刷题APP的天花板）\_软件测试刷题小程序-CSDN博客文章浏览阅读3k次，点赞86次，收藏13次。你知不知道有这么一个软件测试面试的刷题小程序。里面包含了面试常问的软件测试基础题，web自动化测试、app自动化测试、接口测试、性能测试、自动化测试、安全测试及一些常问到的人力资源题目。最主要的是他还收集了像阿里、华为这样的大厂面试真题，还有互动交流板块……\_软件测试刷题小程序编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502编辑https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502![icon-default.png?t=N7T8][icon-default.png_t_N7T8]https://blog.csdn.net/AI\_Green/article/details/134931243?spm=1001.2014.3001.5502][2024_APP_-CSDN_3k_86_13_web_app_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_icon-default.png_t_N7T8_icon-default.png_t_N7T8_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502]

介绍分享一个开源测试工具，atomic-bomb-engine-py:使用rust开发的python压测工具，这个工具是由 qyzhg 同学在TesterHome社区网站的分享。

### **项目背景** ###

公司的原有压测平台是由 go 开发，使用 locust 作为压力引擎使用，在互联网的大环境下，开始各种降本增效，性能测试的 pod 现在缩减到只有 1/8 个物理核心，这种情况下，locust 大概 300 并发都已经出现了 cpu 瓶颈，所以需要一款性能更好的压测引擎作为替代品，刚开始的时候是想直接使用 wrk 集成进项目中，但是遇见了几个比较麻烦的问题:

*  post 请求需要使用 lua
 *  无法对压测过程进行监听
 *  项目过于庞大，二开十分困难

所以经过调研现有的开源压测引擎，没有符合现在的平台化需求的，所以诞生了这个项目，之所以选用 rust，是因为 rust 的性能会非常的好，而且高并发的压测下，没有 gc 对结果产生影响，引擎写完后，可以直接导出一个 c 的入口，go 开启 cgo 会比较简单的将项目集成。

由于引擎是一个独立的项目，无公司的业务部分，可以直接开源。介于现在大部分测试人员的技术栈都是以 python 为主，所以又在引擎外面，使用 pyo3 开发了一个 python 的包装器，可以让更多 python 技术栈的同学直接调用，为了方便大家使用，又开发了一个比较简单的前端页面，可以满足简单的压测需求。

### **项目地址** ###

*  python 包部分：https://github.com/qyzhg/atomic-bomb-engine-py
 *  引擎部分：https://github.com/qyzhg/atomic-bomb-engine
 *  前端部分：https://github.com/GiantAxeWhy/atomic-bomb-engine-front

项目已发布到 pip，可以直接使用 pip 安装使用，ci 部分使用 github actions 进行矩阵编译，支持 python3.8-3.12，linux-x86, mac arm、x86, win-x86，应该可以覆盖大部分环境，如果有特殊需求，可以联系作者添加 action

### **项目内部架构设计图** ###

![图片][2e828a2dfaa7fec9d3752d2faedcce43.png]

### **项目界面** ###

![图片][f17b6090a0db718d53be6c3bd3ea588b.png]

### **使用说明** ###

#### **准备开始** ####

安装：

pip install atomic-bomb-engine

在 python 中导入

import atomic_bomb_engine

异步使用的时候，还需要引用 asyncio

import asyncio

#### **主要方法说明** ####

多接口压测可以使用 batch\_async 方法进行操作，函数签名和解释如下

async def batch_async(
                test_duration_secs: int,
                concurrent_requests: int,
                api_endpoints:List[Dict],
                step_option:Dict[str, int]=None,
                verbose:bool=False,
                should_prevent:bool=False) ->Dict:
       """
           批量压测
           :param test_duration_secs: 测试持续时间
           :param concurrent_requests: 并发数
           :param api_endpoints: 接口信息
           :param step_option: 阶梯加压选项
           :param verbose: 打印详细信息
           :param should_prevent: 是否禁用睡眠
       """

使用 assert\_option 方法可以返回断言选项字典

assert_options=[atomic_bomb_engine.assert_option("$.code", 429),atomic_bomb_engine.assert_option("$.code", 200)])print(result)

jsonpath 如果不会用的话，建议去jsonpath学习  
使用 step\_option 方法可以返回阶梯加压选项字典

def step_option(increase_step: int, increase_interval: int) -> Dict[str, int]:
       """
       生成step option
       :param increase_step: 阶梯步长
       :param increase_interval: 阶梯间隔
       """

同样的本包中也包含了一个对 api\_endpoint 的包装：endpoint 方法，方便调用，endpoint 中的 assert\_options 中也可以套用 assert\_option 方法

async def run_batch():
       result = await atomic_bomb_engine.batch_async(
           test_duration_secs=10,
           concurrent_requests=10,
           api_endpoints=[
               atomic_bomb_engine.endpoint(
                   name="test1",
                   url="https:xxxxx1.xx",
                   method="get",
                   weight=1,
                   timeout_secs=10,
                   assert_options=[atomic_bomb_engine.assert_option(jsonpath="$.code", reference_object=200)]
               ),
               atomic_bomb_engine.endpoint(
                   name="test2",
                   url="https://xxxxx2.xx",
                   method="get",
                   weight=1,
                   timeout_secs=10)
           ])
       print(result)

监听时可以使用 BatchListenIter 生成器

async def listen_batch():
       iterator = atomic_bomb_engine.BatchListenIter()
       for message in iterator:
           if message:
               print(message)
           else:
               await asyncio.sleep(0.3)

同时调用时同单接口

async def main():
       await asyncio.gather(
           run_batch(),
           listen_batch(),
       )if __name__ == "__main__":
       asyncio.run(main())

#### **使用内置 ui 界面** ####

导入

from atomic_bomb_engine import server

导入内置的 server 后，可以使用内置的 http 服务器，开启一个 ui 界面，并且开始监听压测过程中的数据，无需手动迭代数据

import asyncioimport atomic_bomb_enginefrom atomic_bomb_engine import server@server.ui(port=8000)async def run_batch():
       result = await atomic_bomb_engine.batch_async(
           test_duration_secs=120,
           concurrent_requests=100,
           step_option=atomic_bomb_engine.step_option(increase_step=6, increase_interval=5),
           verbose=False,
           api_endpoints=[
               atomic_bomb_engine.endpoint(name="test-baidu",url="https://baidu.com",method="GET",weight=1,timeout_secs=10),
               atomic_bomb_engine.endpoint(name="test-google", url="https://google.com", method="GET", weight=1, timeout_secs=10),
           ])
       print(result)
       return resultif __name__ == '__main__':
       asyncio.run(run_batch())

### **BUG 和需求** ###

由于项目开启时间较短，只有短短一个月左右，可能会有一些 bug 和没有做的需求，如果发现了 bug 和需求，都可以联系作者，由于是工作之余开发，可能不会太保证时效性

行动吧，在路上总比一直观望的要好，未来的你肯定会感谢现在拼搏的自己！如果想学习提升找不到资料，没人答疑解惑时，**请及时加入群： 759968159**，里面有各种测试开发资料和技术可以一起交流哦。

**最后：** 下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取**`【保证100%免费】`**

![06b7bc1a690d4c4f826248342215214c.png][]

**软件测试面试文档**

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。![在这里插入图片描述][cc56d6ee4baf482688db0e1410e2bb51.png]

![在这里插入图片描述][755123cb1321421b96888db976754489.png]

![3010ebc19b1042d4a4e99e0c9334bb1c.gif][]

[5ed80d7da6904639a76a02864c8beec2.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/da1f32294921492cbae1b0559175859c.png
[icon-default.png_t_N7T8]: https://csdnimg.cn/release/blog_editor_html/release2.3.6/ckeditor/plugins/CsdnLink/icons/icon-default.png?t=N7T8
[2024_APP_-CSDN_3k_86_13_web_app_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502_icon-default.png_t_N7T8_icon-default.png_t_N7T8_https_blog.csdn.net_AI_Green_article_details_134931243_spm_1001.2014.3001.5502]: https://blog.csdn.net/AI_Green/article/details/134931243?spm=1001.2014.3001.5502
[2e828a2dfaa7fec9d3752d2faedcce43.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/4cf2ebad24e149378ee8c5f6cd8397f6.png
[f17b6090a0db718d53be6c3bd3ea588b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/44f9064c046e4ab2a304c98f7665e304.png
[06b7bc1a690d4c4f826248342215214c.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/f338c2d2ea874048835b827cf6feeb50.png
[cc56d6ee4baf482688db0e1410e2bb51.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/47f5680b8d60451cbe78a349db573d7b.png
[755123cb1321421b96888db976754489.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/e1a947b6d9f54bd686840c86ec09cc4a.png
[3010ebc19b1042d4a4e99e0c9334bb1c.gif]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/23/856f35dd01394d3db158c1bf425e330f.gif