locust入门 —— 脚本开发代码规范

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收费了吧。

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

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走

---

在此特意为大家准备了一份13G的超实用干货学习资源，涉及的内容非常全面，涵盖功能测试、Python编程语言，接口测试、UI自动化测试、性能测试......包括软件学习路线图，50多天的上课视频、16个突击实战项目，80余个软件测试用软件，37份测试文档，70个软件测试相关问题，40篇测试经验级文章，上千份测试真题分享,还有2022软件测试面试宝典，还有软件测试求职的各类精选简历，希望对大家有所帮助…..关注下方公众号免费获取~