chore: fix tests

faststream/specification/asyncapi/v3_0_0/schema/bindings/amqp/channel.py

@@ -1,4 +1,6 @@
-from typing import Optional, Self
+from typing import Optional
+
+from typing_extensions import Self
 
 from faststream.specification.asyncapi.v2_6_0.schema.bindings.amqp import (
     ChannelBinding as V2Binding,

tests/prometheus/confluent/basic.py

@@ -1,17 +1,13 @@
 from typing import Any
 
 from faststream import AckPolicy
-from faststream.confluent import KafkaBroker
 from faststream.confluent.prometheus import KafkaPrometheusMiddleware
 from tests.brokers.confluent.basic import ConfluentTestcaseConfig
 
 
 class KafkaPrometheusSettings(ConfluentTestcaseConfig):
     messaging_system = "kafka"
 
-    def get_broker(self, apply_types=False, **kwargs: Any) -> KafkaBroker:
-        return KafkaBroker(apply_types=apply_types, **kwargs)
-
     def get_middleware(self, **kwargs: Any) -> KafkaPrometheusMiddleware:
         return KafkaPrometheusMiddleware(**kwargs)
 

tests/prometheus/kafka/basic.py

@@ -1,17 +1,13 @@
 from typing import Any
 
 from faststream import AckPolicy
-from faststream.kafka import KafkaBroker
 from faststream.kafka.prometheus import KafkaPrometheusMiddleware
 from tests.brokers.kafka.basic import KafkaTestcaseConfig
 
 
 class KafkaPrometheusSettings(KafkaTestcaseConfig):
     messaging_system = "kafka"
 
-    def get_broker(self, apply_types=False, **kwargs: Any) -> KafkaBroker:
-        return KafkaBroker(apply_types=apply_types, **kwargs)
-
     def get_middleware(self, **kwargs: Any) -> KafkaPrometheusMiddleware:
         return KafkaPrometheusMiddleware(**kwargs)
 

tests/prometheus/nats/basic.py

@@ -2,9 +2,10 @@
 
 from faststream.nats import NatsBroker
 from faststream.nats.prometheus import NatsPrometheusMiddleware
+from tests.brokers.nats.basic import NatsTestcaseConfig
 
 
-class NatsPrometheusSettings:
+class NatsPrometheusSettings(NatsTestcaseConfig):
     messaging_system = "nats"
 
     def get_broker(self, apply_types=False, **kwargs: Any) -> NatsBroker:

tests/prometheus/nats/test_nats.py

@@ -22,7 +22,9 @@ def stream(queue):
 
 @pytest.mark.nats()
 class TestPrometheus(
-    NatsPrometheusSettings, LocalPrometheusTestcase, LocalRPCPrometheusTestcase
+    NatsPrometheusSettings,
+    LocalPrometheusTestcase,
+    LocalRPCPrometheusTestcase,
 ):
     async def test_metrics_batch(
         self,

tests/prometheus/nats/test_provider.py

@@ -16,7 +16,8 @@
 
 
 class LocalBaseNatsMetricsSettingsProviderTestcase(
-    NatsPrometheusSettings, LocalMetricsSettingsProviderTestcase
+    NatsPrometheusSettings,
+    LocalMetricsSettingsProviderTestcase,
 ):
     def test_get_publish_destination_name_from_cmd(self, queue: str) -> None:
         expected_destination_name = queue

tests/prometheus/rabbit/basic.py

@@ -1,15 +1,11 @@
 from typing import Any
 
-from faststream.rabbit import RabbitBroker
 from faststream.rabbit.prometheus import RabbitPrometheusMiddleware
 from tests.brokers.rabbit.basic import RabbitTestcaseConfig
 
 
 class RabbitPrometheusSettings(RabbitTestcaseConfig):
     messaging_system = "rabbitmq"
 
-    def get_broker(self, apply_types=False, **kwargs: Any) -> RabbitBroker:
-        return RabbitBroker(apply_types=apply_types, **kwargs)
-
     def get_middleware(self, **kwargs: Any) -> RabbitPrometheusMiddleware:
         return RabbitPrometheusMiddleware(**kwargs)

tests/prometheus/rabbit/test_provider.py

@@ -5,18 +5,14 @@
 
 from faststream.prometheus import MetricsSettingsProvider
 from faststream.rabbit.prometheus.provider import RabbitMetricsSettingsProvider
-from tests.prometheus.basic import (
-    LocalMetricsSettingsProviderTestcase,
-    LocalRPCPrometheusTestcase,
-)
+from tests.prometheus.basic import LocalMetricsSettingsProviderTestcase
 
 from .basic import RabbitPrometheusSettings
 
 
 class TestRabbitMetricsSettingsProvider(
     RabbitPrometheusSettings,
     LocalMetricsSettingsProviderTestcase,
-    LocalRPCPrometheusTestcase,
 ):
     @staticmethod
     def get_provider() -> MetricsSettingsProvider:

tests/prometheus/redis/basic.py

@@ -1,15 +1,11 @@
 from typing import Any
 
-from faststream.redis import RedisBroker
 from faststream.redis.prometheus import RedisPrometheusMiddleware
 from tests.brokers.redis.basic import RedisTestcaseConfig
 
 
 class RedisPrometheusSettings(RedisTestcaseConfig):
     messaging_system = "redis"
 
-    def get_broker(self, apply_types=False, **kwargs: Any) -> RedisBroker:
-        return RedisBroker(apply_types=apply_types, **kwargs)
-
     def get_middleware(self, **kwargs: Any) -> RedisPrometheusMiddleware:
         return RedisPrometheusMiddleware(**kwargs)