pytest压力测试:不断发送数据,直到发现数据丢失

示例场景

假设有一个 send_data 函数接受数据并返回成功或失败的状态。

创建一个测试用例,通过逐步增加数据量来测试这个函数,直到返回失败为止。

步骤

  1. 定义压力测试函数

    定义一个函数。不断发送数据,直到发现数据丢失。

  2. 创建 pytest 测试用例

    使用 pytest 运行这个测试函数,记录每一步的结果。

运行成功效果

python 复制代码
import pytest
import time


# 假设这是你的发送数据函数
def send_data(att, data):
    # 模拟发送数据并返回状态
    # 在实际代码中,你应该调用你要测试的服务
    if data > 2400000:  # 设定一个阈值作为测试条件
        assert False , f"Data loss detected with number_of_packets :第{att}次,长度增加到 {data} "
    return True


def test_stress():
    number_of_packets = 1400  # 初始数据
    step_number = 10000       # 每次增加的数据量
    max_attempts = 4          # 最大尝试次数,防止无限循环
    attempts = 0

    while attempts < max_attempts:
        # assert send_data(data): 如果 send_data(data) 返回 False,断言失败,抛出异常并退出循环
        assert send_data(attempts,number_of_packets), f"Data loss detected with number_of_packets :第{attempts}次,长度增加到 {number_of_packets} "

        # 增加数据量
        number_of_packets += step_number
        attempts += 1

        # 可以选择等待一段时间,模拟现实环境中的间隔
        time.sleep(1)  # 延时1秒

    print(f"No data loss detected within the test limits. 每次增加的数据量{step_number},最大尝试次数{max_attempts},最大长度{number_of_packets}")

PASSED [100%]No data loss detected within the test limits. 每次增加的数据量10000,最大尝试次数4,最大长度41400

运行失败效果

python 复制代码
import pytest
import time


# 假设这是你的发送数据函数
def send_data(att, data):
    # 模拟发送数据并返回状态
    # 在实际代码中,你应该调用你要测试的服务
    if data > 24000:  # 设定一个阈值作为测试条件
        assert False , f"Data loss detected with number_of_packets :第{att}次,长度增加到 {data} "
    return True


def test_stress():
    number_of_packets = 1400  # 初始数据
    step_number = 10000       # 每次增加的数据量
    max_attempts = 4          # 最大尝试次数,防止无限循环
    attempts = 0

    while attempts < max_attempts:
        # assert send_data(data): 如果 send_data(data) 返回 False,断言失败,抛出异常并退出循环
        assert send_data(attempts,number_of_packets), f"Data loss detected with number_of_packets :第{attempts}次,长度增加到 {number_of_packets} "

        # 增加数据量
        number_of_packets += step_number
        attempts += 1

        # 可以选择等待一段时间,模拟现实环境中的间隔
        time.sleep(1)  # 延时1秒

    print(f"No data loss detected within the test limits. 每次增加的数据量{step_number},最大尝试次数{max_attempts},最大长度{number_of_packets}")
相关推荐
Mango0000008 分钟前
香港站群服务器有助于提升网站在搜索引擎中的排名
运维·服务器·搜索引擎
嚯——哈哈10 分钟前
筑起数字堡垒:解析AWS高防盾(Shield)的全面防护能力
服务器·微服务·云计算·aws
humors22114 分钟前
阿里云ECS服务器监控报警配置
运维·服务器·安全·阿里云·云计算
杨江16 分钟前
ThingsBoard安装测试
服务器·数据库
小林熬夜学编程1 小时前
【Linux系统编程】第五十弹---构建高效单例模式线程池、详解线程安全与可重入性、解析死锁与避免策略,以及STL与智能指针的线程安全性探究
linux·运维·服务器·c语言·c++·安全·单例模式
Linux运维老纪1 小时前
交换机配置从IP(Switch Configuration from IP)
linux·服务器·网络·安全·运维开发·ip
加载中loading...1 小时前
C/C++实现tcp客户端和服务端的实现(从零开始写自己的高性能服务器)
linux·运维·服务器·c语言·网络
大筒木老辈子1 小时前
Linux笔记---进程:初识进程
linux·服务器·笔记
木木em哈哈1 小时前
archlinux安装waydroid
linux·运维·服务器
程序猿阿伟2 小时前
《进程隔离机制:C++多进程编程安全的坚固堡垒》
服务器·c++·安全