Python Locust -在req/task中自定义超时

Question

我需要用Locust库运行一些负载测试。我需要以特定的方式处理504-gateway超时。是否有可能在请求(或任务)中设置自定义超时。我不需要等待120秒直到超时发生，我需要等待不超过5秒，对我来说这是超时。我的意思是在代码中:

@task(1)
def check_max_5_sec(self):
self.client.get(url=URL_5_sec)
...
@task(1)
def check_max_10_sec(self):
self.client.get(url=URL_10_sec)

在理想的世界中，我希望看到Locust web界面上的失败，如果在给定的超时内没有响应。如何做到这一点?

我知道在FastHttp中有connection_timeout和network_timeout(可能这可能是一个解决方案)，但我可以用HttpUser做到这一点吗?

Answer 1

对于HttpUser/请求只需通过timeout=5。

详情见https://docs.python-requests.org/en/latest/user/quickstart/#timeouts