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']))
body = {
'username': data['username'],
'password': data['password']
}
r = self.client.post('/user/signin', data=body).text
assert r.status_code == 200
class WebsiteUser(Http