From 93eee7bf40e0659fc28b410dd2fcd4f0f5b1b364 Mon Sep 17 00:00:00 2001 From: Jaakko Date: Thu, 5 Sep 2024 15:46:12 +0300 Subject: [PATCH] Pass headers as geventhttpclient Headers object to _make_request --- locust/contrib/fasthttp.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/locust/contrib/fasthttp.py b/locust/contrib/fasthttp.py index f52eac936a..84a4a749da 100644 --- a/locust/contrib/fasthttp.py +++ b/locust/contrib/fasthttp.py @@ -23,7 +23,7 @@ from charset_normalizer import detect from gevent.timeout import Timeout from geventhttpclient._parser import HTTPParseError -from geventhttpclient.client import HTTPClientPool +from geventhttpclient.client import HTTPClientPool, HTTPClient from geventhttpclient.header import Headers from geventhttpclient.response import HTTPConnectionClosed, HTTPSocketPoolResponse from geventhttpclient.useragent import CompatRequest, CompatResponse, ConnectionError, UserAgent @@ -164,10 +164,12 @@ def _send_request_safe_mode(self, method: str, url: str, **kwargs): if hasattr(e, "response"): r = e.response else: + req_headers = HTTPClient.DEFAULT_HEADERS.copy() + req_headers.update(kwargs.get("headers")) req = self.client._make_request( url, method=method, - headers=kwargs.get("headers"), + headers=req_headers, payload=kwargs.get("payload"), params=kwargs.get("params"), )