From bfbba9e1883168bdd864030c46d6415c3af337d8 Mon Sep 17 00:00:00 2001 From: Bruno Casali Date: Wed, 30 Mar 2022 01:42:09 -0300 Subject: [PATCH] Add User-Agent inside the pre-defined headers --- meilisearch/_httprequests.py | 2 ++ tests/client/test_http_requests.py | 15 +++++++++++++++ 2 files changed, 17 insertions(+) create mode 100644 tests/client/test_http_requests.py diff --git a/meilisearch/_httprequests.py b/meilisearch/_httprequests.py index 03099242..1e28f805 100644 --- a/meilisearch/_httprequests.py +++ b/meilisearch/_httprequests.py @@ -7,12 +7,14 @@ MeiliSearchCommunicationError, MeiliSearchTimeoutError, ) +from meilisearch.version import qualified_version class HttpRequests: def __init__(self, config: Config) -> None: self.config = config self.headers = { 'Authorization': f'Bearer {self.config.api_key}', + 'User-Agent': qualified_version(), } def send_request( diff --git a/tests/client/test_http_requests.py b/tests/client/test_http_requests.py new file mode 100644 index 00000000..344d531e --- /dev/null +++ b/tests/client/test_http_requests.py @@ -0,0 +1,15 @@ + +import meilisearch +from tests import BASE_URL, MASTER_KEY + +from meilisearch.config import Config +from meilisearch._httprequests import HttpRequests +from meilisearch.version import qualified_version + +def test_get_headers_from_http_requests_instance(): + """Tests getting defined headers from instance in HttpRequests.""" + config = Config(BASE_URL, MASTER_KEY, timeout=None) + http = HttpRequests(config=config) + + assert http.headers['Authorization'] == f"Bearer {MASTER_KEY}" + assert http.headers['User-Agent'] == qualified_version()