from locust import HttpLocust, TaskSetdef login(l):l.client.post("/login", {"username":"ellen_key", "password":"education"})def logout(l):l.client.post("/logout", {"username":"ellen_key", "password":"education"})def index(l):l.client.get("/")def profile(l):l.client.get("/profile")class UserBehavior(TaskSet):tasks = {index: 2, profile: 1}def on_start(self):login(self)def on_stop(self):logout(self)class WebsiteUser(HttpLocust):task_set = UserBehaviormin_wait = 5000max_wait = 9000

from locust import HttpLocust, TaskSet, taskclass UserBehavior(TaskSet):def on_start(self):""" on_start is called when a Locust start before any task is scheduled """self.login()def on_stop(self):""" on_stop is called when the TaskSet is stopping """self.logout()def login(self):self.client.post("/login", {"username":"ellen_key", "password":"education"})def logout(self):self.client.post("/logout", {"username":"ellen_key", "password":"education"})@task(2)def index(self):self.client.get("/")@task(1)def profile(self):self.client.get("/profile")class WebsiteUser(HttpLocust):task_set = UserBehaviormin_wait = 5000max_wait = 9000

import randomclass WebsiteUser(HttpLocust):task_set = UserBehaviourwait_function = lambda self: random.expovariate(1)*1000

s = requests.session()

# -*- coding: utf_8 -*-
# locust性能测试from locust import HttpLocust, TaskSet, task
import subprocess
import json# 性能测试任务类 TaskSet.
class UserBehavior(TaskSet):# 账户登陆def _loginname(self):url = "/global/do-login.action"body = {"loginName": "%s" % self.locust.loginName,"password": "%s" % self.locust.password,"pcCodeForFocusMedia": 0}h = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36","Content-Type": "application/x-www-form-urlencoded",}a = self.client.post(url=url, data=body, headers=h, verify=False)data = a.json()assert data['status'] == 202# 选择租户def _tenant(self):url = "/global/login-entry.action"body = {"tenantId": "%s" % self.locust.tenantId,"passport.id": self.locust.passport_id,"encryptionKey": "%s" % self.locust.encryptionKey,"loginName": "%s" % self.locust.loginName}h = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36","Content-Type": "application/x-www-form-urlencoded",}b = self.client.post(url=url, data=body, headers=h, verify=False)data = b.json()assert data['status'] == 0def on_start(self):'''任务开始准备工作：只登录一次'''self._loginname()   # 账户登陆self._tenant()   # 选择租户# 任务@task(10)def getTagVals(self):u"""request_url：请求路径request_params：请求头参数request_json：请求json参数"""request_url = "/json/crm_dimension/dim-tree.action" # （待测试的路径）body = {"belongId":35,"dimType":1}response = self.client.post(url=request_url, data=body, verify=False)if response.status_code != 200:print(u"返回异常")print(u"请求返回状态码:", response.status_code)elif response.status_code == 200:print(u"返回正常")# 性能测试配置
class MobileUserLocust(HttpLocust):u"""min_wait ：用户执行任务之间等待时间的下界，单位：毫秒。max_wait ：用户执行任务之间等待时间的上界，单位：毫秒。"""# weight = 3task_set = UserBehaviorhost = "自己公司的url"  # （待测试的ip或者域名）min_wait = 3000max_wait = 6000loginName = "账号"password = 密码tenantId = 公司租户Id # 租户IDencryptionKey = "keyId"  # 保持登陆的keypassport_id = 默认 if __name__ == "__main__":subprocess.Popen("locust -f locustlearn.py", shell=True)