python locust_python locust 性能测试：locust参数-保证并发测试数据唯一性，循环取数据...

from locust import TaskSet, task, HttpLocust

import queue

class UserBehavior(TaskSet):

@task

def test_register(self):

try:

# get_nowait() 取不到数据直接崩溃；get() 取不到数据会一直等待

data = self.locust.user_data_queue.get_nowait() # 取值顺序 ‘username‘: ‘test0000‘、‘username‘: ‘test0001‘、‘username‘: ‘test0002‘...

except queue.Empty: # 取不到数据时，走这里

print(‘account data run out, test ended.‘)

exit(0)

print(‘register with user: {}, pwd: {}‘.format(data[‘username‘], data[‘password‘]))

payload = {

‘username‘: data[‘username‘],

‘password‘: data[‘password‘]

}

r = self.client.post(‘/user/signin?ReturnUrl=https%3A%2F%2Fwww.cnblogs.com%2F‘, data=payload)

self.locust.user_data_queue.put_nowait(data) # 把取出来的数据重新加入队列

assert r.status_code == 200

class WebsiteUser(HttpLocust):

host = ‘https://passport.cnblogs.com‘

task_set = UserBehavior

user_data_queue = queue.Queue() # 创建队列，先进先出

for index in range(100):

data = {

"username": "test%04d" % index,

"password": "pwd%04d" % index,

"email": "test%04d@debugtalk.test" % index,

"phone": "186%08d" % index,

}

user_data_queue.put_nowait(data) # 循环加入队列,循环完，继续执行

min_wait = 1000

max_wait = 3000

参考：http://debugtalk.com/post/head-first-locust-advanced-script/