Fix failing CI tests

src/steps/light_push.py

@@ -23,6 +23,7 @@ class StepsLightPush(StepsCommon):
     test_pubsub_topic = VALID_PUBSUB_TOPICS[0]
     test_payload = "Light push works!!"
     default_message_propagation_delay = 0.1
+    num_shards_in_network = 8
 
     @pytest.fixture(scope="function", autouse=True)
     def light_push_setup(self):

tests/light_push/test_publish.py

@@ -155,6 +155,8 @@ class TestLightPushPublish(StepsLightPush):
         assert not success_content_topics, f"Invalid Content topics that didn't failed: {success_content_topics}"
 
     def test_light_push_with_missing_pubsub_topics(self):
+        # use auto sharding so the shard is derived from the content topic instead of a pubsub topic
+        self.setup_first_lightpush_node(num_shards_in_network=self.num_shards_in_network)
         try:
             self.light_push_node1.send_light_push_message({"message": self.create_message()})
         except Exception as ex:

tests/sharding/test_running_nodes_auto_sharding.py

@@ -101,7 +101,9 @@ class TestRunningNodesAutosharding(StepsSharding):
             self.check_published_message_reaches_relay_peer(pubsub_topic=pubsub_topic)
 
     def test_node_uses_both_auto_and_regular_apis(self):
-        self.setup_main_relay_nodes(cluster_id=self.auto_cluster, pubsub_topic=self.test_pubsub_topic)
+        self.setup_main_relay_nodes(
+            cluster_id=self.auto_cluster, num_shards_in_network=self.num_shards_in_network, pubsub_topic=self.test_pubsub_topic
+        )
         self.subscribe_main_relay_nodes(content_topics=["/toychat/2/huilong/proto"])
         self.check_published_message_reaches_relay_peer(content_topic="/toychat/2/huilong/proto")
         self.subscribe_main_relay_nodes(pubsub_topics=[self.test_pubsub_topic])