【转载】locust性能测试3

#coding=utf-8
from locust import HttpLocust,TaskSet,task
import queue
import random

class test_taskset(TaskSet):
@task
def register(self):
try:
email=self.locust.emailqueue.get() #获取队列里的数据
print(email)
except queue.Empty: #队列取空后，直接退出
print('no data exist')
exit(0)
print('当前注册的邮箱:%s' %email)
body={"user":{"password":"123123","user_email":email,"full_name":"111"},"store_id":"7090","client_id":"9xqgzfbgg3myal2ikn7mls9vjqcj34j","client_secret":"k6fqamfib6x9rx4avrb2w63g9qvguo6"}
self.client.post('/api/v2/signup',json=body) #post方法发送请求

class test_run(HttpLocust):
host='https://trybindo.com'
task_set=test_taskset
# 生成测试手机号
emaildata=[str(1381212000+i)+'@bindo.com' for i in range(100)]

#添加到队列
emailqueue=queue.Queue()
for i in emaildata:
emailqueue.put_nowait(i)

if __name__=='__main__':
import os
os.system('locust -f locustpialiangzhuce.py')