自己上手写性能测试工具 | 干货

依赖库

requests==2.22.0
gevent==20.9.0
numpy==1.19.2
click==7.1.2

click 库

今天的主角是click库。

中文文档:https://www.osgeo.cn/click/index.html

第一个例子(hello.py):

import click

@click.command()
@click.argument('name')
@click.option('-c', default=1, help='number of greetings')
def hello(name, c):
    for x in range(c):
        click.echo('Hello %s!' % name)


if __name__ == "__main__":
    hello()

查看帮助:

> python3 hello.py --help
Usage: hello.py [OPTIONS] NAME

Options:
  -c INTEGER  number of greetings
  --help      Show this message and exit.

使用:

> python3 hello.py 伤心的辣条 -c 3
Hello 伤心的辣条!
Hello 伤心的辣条!
Hello 伤心的辣条!

现在已经掌握了click 的基本用法。

实现命令行性能测试工具

接下来,将click引入到kb.py性能测试脚本中。

from __future__ import print_function
import gevent
from gevent import monkey
monkey.patch_all()
import time
import click
import requests
from numpy import mean


class statistical:
    pass_number = 0
    fail_number = 0
    run_time_list = []


def running(url, numbers):
    for _ in range(numbers):
        start_time = time.time()
        r = requests.get(url)
        if r.status_code == 200:
            statistical.pass_number = statistical.pass_number + 1
            print(".", end="")
        else:
            statistical.fail_number = statistical.fail_number + 1
            print("F", end="")

        end_time = time.time()
        run_time = round(end_time - start_time, 4)
        statistical.run_time_list.append(run_time)


@click.command()
@click.argument('url')
@click.option('-u', default=1, help='运行用户的数量,默认 1', type=int)
@click.option('-q', default=1, help='单个用户请求数,默认 1', type=int)
def main(url, u, q):
    print("请求URL: {url}".format(url=url))
    print("用户数:{},循环次数: {}".format(u, q))
    print("============== Running ===================")

    jobs = [gevent.spawn(running, url, q) for _url in range(u)]
    gevent.wait(jobs)

    print("\n============== Results ===================")
    print("最大:       {} s".format(str(max(statistical.run_time_list))))
    print("最小:       {} s".format(str(min(statistical.run_time_list))))
    print("平均:       {} s".format(str(round(mean(statistical.run_time_list), 4))))
    print("请求成功", statistical.pass_number)
    print("请求失败", statistical.fail_number)
    print("============== end ===================")


if __name__ == "__main__":
    main()

查看帮助:

python3 kb.py --help
Usage: kb.py [OPTIONS] URL

Options:
  -u INTEGER  运行用户的数量,默认 1
  -q INTEGER  单个用户请求数,默认 1
  --help      Show this message and exit.

使用方法:

> python3 kb.py https://wwww.baidu.com -u 10 -q 10

请求URL: https://wwww.baidu.com
用户数:10,循环次数: 10
============== Running ===================
....................................................................................................
============== Results ===================
最大:       0.955 s
最小:       0.2573 s
平均:       0.4585 s
请求成功 100
请求失败 0
============== end ===================

项目代码:https://github.com/SeldomQA/kb
在这里插入图片描述
另外,欢迎加入软件测试技术交流群 313782132 ~进群可领取免费软件测试资料以及群内测试大牛解惑!

测试工程师职业发展路线图

功能测试 — 接口测试 — 自动化测试 — 测试开发 — 测试架构师

加油吧,测试人!如果你需要提升规划,那就行动吧,在路上总比在起点观望的要好。事必有法,然后有成。

资源不错就给个推荐吧~

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值