From 78cab33b5072387bf7683055b13152807ef692e4 Mon Sep 17 00:00:00 2001 From: Florin Barbu Date: Fri, 13 Sep 2024 10:43:34 +0300 Subject: [PATCH] fixes --- tests/sharding/test_filter.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/sharding/test_filter.py b/tests/sharding/test_filter.py index 4abacddc..caf57611 100644 --- a/tests/sharding/test_filter.py +++ b/tests/sharding/test_filter.py @@ -34,7 +34,7 @@ class TestFilterStaticSharding(StepsSharding): class TestFilterAutoSharding(StepsSharding): def test_filter_works_with_auto_sharding(self): self.setup_first_relay_node_with_filter( - cluster_id=self.auto_cluster, content_topic=self.test_content_topic, pubsub_topic=self.test_pubsub_topic + cluster_id=self.auto_cluster, content_topic=self.test_content_topic, pubsub_topic=self.test_pubsub_topic, num_shards_in_network=1 ) self.setup_second_node_as_filter(cluster_id=self.auto_cluster, content_topic=self.test_content_topic, pubsub_topic=self.test_pubsub_topic) self.subscribe_first_relay_node(content_topics=[self.test_content_topic]) @@ -43,7 +43,10 @@ class TestFilterAutoSharding(StepsSharding): def test_filter_auto_sharding_multiple_content_topics(self): self.setup_first_relay_node_with_filter( - cluster_id=self.auto_cluster, content_topic=CONTENT_TOPICS_DIFFERENT_SHARDS, pubsub_topic=PUBSUB_TOPICS_SAME_CLUSTER + cluster_id=self.auto_cluster, + content_topic=CONTENT_TOPICS_DIFFERENT_SHARDS, + pubsub_topic=PUBSUB_TOPICS_SAME_CLUSTER, + num_shards_in_network=8, ) self.setup_second_node_as_filter( cluster_id=self.auto_cluster, content_topic=CONTENT_TOPICS_DIFFERENT_SHARDS, pubsub_topic=self.test_pubsub_topic