压力测试-Locust的一些常用类及方法

HttpUser类

使用此类时,表示进行压力测试的Http虚拟用户,每个用户都获得一个client属性,可用于发出Http请求,使用getpostputdelete等方法。在该类下,可以直接利用@task装饰器声明任务,也可由tasks属性定义。

@task装饰器

它在HttpUser类中起到声明任务的作用,可选择设置权重,从而控制执行率。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    @task(10)
    def one(self):
        self.client.get("/one")

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

上述脚本表示执行one的概率是two的10倍

@tag装饰器

@tag装饰器用于标识在运行压力测试时执行什么任务。例如下方脚本中,启动测试时应用--tags test参数,则仅会执行test任务,不会执行test2任务。

class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    @tag('test')
    @task
    def one(self):
        self.client.get("/one")
    @tag('test2')
    @task(2)
    def two(self):
        self.client.get("/two")

on_start/on_stop

开始/停止执行所在任务列表时调用。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    def on_start(self):
        self.client.post("/about",{"username":"name","password":"password"})
        print("1111111")
    def on_stop(self):
        print("2222222")
    @task
    def test(self):
        print("55555555")

上述脚本表示,开始运行时直接执行on_start任务,进行过程中执行test任务,结束任务时执行on_stop任务。可用于基于登录操作的流程测试。

weight

在一个文件中,建立多个HttpUser类时,weight代表执行该类下测试的可能性,数值越大,执行时被选择的可能性越高。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    weight=3
    ...
class startUser2(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    weight=1

执行startUser1的概率是startUser2的3倍。

tasks

tasks属性可以是列表,每次随机选择任务;也可以是字典,按设置的比例执行任务。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    @task
    def one(self):
        self.client.get("/one")
    def two(self):
        self.client.get("/two")
    tasks=[one,two]
    #tasks ={one:3,two:1} 
    #one及two代表已建立TaskSet任务实例,执行可能性是two的3倍

TaskSet类

目前可在HttpUser中直接定义任务,也可在TaskSet中定义压力测试中用户执行的任务。

注意如果是继承TaskSet定义任务,则需要引入User类,而后在启动执行User类或HttpUser类,否则只是声明了任务,执行会报错No User Class Found

开始运行时,会从tasks属性中选择一个任务,休眠wait_time定义的时间长度后,执行下个任务。

interrupt(reschedule=True)

TaskSet可以嵌套,在tasks属性中可以包含另一个TaskSet,如果需要中断,则需要在子TaskSet中调用interrupt,中断任务并将执行控制移交给父TaskSet
如果rescheduleTrue(默认),则父TaskSet将立即重新安排并执行新任务。

from locust import  HttpUser,task,TaskSet,constant
class startUser1(TaskSet):
    wait_time = constant(1)
    @task
    def onetask(self):
        self.client.get("/")
        print("55555555")
    @task
    def stop(self):
        self.interrupt()

class startUser2(HttpUser):
    host='https://www.baidu.com'
    wait_time = constant(1)
    tasks = [startUser1]
    @task
    def twotask(self):
        self.client.get("/")
        print("33333")

SequentialTaskSet类

SequentialTaskSet类其实是TaskSet类,但该类下的任务是按顺序执行。

from locust import  User,task,SequentialTaskSet,constant
class startUser1(SequentialTaskSet):
    wait_time = constant(1)
    @task
    def fist_task(self):
        print("1")
    @task
    def second_task(self):
        print("2")
    @task
    def third_task(self):
        print("3")
class startUser2(User):
    wait_time = constant(1)
    tasks=[startUser1]
#执行任务时,会依次打印出1、2、3

手动控制请求成功or失败

通过使用catch_response参数(catch_response值为布尔类型,如果设置为True, 允许该请求被标记为成功或失败)和 with 语句,设定请求失败or成功条件。

标记失败

虽然有时候最终接口访问正常,但仍可设计一些条件,将其标记为失败。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    @task
    def one(self):
        with self.client.get("/",catch_response=True) as response:
            if response.elapsed.total_seconds() > 0.01:
                response.failure("Request took too long")

请求时长大于0.01秒,标记失败
elapsed表示从发送请求到响应到达之间经过的时间

标记成功

有的时候虽然返回4**的状态码,但是是符合预期的,这个时候就需要我们主动将其标记为成功。

from locust import HttpUser,constant
class startUser1(HttpUser):
    host='https://www.***.com'
    wait_time = constant(1)
    @task
    def fifth_task(self):
        with self.client.get("/",catch_response=True) as response:
            if response.status_code==404:
                response.success()

明明是之前整理的知识点,但是重新写一次仍然用了很久....甚至比新写的用的时间还久....再次复习了一次...但是自己已经写毛了....

自我记录,有错误欢迎指正~

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

推荐阅读更多精彩内容