Updated the AnyIO tests based on updated asyncio tests

Author: agronholm

tests/test_anyio/test_bloom.py

@@ -334,6 +334,34 @@ async def test_topk(decoded_r: redis.Redis):
     assert 0.9 == round(float(info["decay"]), 1)
 
 
+@pytest.mark.redismod
+async def test_topk_list_with_special_words(decoded_r: redis.Redis):
+    # test list with empty buckets
+    assert await decoded_r.topk().reserve("topklist:specialwords", 5, 20, 4, 0.9)
+    assert await decoded_r.topk().add(
+        "topklist:specialwords",
+        "infinity",
+        "B",
+        "nan",
+        "D",
+        "-infinity",
+        "infinity",
+        "infinity",
+        "B",
+        "nan",
+        "G",
+        "D",
+        "B",
+        "D",
+        "infinity",
+        "-infinity",
+        "-infinity",
+    )
+    assert ["infinity", "B", "D", "-infinity", "nan"] == await decoded_r.topk().list(
+        "topklist:specialwords"
+    )
+
+
 @pytest.mark.redismod
 async def test_topk_incrby(decoded_r: redis.Redis):
     await decoded_r.flushdb()

tests/test_anyio/test_cluster.py

@@ -57,7 +57,7 @@
 if sys.version_info < (3, 11):
     from exceptiongroup import ExceptionGroup
 
-pytestmark = [pytest.mark.anyio, pytest.mark.onlycluster]
+pytestmark = [pytest.mark.onlycluster, pytest.mark.anyio]
 
 
 default_host = "127.0.0.1"
@@ -67,12 +67,6 @@
     [8192, 16383, ["127.0.0.1", 7001, "node_1"], ["127.0.0.1", 7002, "node_2"]],
 ]
 
-@pytest.fixture(autouse=True)
-def run_gc():
-    import gc
-
-    yield
-    gc.collect()
 
 class NodeProxy:
     """A class to proxy a node connection to a different port"""

tests/test_anyio/test_search.py

@@ -51,8 +51,8 @@ async def decoded_r(create_redis, stack_url):
 async def waitForIndex(env, idx, timeout=None):
     delay = 0.1
     while True:
-        res = await env.execute_command("FT.INFO", idx)
         try:
+            res = await env.execute_command("FT.INFO", idx)
             if int(res[res.index("indexing") + 1]) == 0:
                 break
         except ValueError:
@@ -64,7 +64,7 @@ async def waitForIndex(env, idx, timeout=None):
             except ValueError:
                 break
 
-        time.sleep(delay)
+        await anyio.sleep(delay)
         if timeout is not None:
             timeout -= delay
             if timeout <= 0:
@@ -1767,7 +1767,7 @@ async def test_binary_and_text_fields(decoded_r: redis.Redis):
     mixed_data = {"first_name": "🐍python", "vector_emb": fake_vec.tobytes()}
     await decoded_r.hset(f"{index_name}:1", mapping=mixed_data)
 
-    schema = (
+    schema = [
         TagField("first_name"),
         VectorField(
             "embeddings_bio",
@@ -1778,14 +1778,15 @@ async def test_binary_and_text_fields(decoded_r: redis.Redis):
                 "DISTANCE_METRIC": "COSINE",
             },
         ),
-    )
+    ]
 
     await decoded_r.ft(index_name).create_index(
         fields=schema,
         definition=IndexDefinition(
             prefix=[f"{index_name}:"], index_type=IndexType.HASH
         ),
     )
+    await waitForIndex(decoded_r, index_name)
 
     query = (
         Query("*")
@@ -1795,6 +1796,12 @@ async def test_binary_and_text_fields(decoded_r: redis.Redis):
     result = await decoded_r.ft(index_name).search(query=query, query_params={})
     docs = result.docs
 
+    if len(docs) == 0:
+        hash_content = await decoded_r.hget(f"{index_name}:1", "first_name")
+    assert len(docs) > 0, (
+        f"Returned search results are empty. Result: {result}; Hash: {hash_content}"
+    )
+
     decoded_vec_from_search_results = np.frombuffer(
         docs[0]["vector_emb"], dtype=np.float32
     )

tests/test_anyio/test_timeseries.py

@@ -6,6 +6,8 @@
 from tests.conftest import (
     assert_resp_response,
     is_resp2_connection,
+    skip_if_server_version_gte,
+    skip_if_server_version_lt,
     skip_ifmodversion_lt,
 )
 
@@ -77,7 +79,24 @@ async def test_alter(decoded_r: redis.Redis):
 
 @pytest.mark.redismod
 @skip_ifmodversion_lt("1.4.0", "timeseries")
+@skip_if_server_version_lt("7.9.0")
 async def test_alter_duplicate_policy(decoded_r: redis.Redis):
+    assert await decoded_r.ts().create(1)
+    info = await decoded_r.ts().info(1)
+    assert_resp_response(
+        decoded_r, "block", info.get("duplicate_policy"), info.get("duplicatePolicy")
+    )
+    assert await decoded_r.ts().alter(1, duplicate_policy="min")
+    info = await decoded_r.ts().info(1)
+    assert_resp_response(
+        decoded_r, "min", info.get("duplicate_policy"), info.get("duplicatePolicy")
+    )
+
+
+@pytest.mark.redismod
+@skip_ifmodversion_lt("1.4.0", "timeseries")
+@skip_if_server_version_gte("7.9.0")
+async def test_alter_duplicate_policy_prior_redis_8(decoded_r: redis.Redis):
     assert await decoded_r.ts().create(1)
     info = await decoded_r.ts().info(1)
     assert_resp_response(
@@ -724,7 +743,27 @@ async def test_info(decoded_r: redis.Redis):
 
 @pytest.mark.redismod
 @skip_ifmodversion_lt("1.4.0", "timeseries")
+@skip_if_server_version_lt("7.9.0")
 async def test_info_duplicate_policy(decoded_r: redis.Redis):
+    await decoded_r.ts().create(
+        1, retention_msecs=5, labels={"currentLabel": "currentData"}
+    )
+    info = await decoded_r.ts().info(1)
+    assert_resp_response(
+        decoded_r, "block", info.get("duplicate_policy"), info.get("duplicatePolicy")
+    )
+
+    await decoded_r.ts().create("time-serie-2", duplicate_policy="min")
+    info = await decoded_r.ts().info("time-serie-2")
+    assert_resp_response(
+        decoded_r, "min", info.get("duplicate_policy"), info.get("duplicatePolicy")
+    )
+
+
+@pytest.mark.redismod
+@skip_ifmodversion_lt("1.4.0", "timeseries")
+@skip_if_server_version_gte("7.9.0")
+async def test_info_duplicate_policy_prior_redis_8(decoded_r: redis.Redis):
     await decoded_r.ts().create(
         1, retention_msecs=5, labels={"currentLabel": "currentData"}
     )