locust(四) - HTTP请求的压力测试

在前面几篇文章中,只介绍了如何创建Locust类、TaskSet类去模拟用户,本文介绍如何对提供HTTP接口的系统进行压力测试。Locust已经有HttpLocust类,每个HttpLocust类的实例都有一个client属性用于构造HTTP请求。

HttpLocust类

一个HttpLocust类的实例可以表示一组用于HTTP压力测试的“用户”,该用户的行为由task_set属性定义,这些与父类Locust类都一致。

HttpLocust类有个额外的client属性,用于建立与保持HTTP请求会话。熟悉Python的朋友一般都知道requests库,HttpLocust类的client正是封装了该库,用法基本一致。

当你的Locust类继承自HttpLocust,那你指向的TaskSet可以直接使用client属性发起HTTP请求,下面是一个例子,可以用来对//about两个URL进行压力测试:

from locust import HttpLocust, TaskSet, task

class MyTaskSet(TaskSet):
    @task(2)
    def index(self):
        self.client.get("/")

    @task(1)
    def about(self):
        self.client.get("/about/")

class MyLocust(HttpLocust):
    task_set = MyTaskSet
    min_wait = 5000
    max_wait = 15000

上面的示例代码中,每个模拟用户在5-15秒的等待间隔会发起一次请求,task的权重显示了对URL / 的请求会是 /about 的两倍。

有心的读者会发现,我们可以在TaskSet中直接使用self.client调用HttpSession,而不是self.locust.client。这是因为为了用户方便,TaskSet类中的self.client已经直接指向self.locust.client

HTTP client基本用法

client属性中,每个HttpLocust实例都有一个HttpSession实例。 HttpSession类实际上是requests.Session的一个子类,可用于发出getpostputdeleteheadpatchoptions等HTTP请求并报告给Locust的用于统计。 HttpSession实例会在请求之间保存cookie,以便它可以用来登录网站、保持请求会话。

下面是一个简单的例子,它向 /about 发出GET请求(假设self是TaskSet或HttpLocust类的实例:

response = self.client.get("/about")
print("Response status code:", response.status_code)
print("Response content:", response.text)

下面是POST请求的例子:

response = self.client.post("/login", {"username":"testuser", "password":"secret"})

人为标记成功失败

默认情况下,Locust将HTTP响应码为OK(2xx)的请求标记为成功,其他的就标记为失败。可能大多数时候这种处理方式就是我们想要的,但是有的时候就是想测试返回404的情况呢?或者server在返回错误时,错误信息在响应的消息体中而不体现在响应码,这些情况的处理就需要手动控制成功/失败。

使用catch_response参数和with,可以获取response的内容并标记失败:

with client.get("/", catch_response=True) as response:
    if response.content != b"Success":
        response.failure("Got wrong response")

正如可以将请求OK响码标记为失败一样,用catch_response参数与with语句也可以将请求不为2xx的响应码标记为成功:

with client.get("/does_not_exist/", catch_response=True) as response:
    if response.status_code == 404:
        response.success()

设置检查点

Locust中设置检查点是很简单的,可以直接用assert设置检查点,对response的内容检查,并输出错误信息:

response = self.client.get('/test_assert')
assert '‘success' in response.content, "Respense error: " + response

动态参数请求分组

网站的实参(Query String)经常是动态的,比如/blog?id=diff_id,一次查询一个新的id,如果不指定分组,在Locust的测试结果中和会看到默认以"/blog?id=diff_id"的很多分组,而这些请求都是在测试获取某个id的blog,其实可以被分为一组。

通过给client传入name参数能够实现分组:

# 这些id不同的请求都会被分组到 /blog/?id=[id] 中
for i in range(10):
    client.get("/blog?id=%i" % i, name="/blog?id=[id]")

参考

[1] Locust Documentation

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,258评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,335评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,225评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,126评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,140评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,098评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,018评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,857评论 0 273
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,298评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,518评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,678评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,400评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,993评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,638评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,801评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,661评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,558评论 2 352

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,651评论 18 139
  • Locust文件就是一般的Python文件。唯一的需求就是它至少需要一个继承于Locust的类. Locust类 ...
    Yuan_Jie阅读 4,036评论 0 7
  • Locust是什么? Locust 是一个开源负载测试工具。使用 Python 代码定义用户行为,也可以仿真百万个...
    zychen143阅读 7,143评论 1 2
  • 欲饮桃花酒魂断相思烟雨楼与尔同消万古愁十度的酒也胡诌他乡酒肉臭非重阳亦作重九我的洒脱不掣肘不会有勉强开口小桥流水柔...
    李帅帅angkuLC阅读 558评论 0 0
  • 想学绘画很久了,终于下定决心要好好来学,坚持每天绘画!绘画快一个月了,关注了一些大神,每次看着他们的作品真是各种神...
    一萌HM阅读 280评论 7 7