Question

是否有可能让Locust传递带有安全令牌的标头命令来对API进行负载测试？

由于正在测试的服务器必须接收带有请求的令牌，因此我正在尝试针对带有令牌的标头标志的编码器测试我们的api。

curl -H“授权：令牌令牌字符串” http://someserver

Answer 1

是的，您可以使用：

token_string = "token string"

resp = self.client.post(
            url="http://someserver",
            data=json.dumps(data),
            auth=None,
            headers={"authorization": "Token " + token_string},
            name="http://someserver",
        )

Answer 2

如果您想为每个请求使用相同的标头，您还可以在 on_start 方法中将它们设置为客户端。它们将自动用于每个客户端请求。

class User(HttpUser):

    def on_start(self):
        self.client.headers = {'Authorization': 'my-auth-token'}

    @task
    def my_authenticated_task(self):
        self.client.post('enspoint')  # this will use headers we set earlier

蝗虫传递标头

2 个答案: