Python locust性能测试框架模板

locust框架模板

from locust import HttpLocust, TaskSet, task
import Queue


class UserBehavior(TaskSet):

    def setup(self):
        print(\'task setup\')

    def teardown(self):
        print(\'task teardown\')

    def on_start(self):

        # key = self.login()
        request_url = "/api/user/getkey"
        request_json = {"username": "qxzn", "userpwd": "qx6688"}

        response = self.client.post(url=request_url, json=request_json)
        # print response.status_code
        # print response.text
        if response.status_code != 200:
            print u"返回异常,请求返回状态码:", response.status_code
        elif response.status_code == 200:
            print u"返回正常:", response.status_code
        global key_1, key_2
        key_1 = response.text.split(\'\'\'appkey":"\'\'\')[1].split(\'"\')[0]
        key_2 = response.text.split(\'\'\'infokey":"\'\'\')[1].split(\'"\')[0]
        print key_1 + \'-\' + key_2
        data = self.locust.user_data_queu
        print data

    def on_stop(self):
        # 虚拟用户结束Task时运行
        print(\'end\')

    @task(1)
    def resList(self):

        # key = self.login()
        header = {\'Accept\': \'application/json, text/plain, */*\',
                  \'Accept-Encoding\': \'gzip, deflate\',
                  \'Accept-Language\': \'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2\',
                  \'Connection\': \'keep-alive\',
                  \'User-Agent\': \'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:22.0) Gecko/20100101 Firefox/22.0\',
                  \'Content-Type\': \'application/json;charset=utf-8\',
                  \'Authorization\': key_1 + \'-\' + key_2}
        data = {"num": 15, "current_page": 1, "province": "", "city": "", "type": ""}
        request_url = "/api/desk_device/devicelist"

        response = self.client.post(url=request_url, headers=header, json=data)
        if response.status_code != 200:
            print u"返回异常,请求返回状态码:", response.status_code
        elif response.status_code == 200:
            print u"返回正常", response.status_code


class WebsiteUser(HttpLocust):

    def setup(self):
        print(\'locust setup\')

    def teardown(self):
        print(\'locust teardown\')

    user_data_queue = Queue.Queue()
    task_set = UserBehavior
    host = "http://xxxx.com:8095"
    min_wait = 5000
    max_wait = 8000