feat: ruff

airtai · Dec 20, 2024 · 2f6abd9 · 2f6abd9
1 parent c9de6ca
commit 2f6abd9
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 25 deletions.
diff --git a/tests/prometheus/test_metrics.py b/tests/prometheus/test_metrics.py
@@ -1,17 +1,23 @@
 import random
-from typing import Optional, Any
+from typing import Any, Optional
 
 import pytest
 from prometheus_client import CollectorRegistry
 
 from faststream.prometheus.container import MetricsContainer
 from faststream.prometheus.manager import MetricsManager
 from faststream.prometheus.types import ProcessingStatus, PublishingStatus
-from tests.prometheus.utils import get_received_messages_metric, get_received_messages_size_bytes_metric, \
-    get_received_messages_in_process_metric, get_received_processed_messages_metric, \
-    get_received_processed_messages_duration_seconds_metric, get_received_processed_messages_exceptions_metric, \
-    get_published_messages_metric, get_published_messages_duration_seconds_metric, \
-    get_published_messages_exceptions_metric
+from tests.prometheus.utils import (
+    get_published_messages_duration_seconds_metric,
+    get_published_messages_exceptions_metric,
+    get_published_messages_metric,
+    get_received_messages_in_process_metric,
+    get_received_messages_metric,
+    get_received_messages_size_bytes_metric,
+    get_received_processed_messages_duration_seconds_metric,
+    get_received_processed_messages_exceptions_metric,
+    get_received_processed_messages_metric,
+)
 
 
 class TestCaseMetrics:
@@ -178,7 +184,7 @@ def test_remove_received_message_in_process(
             app_name=app_name,
             broker=broker,
             queue=queue,
-            messages_amount=messages_amount-1,
+            messages_amount=messages_amount - 1,
         )
 
         manager.add_received_message_in_process(

diff --git a/tests/prometheus/utils.py b/tests/prometheus/utils.py
@@ -1,4 +1,5 @@
-from typing import Sequence, cast
+from collections.abc import Sequence
+from typing import cast
 
 from dirty_equals import IsPositiveFloat, IsStr
 from prometheus_client import Histogram, Metric
@@ -8,12 +9,12 @@
 
 
 def get_received_messages_metric(
-        *,
-        metrics_prefix: str,
-        app_name: str,
-        broker: str,
-        queue: str,
-        messages_amount: int,
+    *,
+    metrics_prefix: str,
+    app_name: str,
+    broker: str,
+    queue: str,
+    messages_amount: int,
 ) -> Metric:
     metric = Metric(
         name=f"{metrics_prefix}_received_messages",
@@ -51,11 +52,11 @@ def get_received_messages_size_bytes_metric(
     size: int,
 ) -> Metric:
     metric = Metric(
-            name=f"{metrics_prefix}_received_messages_size_bytes",
-            documentation="Histogram of received messages size in bytes by broker and handler",
-            unit="",
-            typ="histogram",
-        )
+        name=f"{metrics_prefix}_received_messages_size_bytes",
+        documentation="Histogram of received messages size in bytes by broker and handler",
+        unit="",
+        typ="histogram",
+    )
     metric.samples = [
         *[
             Sample(
@@ -99,12 +100,12 @@ def get_received_messages_size_bytes_metric(
 
 
 def get_received_messages_in_process_metric(
-        *,
-        metrics_prefix: str,
-        app_name: str,
-        broker: str,
-        queue: str,
-        messages_amount: int,
+    *,
+    metrics_prefix: str,
+    app_name: str,
+    broker: str,
+    queue: str,
+    messages_amount: int,
 ) -> Metric:
     metric = Metric(
         name=f"{metrics_prefix}_received_messages_in_process",