[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Unleash · Dec 13, 2023 · f9e93d1 · f9e93d1
1 parent a99dea8
commit f9e93d1
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 7 deletions.
diff --git a/UnleashClient/api/features.py b/UnleashClient/api/features.py
@@ -42,7 +42,9 @@ def get_feature_toggles(
     :return: (Feature flags, etag) if successful, ({},'') if not
     """
     try:
-        backoff_strategy = backoff_strategy or BackoffStrategy() # TODO creating it here doesn't make sense
+        backoff_strategy = (
+            backoff_strategy or BackoffStrategy()
+        )  # TODO creating it here doesn't make sense
         if not backoff_strategy.performAction():
             backoff_strategy.skipped()
             return {}, ""

diff --git a/UnleashClient/api/metrics.py b/UnleashClient/api/metrics.py
@@ -1,8 +1,8 @@
 import json
 
 import requests
-from UnleashClient.api.backoff import BackoffStrategy
 
+from UnleashClient.api.backoff import BackoffStrategy
 from UnleashClient.constants import APPLICATION_HEADERS, METRICS_URL
 from UnleashClient.utils import LOGGER, log_resp_info
 

diff --git a/UnleashClient/periodic_tasks/send_metrics.py b/UnleashClient/periodic_tasks/send_metrics.py
@@ -21,11 +21,13 @@ def aggregate_and_send_metrics(
     backoff_strategy: Optional[BackoffStrategy] = None,
 ) -> None:
     feature_stats_list = []
-    backoff_strategy = backoff_strategy or BackoffStrategy() # TODO creating it here doesn't make sense
+    backoff_strategy = (
+        backoff_strategy or BackoffStrategy()
+    )  # TODO creating it here doesn't make sense
     if not backoff_strategy.performAction():
         backoff_strategy.skipped()
         return {}, ""
-    
+
     for feature_name in features.keys():
         if not (features[feature_name].yes_count or features[feature_name].no_count):
             continue
@@ -53,7 +55,12 @@ def aggregate_and_send_metrics(
 
     if feature_stats_list:
         send_metrics(
-            url, metrics_request, custom_headers, custom_options, request_timeout, backoff_strategy
+            url,
+            metrics_request,
+            custom_headers,
+            custom_options,
+            request_timeout,
+            backoff_strategy,
         )
         # TODO should we do if send_metrics then update cache? We're also updating in the case of an exception
         cache.set(METRIC_LAST_SENT_TIME, datetime.now(timezone.utc))

diff --git a/tests/unit_tests/api/test_metrics.py b/tests/unit_tests/api/test_metrics.py
@@ -1,7 +1,6 @@
 import responses
 from pytest import mark, param
 from requests import ConnectionError
-from UnleashClient.api.backoff import BackoffStrategy
 
 from tests.utilities.mocks.mock_metrics import MOCK_METRICS_REQUEST
 from tests.utilities.testing_constants import (
@@ -11,6 +10,7 @@
     URL,
 )
 from UnleashClient.api import send_metrics
+from UnleashClient.api.backoff import BackoffStrategy
 from UnleashClient.constants import METRICS_URL
 
 FULL_METRICS_URL = URL + METRICS_URL
@@ -34,7 +34,12 @@ def test_send_metrics(payload, status, expected):
     responses.add(responses.POST, FULL_METRICS_URL, **payload, status=status)
 
     result = send_metrics(
-        URL, MOCK_METRICS_REQUEST, CUSTOM_HEADERS, CUSTOM_OPTIONS, REQUEST_TIMEOUT, BackoffStrategy()
+        URL,
+        MOCK_METRICS_REQUEST,
+        CUSTOM_HEADERS,
+        CUSTOM_OPTIONS,
+        REQUEST_TIMEOUT,
+        BackoffStrategy(),
     )
 
     assert len(responses.calls) == 1