From fe78f54c674332fea4242e4661ac8ad947d6b203 Mon Sep 17 00:00:00 2001 From: joseph-sentry Date: Mon, 17 Jun 2024 17:55:15 -0400 Subject: [PATCH] test: fix tests --- codecov_cli/helpers/request.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/codecov_cli/helpers/request.py b/codecov_cli/helpers/request.py index a4791c7e..3771bec8 100644 --- a/codecov_cli/helpers/request.py +++ b/codecov_cli/helpers/request.py @@ -7,6 +7,7 @@ from codecov_cli import __version__ from codecov_cli.types import RequestError, RequestResult +from typing import Optional logger = logging.getLogger("codecovcli") @@ -15,7 +16,7 @@ USER_AGENT = f"codecov-cli/{__version__}" -def _set_user_agent(headers: dict | None = None) -> dict: +def _set_user_agent(headers: Optional[dict] = None) -> dict: headers = headers or {} headers.setdefault("User-Agent", USER_AGENT) return headers @@ -37,7 +38,10 @@ def put(url: str, data: dict = None, headers: dict = None) -> requests.Response: def post( - url: str, data: dict | None = None, headers: dict | None = None, params: dict | None = None + url: str, + data: Optional[dict] = None, + headers: Optional[dict] = None, + params: Optional[dict] = None, ) -> requests.Response: headers = _set_user_agent(headers) return requests.post(url, json=data, headers=headers, params=params) @@ -82,7 +86,10 @@ def wrapper(*args, **kwargs): @retry_request def send_post_request( - url: str, data: dict | None = None, headers: dict | None = None, params: dict | None = None + url: str, + data: Optional[dict] = None, + headers: Optional[dict] = None, + params: Optional[dict] = None, ): return request_result(post(url=url, data=data, headers=headers, params=params)) @@ -94,7 +101,8 @@ def get_token_header_or_fail(token: str) -> dict: ) return {"Authorization": f"token {token}"} -def get_token_header(token: str) -> dict | None: + +def get_token_header(token: str) -> Optional[dict]: if token is None: return None return {"Authorization": f"token {token}"}