蝗虫传递标头

Question

Locust是否可以通过带有安全令牌的标头命令来对API进行负载测试？

我正在尝试为我们的编码器的API测试带有令牌标记的标头标志，因为被测试的服务器必须接收带有请求的令牌，即。

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

Answer 1

此外，如果您想对每个请求使用相同的标头，您还可以在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

Answer 2

是的，您可以使用：

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",
        )