ruff

Author: rerpha

tests/test_sniff.py

@@ -1,11 +1,16 @@
 from unittest.mock import patch
 
-from confluent_kafka.admin import ClusterMetadata, BrokerMetadata, TopicMetadata
+from confluent_kafka.admin import BrokerMetadata, ClusterMetadata, TopicMetadata
 
 from saluki.sniff import sniff
 
+
 def test_sniff_with_two_partitions_in_a_topic():
-    with patch("saluki.sniff.AdminClient") as a, patch("saluki.sniff.Consumer") as c, patch("saluki.sniff.logger") as logger:
+    with (
+        patch("saluki.sniff.AdminClient") as a,
+        patch("saluki.sniff.Consumer") as c,
+        patch("saluki.sniff.logger") as logger,
+    ):
         fake_cluster_md = ClusterMetadata()
         broker1 = BrokerMetadata()
         broker1.id = "id1"
@@ -14,16 +19,13 @@ def test_sniff_with_two_partitions_in_a_topic():
         fake_cluster_md.brokers = {0: broker1}
 
         topic1 = TopicMetadata()
-        topic1.partitions = {0:{}}
+        topic1.partitions = {0: {}}
         topic2 = TopicMetadata()
-        topic2.partitions = {0:{}, 1:{}}
+        topic2.partitions = {0: {}, 1: {}}
 
-        fake_cluster_md.topics = {
-            "topic1": topic1,
-            "topic2": topic2
-        }
+        fake_cluster_md.topics = {"topic1": topic1, "topic2": topic2}
         a().list_topics.return_value = fake_cluster_md
-        c().get_watermark_offsets.return_value = 1,2
+        c().get_watermark_offsets.return_value = 1, 2
         sniff("whatever")
 
         brokers_call = logger.info.call_args_list[2]
@@ -38,4 +40,3 @@ def test_sniff_with_two_partitions_in_a_topic():
 
         topic2_call2 = logger.info.call_args_list[8]
         assert "1 - low:1, high:2, num_messages:1" in topic2_call2.args[0]
-

tests/test_utils.py

@@ -6,7 +6,7 @@
 from streaming_data_types.forwarder_config_update_fc00 import (
     ConfigurationUpdate,
     StreamInfo,
-    serialise_fc00
+    serialise_fc00,
 )
 
 from saluki.utils import (
@@ -87,7 +87,9 @@ def test_deserialising_with_schema_list_ignores_messages_with_schema_not_in_list
         mock_message.error.return_value = False
         mock_message.timestamp.return_value = 2, 1
 
-        deserialise_and_print_messages([mock_message, ok_message], None, schemas_to_filter_to=["fc00"])
+        deserialise_and_print_messages(
+            [mock_message, ok_message], None, schemas_to_filter_to=["fc00"]
+        )
         assert logger.info.call_count == 1