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/
