在 Locust 中通过顺序任务将数据从一个任务函数逐一传递到另一个任务函数时出现问题

Question

我们正在尝试实现一种基于用户场景的负载测试，它基本上在一个 url 中创建一个订单，然后我们必须在下一个 url 中传递订单 id 来获取该订单的状态。

我们正在使用 Locust 顺序任务。因为我们希望它按第一个请求 -> 第二个请求 -> 第三个请求的顺序运行。我们已经按预期获得了响应数据，但我们无法将该变量数据发送到第三个任务以显示订单状态。

import json
from locust import HttpLocust, TaskSet, task, TaskSequence, seq_task

class MyTaskSequence(TaskSequence):

    response_data = ""

    @seq_task(1)
    def index(self):
        print("--- First Task")
        response = self.client.get("/order/testing-06a5c/")
        print(response.status_code)

    @seq_task(2)
    def get_details(self):
        print("--- Second Task")
        response = self.client.post(return_data, headers={"authority": "staging.domain.com", "referer":"https://staging.domain.com/"})
        print(response.status_code)
        response_data = json.loads(response.text)
        print(response_data["details"]["claim_uri"])
        self.response_data

    def on_start(self):
        self.get_details()

    @seq_task(3)     
    def post_details(self):
        print(self.get_details())
        print("-- Third Task", self.response_data)
        #return_data = self.response_data["details"]["claim_uri"]
        #response = self.client.post(return_data, headers={"authority": "staging.domain.com", "referer":"https://staging.domain.com/"})
        #print(response.text)

class MyLocust(HttpLocust):
    task_set = MyTaskSequence
    min_wait = 5000
    max_wait = 15000
    host = 'https://staging.domain.com'

输出：

[2018-11-19 19:24:19,784] system.local/INFO/stdout:
[2018-11-19 19:24:19,784] system.local/INFO/stdout: --- First Task
[2018-11-19 19:24:19,785] system.local/INFO/stdout:
[2018-11-19 19:24:20,235] system.local/INFO/stdout: 200
[2018-11-19 19:24:20,235] system.local/INFO/stdout:
[2018-11-19 19:24:29,372] system.local/INFO/stdout: --- Second Task
[2018-11-19 19:24:29,373] system.local/INFO/stdout:
[2018-11-19 19:24:29,653] system.local/INFO/stdout: 200
[2018-11-19 19:24:29,654] system.local/INFO/stdout:
[2018-11-19 19:24:29,654] system.local/INFO/stdout: /payment/initiate/claim/bc6d5024-f608-41af-8e78-191798c31a69/
[2018-11-19 19:24:29,654] system.local/INFO/stdout:
[2018-11-19 19:24:37,089] system.local/INFO/stdout: --- Second Task
[2018-11-19 19:24:37,089] system.local/INFO/stdout:
[2018-11-19 19:24:37,367] system.local/INFO/stdout: 200
[2018-11-19 19:24:37,367] system.local/INFO/stdout:
[2018-11-19 19:24:37,367] system.local/INFO/stdout: /payment/initiate/claim/72217a35-01fc-488e-885e-aea81a57a463/
[2018-11-19 19:24:37,368] system.local/INFO/stdout:
[2018-11-19 19:24:37,368] system.local/INFO/stdout: None
[2018-11-19 19:24:37,368] system.local/INFO/stdout:
[2018-11-19 19:24:37,368] system.local/INFO/stdout: ('-- Third Task', '')
[2018-11-19 19:24:37,368] system.local/INFO/stdout:
^C[2018-11-19 19:24:40,598] system.local/ERROR/stderr: KeyboardInterrupt

我们想将response_data["details"]["claim_uri"]变量传递给@seq_task(3)。这样我们就可以形成一个动态的目标domain.com/response_data["details"]["claim_uri"]。该变量将是字符串类型，并且每次第三个任务调用时都必须单独传递。

我们尝试了这种方法，但输出中没有任何结果。

有什么遗漏的吗？

Answer 1

问题在于可变响应数据。在没有 self 的情况下调用变量时，变量的作用域是本地的。

下面的代码有效。

@seq_task(2)
    def get_details(self):
        <-- function data -->
        self.response_data = json.loads(response.text)