This commit is contained in:
Florin Barbu 2024-09-13 10:43:34 +03:00
parent 48f844c186
commit 78cab33b50
No known key found for this signature in database
GPG Key ID: 593D6DBC6D9E5095

View File

@ -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