Start fixing some of the issues reported by the pipeline

datacosmos/client.py

@@ -1,12 +1,12 @@
 import logging
 import os
 from datetime import datetime, timedelta, timezone
-from typing import Optional, Any
+from typing import Any, Optional
 
 import requests
-from requests_oauthlib import OAuth2Session
 from oauthlib.oauth2 import BackendApplicationClient
 from requests.exceptions import RequestException
+from requests_oauthlib import OAuth2Session
 
 from config.config import Config
 
@@ -18,7 +18,9 @@ class DatacosmosClient:
     """
 
     def __init__(
-        self, config: Optional[Config] = None, config_file: str = "config/config.yaml"
+        self,
+        config: Optional[Config] = None,
+        config_file: str = "config/config.yaml",
     ):
         """
         Initialize the DatacosmosClient.
@@ -42,7 +44,9 @@ def _load_config(self, config_file: str) -> Config:
             if os.path.exists(config_file):
                 self.logger.info(f"Loading configuration from {config_file}")
                 return Config.from_yaml(config_file)
-            self.logger.info("Loading configuration from environment variables")
+            self.logger.info(
+                "Loading configuration from environment variables"
+            )
             return Config.from_env()
         except Exception as e:
             self.logger.error(f"Failed to load configuration: {e}")
@@ -73,7 +77,9 @@ def _authenticate_and_initialize_client(self) -> requests.Session:
 
             # Initialize the HTTP session with the Authorization header
             http_client = requests.Session()
-            http_client.headers.update({"Authorization": f"Bearer {self.token}"})
+            http_client.headers.update(
+                {"Authorization": f"Bearer {self.token}"}
+            )
             return http_client
         except RequestException as e:
             self.logger.error(f"Request failed during authentication: {e}")
@@ -97,7 +103,9 @@ def get_http_client(self) -> requests.Session:
         self._refresh_token_if_needed()
         return self._http_client
 
-    def request(self, method: str, url: str, *args: Any, **kwargs: Any) -> requests.Response:
+    def request(
+        self, method: str, url: str, *args: Any, **kwargs: Any
+    ) -> requests.Response:
         """
         Send an HTTP request using the authenticated session.
         Logs request and response details.
@@ -107,7 +115,9 @@ def request(self, method: str, url: str, *args: Any, **kwargs: Any) -> requests.
             self.logger.info(f"Making {method.upper()} request to {url}")
             response = self._http_client.request(method, url, *args, **kwargs)
             response.raise_for_status()
-            self.logger.info(f"Request to {url} succeeded with status {response.status_code}")
+            self.logger.info(
+                f"Request to {url} succeeded with status {response.status_code}"
+            )
             return response
         except RequestException as e:
             self.logger.error(f"HTTP request failed: {e}")

tests/unit/datacosmos/client/test_client_authentication.py

@@ -1,10 +1,14 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
 @patch("datacosmos.client.OAuth2Session.fetch_token")
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client", autospec=True)
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client",
+    autospec=True,
+)
 def test_client_authentication(mock_auth_client, mock_fetch_token):
     """
     Test that the client correctly fetches a token during authentication.

tests/unit/datacosmos/client/test_client_delete_request.py

@@ -1,9 +1,12 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 def test_delete_request(mock_auth_client):
     """
     Test that the client performs a DELETE request correctly.
@@ -27,7 +30,6 @@ def test_delete_request(mock_auth_client):
     # Assertions
     assert response.status_code == 204
     mock_http_client.request.assert_called_once_with(
-        "DELETE",
-        "https://mock.api/some-endpoint"
+        "DELETE", "https://mock.api/some-endpoint"
     )
     mock_auth_client.call_count == 2

tests/unit/datacosmos/client/test_client_get_request.py

@@ -1,9 +1,12 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 def test_client_get_request(mock_auth_client):
     """
     Test that the client performs a GET request correctly.
@@ -29,7 +32,6 @@ def test_client_get_request(mock_auth_client):
     assert response.status_code == 200
     assert response.json() == {"message": "success"}
     mock_http_client.request.assert_called_once_with(
-        "GET",
-        "https://mock.api/some-endpoint"
+        "GET", "https://mock.api/some-endpoint"
     )
     mock_auth_client.call_count == 2

tests/unit/datacosmos/client/test_client_initialization.py

@@ -1,9 +1,12 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 @patch("os.path.exists", return_value=False)
 @patch("config.Config.from_env")
 def test_client_initialization(mock_from_env, mock_exists, mock_auth_client):

tests/unit/datacosmos/client/test_client_post_request.py

@@ -1,9 +1,12 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 def test_post_request(mock_auth_client):
     """
     Test that the client performs a POST request correctly.
@@ -31,8 +34,6 @@ def test_post_request(mock_auth_client):
     assert response.status_code == 201
     assert response.json() == {"message": "created"}
     mock_http_client.request.assert_called_once_with(
-        "POST",
-        "https://mock.api/some-endpoint",
-        json={"key": "value"}
+        "POST", "https://mock.api/some-endpoint", json={"key": "value"}
     )
     mock_auth_client.call_count == 2

tests/unit/datacosmos/client/test_client_put_request.py

@@ -1,9 +1,12 @@
-from unittest.mock import patch, MagicMock
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 def test_put_request(mock_auth_client):
     """
     Test that the client performs a PUT request correctly.
@@ -31,8 +34,6 @@ def test_put_request(mock_auth_client):
     assert response.status_code == 200
     assert response.json() == {"message": "updated"}
     mock_http_client.request.assert_called_once_with(
-        "PUT",
-        "https://mock.api/some-endpoint",
-        json={"key": "updated-value"}
+        "PUT", "https://mock.api/some-endpoint", json={"key": "updated-value"}
     )
     mock_auth_client.call_count == 2

tests/unit/datacosmos/client/test_client_token_refreshing.py

@@ -1,10 +1,13 @@
-from unittest.mock import patch, MagicMock
 from datetime import datetime, timedelta, timezone
-from datacosmos.client import DatacosmosClient
+from unittest.mock import MagicMock, patch
+
 from config.config import Config
+from datacosmos.client import DatacosmosClient
 
 
-@patch("datacosmos.client.DatacosmosClient._authenticate_and_initialize_client")
+@patch(
+    "datacosmos.client.DatacosmosClient._authenticate_and_initialize_client"
+)
 def test_client_token_refreshing(mock_auth_client):
     """
     Test that the client refreshes the token when it expires.
@@ -44,6 +47,5 @@ def test_client_token_refreshing(mock_auth_client):
 
     # Verify the request was made correctly
     mock_http_client.request.assert_called_once_with(
-        "GET",
-        "https://mock.api/some-endpoint"
+        "GET", "https://mock.api/some-endpoint"
     )