一个完整业务流程的测试，一些动态数据基本都要进行参数化，比如虚拟用户的登录账号，或者随机访问商品详情的商品id

  from locust import HttpUser,TaskSet,between,task
  import os,sys
  curPath = os.path.abspath(os.path.dirname(__file__))
  rootPath = os.path.split(curPath)[0]
  PathProject = os.path.split(rootPath)[0]
  sys.path.append(rootPath)
  sys.path.append(PathProject)

  n = 0

  class UserBehavior(TaskSet):

      def get_account(self):
          global n
          mobiles = ['1300000000','1300000001','1300000002','1300000003']
          n += 1

          #数据循环使用
          if n>=len(mobiles):
              n = 0
          return mobiles[n]

      @task(4)
      def test_login(self):
          """
          登录用户
          :return:
          """
          mobile = self.get_account()
          url = '/login'
          param={'mobile':mobile}
          with self.client.post(url,json=param,headers={},catch_response = True) as response:
              print("成功登录账号",mobile)



  class WebsiteUser(HttpUser):
      host = 'http://127.0.0.1'
      tasks = [UserBehavior]

      wait_time = between(1, 2)

  if __name__ == '__main__':
      os.system("locust -f ccc.py")

使用队列进行参数化

一、

from locust import HttpUser,TaskSet,between,task
import os,sys
curPath = os.path.abspath(os.path.dirname(__file__))
rootPath = os.path.split(curPath)[0]
PathProject = os.path.split(rootPath)[0]
sys.path.append(rootPath)
sys.path.append(PathProject)
import queue

n = 0

class UserBehavior(TaskSet):


    @task(4)
    def test_login(self):
        """
        登录用户
        :return:
        """
        try:
            mobile = self.parent.queueData.get()  # 获取队列里的数据

            url = '/login'
            param = {'mobile': mobile}
            with self.client.post(url, json=param, headers={}, catch_response=True) as response:
                print("成功登录账号", mobile)

        except queue.Empty:

            # 队列取空后，直接退出
            print('no data exist')




class WebsiteUser(HttpUser):
    host = 'http://127.0.0.1'
    tasks = [UserBehavior]

    wait_time = between(1, 2)

    queueData = queue.Queue()  # 队列实例化
    for count in range(0,5):   # 循环数据生成
        data = {
            "username": str(13000000000 + count)
        }
        queueData.put_nowait(data)

if __name__ == '__main__':
    os.system("locust -f ccc.py")

二：
上面队列的例子，参数化最多数据是5个，如果在并发参数设置为100个用户的时候，最多并发还是只有5个

我们可以进行数据循环使用

from locust import HttpUser,TaskSet,between,task
import os,sys
curPath = os.path.abspath(os.path.dirname(__file__))
rootPath = os.path.split(curPath)[0]
PathProject = os.path.split(rootPath)[0]
sys.path.append(rootPath)
sys.path.append(PathProject)
import queue

n = 0

class UserBehavior(TaskSet):


    @task(4)
    def test_login(self):
        """
        登录用户
        :return:
        """
        try:
            mobile = self.parent.queueData.get()  # 获取队列里的数据
            self.parent.queueData.put_nowait(mobile)  # 再将取出的数据插入队尾，对数据进行循环使用

            url = '/login'
            param = {'mobile': mobile}
            with self.client.post(url, json=param, headers={}, catch_response=True) as response:
                print("成功登录账号", mobile)

        except queue.Empty:

            # 队列取空后，直接退出
            print('no data exist')




class WebsiteUser(HttpUser):
    host = 'http://127.0.0.1'
    tasks = [UserBehavior]

    wait_time = between(1, 2)

    queueData = queue.Queue()  # 队列实例化
    for count in range(0,5):   # 循环数据生成
        queueData.put_nowait(str(13000000000 + count))

if __name__ == '__main__':
    os.system("locust -f ccc.py")

locust性能测试：参数化（七）

locust性能测试：参数化（七）

一、