处理多个类中的变量 Python Locust

Question

我正在尝试编写一个简单的蝗虫性能测试任务。 我有两个顺序的 HTTP 任务集，如果我将它们写在一个 class 中它工作正常。 但是我想将它分为两个类，如下所示。 第一个 class [内部方法] 中的一个变量我需要在另一个 class 中使用。 你们能帮助我如何实现它。 下面的代码

# FIRST CLASS
from locust import SequentialTaskSet
class MySeqTask1(SequentialTaskSet):

@task
def get_status1(self):
    r = self.client.get("/200")
    print("1. Status of 200")
    self.data = 20 # ***VARIABLE TO BE SHARED TO SECOND CLASS***

#SECOND CLASS 
from locust import SequentialTaskSet, HttpUser, constant, task,FastHttpUser
from sampleLocust import MySeqTask1


class MySeqTask2(SequentialTaskSet):

@task
def get_status3(self):
    obj = MySeqTask1(self)
    self.client.get(f"/301")
    print("3. Status of 301")
    print(obj.data) # ***USING VARIABLE FROM FIRST CLASS***

class MyLoadTest(FastHttpUser):
   host = "https://http.cat"
   tasks = [MySeqTask1,MySeqTask2]
   wait_time = constant(1)

我得到以下错误。

   Command to run the program : locust -f utils/Baseclassone.py --headless -u 1 -r 1 -t 1m


print(obj.data) # ***USING VARIABLE FROM FIRST CLASS***
AttributeError: 'MySeqTask1' object has no attribute 'data'

Answer 1

您可以在用户而不是任务集上设置变量。 它可以作为任务集中的 self.user 访问，例如

self.user.data = 20