linting + formatting

airtai · May 18, 2024 · d880ac3 · d880ac3
1 parent e78ce9f
commit d880ac3
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/faststream/confluent/testing.py b/faststream/confluent/testing.py
@@ -6,7 +6,6 @@
 
 from faststream.broker.message import encode_message, gen_cor_id
 from faststream.confluent.broker import KafkaBroker
-from faststream.confluent.client import TopicPartition
 from faststream.confluent.publisher.asyncapi import AsyncAPIBatchPublisher
 from faststream.confluent.publisher.producer import AsyncConfluentFastProducer
 from faststream.confluent.subscriber.asyncapi import AsyncAPIBatchSubscriber
@@ -37,17 +36,10 @@ def create_publisher_fake_subscriber(
         broker: KafkaBroker,
         publisher: "AsyncAPIPublisher[Any]",
     ) -> "HandlerCallWrapper[Any, Any, Any]":
-        if publisher.partition:
-            tp = TopicPartition(topic=publisher.topic, partition=publisher.partition)
-            sub = broker.subscriber(
-                partitions=[tp],
-                batch=isinstance(publisher, AsyncAPIBatchPublisher),
-            )
-        else:
-            sub = broker.subscriber(
-                publisher.topic,
-                batch=isinstance(publisher, AsyncAPIBatchPublisher),
-            )
+        sub = broker.subscriber(  # type: ignore[call-overload,misc]
+            publisher.topic,
+            batch=isinstance(publisher, AsyncAPIBatchPublisher),
+        )
 
         if not sub.calls: