locust入门 -4 实战脚本开发

孟船长

发布于 2022-04-12 14:37:18

3020

发布于 2022-04-12 14:37:18

我们已经把基础知识都了解了，现在就可以进入实战脚本开发，看一下效果了。我们还是用微信公众号提供的接口，有兴趣的可以复制下面代码，可以直接在本地执行了。

我们先分析一下需求，我们的目标是压测创建标签接口，这个接口有一个条件，就是首先要登录，就是获取access_token。所以我们在写创建标签以前要先获取access_token，可是如果每次执行都先登录，那就成了压测登录+创建标签两个接口了，这样很难看到单个接口的真实数据。所以我们得这样写：

先执行获取access_token接口，获取access_token
把access_token传到创建标签接口，实现创建标签
创建标签名不能重复，所以要把标签名设置成变化的。
最后可以查询一下标签，看是否创建成功

这里还有一点要注意，就是我们模拟多用户的时候，这些用户都是并行的，模拟了多少个用户，on_start和on_stop就会执行多少次。

代码：

# coding: utf-8

import random
from locust import HttpUser, task, between, TaskSet


class TokenTask(TaskSet):

    token = None

    def on_start(self):
        data = {'appid': 'wx4d8f7de320aca501',
                'secret': 'd096e22047235cad799d917a517c6246',
                'grant_type': 'client_credential'}
        with self.client.request('POST',
                                 '/cgi-bin/token',
                                 params=data,
                                 verify=False,
                                 name='获取access_token') as res:
            access_token = res.json()['access_token']
            print(access_token)
            self.token = access_token

    @task
    def create_tags(self):
        print("执行create_tags方法")
        s = random.randint(1, 100)
        data = {"tag": {"name": f"Beijing{s}"}}
        with self.client.request('POST',
                                 '/cgi-bin/tags/create?access_token=' + self.token,
                                 json=data, name='新建标签') as r:
            print(r.text)

    def on_stop(self):
        with self.client.request('POST',
                                 '/cgi-bin/tags/get?access_token=' + self.token,
                                 name='查询标签') as r:
            print(r.text)


class QuickstartUser(HttpUser):
    tasks = [TokenTask]
    host = "https://api.weixin.qq.com"
    wait_time = between(1, 2)

执行文件：locust -f testindex.py可以看到

locust结果