#业务场景测试，登录，注册，查询功能组合
from locust import HttpLocust,TaskSet,task
#--coding:utf-8 --
import csv
import datetime
class UserBehavior(TaskSet):
‘’’ #登录业务，占比1成
@task(1)
def login_page(self):
file_login_result = open(“result_login.csv”,“a”)
nowtime = datetime.datetime.now()
for i in range(1,11):
username = “babai”+str(i)
login_data = {“login_info”:username,“password”:“liulong0706”}
login_response = self.client.post("/index.php?controller=simple&action=login_act",data = login_data).text
login_location = login_response.find(“您好”)
if login_location >= 0:
print(str(nowtime)+str(username)+“登录成功”)
file_login_result.write(str(nowtime)+str(username)+“登录成功”+’\n’)
else:
print(str(nowtime)+str(username)+“登录失败”)
file_login_result.write(str(nowtime)+str(username)+“登录失败”+’\n’)
file_login_result.close()’’’
#注册业务，占比2成
@task(2)
def register_page(self):
file_register_result = open(“result_register.csv”,“a”)
nowtime = datetime.datetime.now()
with open(“iwebshop_register.csv”,“r”) as register_file:
rows = csv.reader(register_file)
for row in rows:
register_data = {“email”:str(row[0]),“username”:str(row[1]),“password”:str(row[2]),“repassword”:str(row[3]),“captcha”:str(row[4])}
register_response = self.client.post("/index.php?controller=simple&action=reg_act",data = register_data).text
#print(register_response)
register_location = register_response.find(“恭喜”)
if register_location >= 0:
print(str(nowtime)+str(rows[0])+" 注册成功")
file_register_result.write(str(nowtime)+str(row[0])+" 注册成功"+’\n’)
else:
print(str(nowtime)+str(rows[0])+" 注册失败")
file_register_result.write(str(nowtime)+str(row[0])+" 注册失败"+’\n’)
file_register_result.close()
‘’’#查询业务，占比3成
@task(3)
def query_page(self):
nowtime = datetime.datetime.now()
file_query_result = open(“result_query.csv”,“a”)
with open(“query_data.csv”,“r”) as query_file:
rows = csv.reader(query_file)
for element in rows:
query_response = self.client.get("/index.php?controller=site&action=search_list&word="+str(element)).text
query_location = query_response.find(str(element))
if query_location >= 0:
print(str(nowtime)+str(element)+" 查找成功")
file_query_result.write(str(nowtime)+str(element)+" 查找成功"+’\n’)
else:
print(str(nowtime)+str(element)+" 查找失败")
file_query_result.write(str(nowtime)+str(element)+" 查找失败"+’\n’)
file_query_result.close()’’’
#设置登录和搜索的权重
‘’‘tasks = {login_page:1,register_page:2,query_page:3}’’’

class WebSiteUser(HttpLocust):
host = “http://192.168.0.5/iwebshop”
task_set = UserBehavior
min_wait = 2000
max_wait = 5000

代码段先放上，task(2)和task(3)标签的内容都一致，后面在task(2)中加了读取csv的for循环，向表格中写入输出结果的rows忘记改为row，导致此错误