locust入门 -3 脚本开发代码规范

孟船长

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

3970

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

文章被收录于专栏：自动化测试实战

locust代码规范

上文我们讲了locust的基本语法，现在我们先来认识一下locust代码规范。了解这些以后我们开发脚本的时候就会思路更清晰。

我们要把控制用户行为，比如等待时间wait_time、主机地址host、定义任务等，写在继承了HttpUser类中。

写法一：

全部写在继承了HttpUser类下

from locust import HttpUser, constant, task

class QuickStart(HttpUser):
  host='https://api.weixin.qq.com'
  wait_time = constant(2)
  
  @task
  def myIndex(self):
    self.client.request(method='GET', url='/cig/index', name='打开首页')

写法二[用的比较少]：

from locust import HttpUser, constant, task

@task
def myIndex(user):
  user.client.request(method='GET', url='/cig/index', name='打开首页')

class QuickStart(HttpUser):
  host='https://api.weixin.qq.com'
  wait_time = constant(2)
  tasks = [myIndex]

写法三：

from locust import HttpUser, constant, task, TaskSet

class TaskBox(TaskSet):

  @task
  def myIndex(self):
    # 如果是请求这么写，下面接口请求不通
    # 这里的self.client其实是self.user.client，查看源码就明白了
    self.client.request(method='GET', url='/cig/index', name='打开首页')
  
  @task
  def myIndex2(self):
    print("用例2")

class QuickStart(HttpUser):
  host='https://api.weixin.qq.com'
  wait_time = constant(2)
  tasks = [TaskBox]

我们已经知道，@task标签不能定义任务执行次序，那我们想按顺序执行时，就可以用下面这种写法：

from locust import HttpUser, constant, task, SequentialTaskSet

# 继承SequentialTaskSetlei类
class SequExec(SequentialTaskSet):
  
    @task
    def create_tags(self):
        print('标记为task（2）的用例')

    @task(3)
    def get_access_token(self):
        print("标记为task（1）的用例")

    @task
    def my_test(self):
        print("标记为task(3)的用例")


class QuickStart(HttpUser):
  host='https://api.weixin.qq.com'
  wait_time = constant(2)
  tasks = [SequExec]

此时执行事，就会先执行create_tags一次，然后执行get_access_token3次，再执行my_test一次。再重复这个过程。

因为我们已经在User类里定义了host，执行时就可以不加--host xxx参数了：

locust -f 文件名.py

从上面几种写法可以看出，如果使用第一种写法，代码会很乱，所以实际工作中第三种写法是我们用的比较多的。

再补充一句，从locust写代码过程中可以看出，locust比较适合做接口压测，因为locust请求的是接口，而没有涉及到页面数据加载这些内容。我们在实际请求一个页面时，我们只是请求了一个接口，但是浏览器为了渲染前端，还会自动请求很多接口(比如加载图片)，locust对于浏览器自动请求的其他接口完全没有涉及，jmeter可以通过「高级」涉及一部分(软件限制，不能保证一模一样)，loadrunner可以实现和前端渲染页面完全一致的请求过程。ps：现在知道为什么loadrunner收费了吧。

现在就可以进入实战的环节了。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2022-03-23，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法