fix: add timeout to tests before lora to unblock (#4808)

Signed-off-by: Keiven Chang <[email protected]>
Co-authored-by: Keiven Chang <[email protected]>

Author: keivenchang

lib/bindings/python/tests/test_kv_bindings.py

@@ -36,8 +36,8 @@ async def distributed_runtime():
     runtime.shutdown()
 
 
-@pytest.mark.asyncio
-async def test_radix_tree_binding(distributed_runtime):
+@pytest.mark.timeout(5)  # Expected: ~1s, timeout set to 5x for safety
+def test_radix_tree_binding():
     """Test RadixTree binding directly with store event and find matches"""
     import json
 
@@ -102,13 +102,12 @@ async def test_radix_tree_binding(distributed_runtime):
     )
 
 
-@pytest.mark.asyncio
+@pytest.mark.timeout(5)  # Expected: ~1s, timeout set to 5x for safety
 @pytest.mark.parametrize("num_threads", [2, 3, 5, 128])
 @pytest.mark.parametrize("prepopulate_worker_ids", [True, False])
 @pytest.mark.parametrize("expiration_duration_secs", [None])
 @pytest.mark.parametrize("is_threaded", [True, False])
-async def test_radix_tree_thread_safety(
-    distributed_runtime,
+def test_radix_tree_thread_safety(
     num_threads,
     prepopulate_worker_ids,
     expiration_duration_secs,
@@ -205,6 +204,7 @@ def worker(worker_id, prepopulate_worker_ids: bool = False):
 
 
 @pytest.mark.asyncio
+@pytest.mark.timeout(5)  # Expected: ~1s, timeout set to 5x for safety
 async def test_event_handler(distributed_runtime):
     kv_block_size = 32
     namespace = "kv_test"
@@ -247,6 +247,7 @@ async def test_event_handler(distributed_runtime):
 
 
 @pytest.mark.asyncio
+@pytest.mark.timeout(5)  # Expected: ~1s, timeout set to 5x for safety
 async def test_approx_kv_indexer(distributed_runtime):
     """Test ApproxKvIndexer with TTL-based block tracking"""
     kv_block_size = 32