将令牌传递给 Locust 请求

Question

我正在尝试将生成的 Oauth 令牌传递到@task request中

这会导致 401 错误

from locust import HttpUser, constant, task
import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)


class ProcessRequests(HttpUser):

    def on_start(self):

        tenant_id = "tenant123"
        client_id = "client123"
        secret = "secret123"
        scope = "api://123/.default"

        body ="grant_type=client_credentials&client_id=" + client_id + "&client_secret=" + secret + "&scope=" + scope

        tokenResponse = self.client.post(
                f"https://login.microsoftonline.com/{tenant_id}/oauth2/v2.0/token",
                body,
                headers = { "ContentType": "application/x-www-form-urlencoded"} 
            )

        response = tokenResponse.json()
        responseToken = response['access_token']
        

        self.headers = {'Authorization': 'Bearer' + responseToken}


    @task
    def get_labware(self):
        self.client.get("https://url/123", name="Labware",headers=self.headers)

    @task
    def get_instruments(self):
        self.client.get("https://url/456", name="Instruments", headers=self.headers)

我让它在 K6/javascript 中工作，所以我知道参数是正确的。 我只是不能在 python 中正确解析或传递它们。

Answer 1

您需要在Bearer和您的令牌之间留一个空格。 尝试这个：

self.headers = {'Authorization': 'Bearer ' + responseToken}