Added timeouts for index creation

Author: vladvildanov

tests/test_asyncio/test_command_policies.py

@@ -1,3 +1,4 @@
+import asyncio
 import random
 
 import pytest
@@ -114,6 +115,9 @@ async def wrapper(*args, request_policy: RequestPolicy, **kwargs):
             )
             assert determined_nodes[0] == primary_nodes[0]
 
+            # Wait for index creation
+            await asyncio.sleep(1)
+
             # Routed to another random primary node
             info = await r.ft().info()
             assert info["index_name"] == "idx"

tests/test_command_policies.py

@@ -1,4 +1,5 @@
 import random
+from time import sleep
 from unittest.mock import Mock, patch
 
 import pytest
@@ -11,6 +12,7 @@
 from redis.commands.search.aggregation import AggregateRequest
 from redis.commands.search.field import TextField, NumericField
 from tests.conftest import skip_if_server_version_lt
+from tests.test_search import waitForIndex
 
 
 @pytest.mark.onlycluster
@@ -115,6 +117,9 @@ def wrapper(*args, request_policy: RequestPolicy, **kwargs):
             )
             assert determined_nodes[0] == primary_nodes[0]
 
+            # Wait for index creation
+            sleep(1)
+
             # Routed to another random primary node
             info = r.ft().info()
             assert info["index_name"] == "idx"