# -*- coding: utf-8 -*-

from locust import HttpUser,task,TaskSet
import jpath
import json
from ensure import ensure

# 定义ScriptTasks类，继承TaskSet类
class ScriptTasks(TaskSet):
    #初始化，每个locust用户开始做的第一件事
    def on_start(self):
        #放置 用户名和密码
        res = self.client.post('/api/user/login', {
        "username":"13700998877",
        "password":'c123456'
        })
        #断言
        ensure(res.status_code).equals(200)
        #值
        code = jpath.get('code', json.loads(res.text))
        ensure(code).equals(0)
        msg = jpath.get('msg', json.loads(res.text))
        ensure(msg).equals("登录成功")


    #@task()装饰的方法为一个事务，方法的参数用于指定该行为的执行权重，参数越大每次被虚拟用户执行的概率越高，默认为1
    @task(1)
    def about(self):
        #self.client 属性使用python的request库的方法，调用和使用方法和request一样
        task_rel = self.client.get('/api/user/user_info')
        #断言
        ensure(task_rel.status_code).equals(200)
        #值
        code = jpath.get('code', json.loads(task_rel.text))
        ensure(code).equals(0)
        msg = jpath.get('msg', json.loads(task_rel.text))
        ensure(msg).equals("操作成功")

#    @task(2)
#    def demo(self):
#        payload = {}
#        headers = {}
#        self.client.post('/demo', data = payload,headers = headers)

#TaskSet类，该类定义用户任务信息(模拟用户信息)，

class WebsitUser(HttpUser):
    #指向一个定义的用户行为
    task_set = task(ScriptTasks)
    #被测系统的host，
    host = 'http://sky.nnzhp.cn'
    #每个用户执行两个任务间隔时间最小值，单位是（毫秒，默认是1000ms）
    min_wait = 100
    # 每个用户执行两个任务间隔时间最大值，单位是（毫秒）
    max_wait = 5000