• 首页

  • 分类&标签

  • 归档

  • 手册

  • 项目池

  • 友链

  • 关于
伯 乐 讲 堂
伯 乐 讲 堂

查看「系列思维导图」

侠客 · Mr.潘

获取中...

05
27
Locust 性能测试

Locust 脚本加强篇(关联、检查点、集合点)

发表于 2021-05-27 • 专项测试 Locust • 被 408 人看爆

1、关联:通常在业务流程中有很多一系列的接口调用,从后面的接口依赖前边接口的结果数据

from lxml import etree
from locust import TaskSet, task, HttpUser
class UserBehavior(TaskSet):
    @staticmethod
    def get_session(html):
        tree = etree.HTML(html)
        return tree.xpath("//div[@class='btnbox']/input[@name='session']/@value")[0] \
    @task(10)
    def test_login(self):
        html = self.client.get('/login').text
        username = 'user@compay.com'
        password = '123456'
        session = self.get_session(html)
        payload = { 'username': username, 'password': password, 'session': session }
        self.client.post('/login', data=payload)
class WebsiteUser(HttpUser):
    host = 'http://debugtalk.com'
    # task_set = UserBehavior
    tasks = [UserBehavior ]
    min_wait = 1000
    max_wait = 3000

2、检查点:用来判断返回值是否符合要求

from lxml import etree
from locust import TaskSet, task, HttpUser
class UserBehavior(TaskSet):
    @staticmethod
    def get_session(html):
        tree = etree.HTML(html)
        return tree.xpath("//div[@class='btnbox']/input[@name='session']/@value")[0] \
    @task(10)
    def test_login(self):
        html = self.client.get('/login').text
        assert "200" in html
        username = 'user@compay.com'
        password = '123456'
        session = self.get_session(html)
        payload = { 'username': username, 'password': password, 'session': session }
        self.client.post('/login', data=payload)
class WebsiteUser(HttpUser):
    host = 'http://debugtalk.com'
    # task_set = UserBehavior
    tasks = [UserBehavior ]
    min_wait = 1000
    max_wait = 3000

3、集合点:提高某个接口的并发度,当所有用户运行到指定位置后集合等待,同时向下执行

from gevent._semaphore import Semaphore
from locust import TaskSet, events


all_locusts_spawned = Semaphore()
all_locusts_spawned.acquire()

def on_hatch_complete(**kwargs):
    # 创建钩子方法
    all_locusts_spawned.release() 

# 挂载到locust钩子函数(所有的Locust实例产生完成时触发)
events.hatch_complete += on_hatch_complete

class TestTask(TaskSet):
    def login(self):
        headers = {"Content-Type": "application/json"}
        res = self.client.post("/api/v1/user/login",
                               json={"loginName": "18550208315", "passwordMd5": "e10adc3949ba59abbe56e057f20f883e"},
                               headers=headers)
        self.user.token = res.json()['data']

    def on_start(self):
        """ on_start is called when a Locust start before any task is scheduled """
        # 限制在所有用户准备完成前处于等待状态
        all_locusts_spawned.wait()
        self.login()

    @task
    def goods_search(self):
        """商品查询"""
        headers = {
            "Content-Type": "application/json",
            "token": self.user.token
        }
        self.client.get("/api/v1/search", params={"keyword": "Apple"}, headers=headers)


class WebSiteUser(HttpUser):
    """HTTP 用户"""
    wait_time = between(1, 3)
    tasks = [UserBehavior]
标题:Locust 脚本加强篇(关联、检查点、集合点)
作者:侠客 · Mr.潘

读后有收获可以支付宝请作者喝咖啡,读后有疑问请加在下微信(pansfy)讨论:

CentOS7 监控网络流量
Locust 脚本初级篇
侠客 · Mr.潘

侠客 · Mr.潘

未来的你,会感谢今天仍正在奋斗的你

Github QQ Email RSS
看爆 Top10
  • 助力项目池 3,538次看爆
  • 第 4 节 yum 版的 LAMP 环境部署 2,535次看爆
  • 你的生产力工具集成就高效人士 1,880次看爆
  • 第 5 节 yum 版的 LNMP 环境部署 1,676次看爆
  • 第 3 节 企业级系统环境之上云篇 1,387次看爆
  • 服务端的架构的演进之路 1,284次看爆
  • 第 1 节 Docker 实践 1,190次看爆
  • 镜像仓库一文打尽 982次看爆
  • Docker本地私有镜像仓库Harbor搭建及配置 980次看爆
  • 第 6 节 源码版的LAMP环境部署 945次看爆

Copyright © 2023 侠客 · Mr.潘 · 苏ICP备19067937号

Proudly published with Halo · Theme by fyang · 站点地图