1
0
Fork 0
mirror of https://github.com/dragonflydb/dragonfly.git synced 2024-12-14 11:58:02 +00:00
dragonflydb-dragonfly/tests/dragonfly/list_family_test.py
Kostas Kyrimis 7944af3c62
feat: Add black formatter to the project (#1544)
Add black formatter and run it on pytests
2023-07-17 13:13:12 +03:00

22 lines
902 B
Python

import asyncio
from redis import asyncio as aioredis
import pytest
@pytest.mark.parametrize("index", range(50))
class TestBlPop:
async def async_blpop(client: aioredis.Redis):
return await client.blpop(["list1{t}", "list2{t}", "list2{t}", "list1{t}"], 0.5)
async def blpop_mult_keys(async_client: aioredis.Redis, key: str, val: str):
task = asyncio.create_task(TestBlPop.async_blpop(async_client))
await async_client.lpush(key, val)
result = await asyncio.wait_for(task, 3)
assert result[1] == val
watched = await async_client.execute_command("DEBUG WATCHED")
assert watched == ["awaked", [], "watched", []]
async def test_blpop_multiple_keys(self, async_client: aioredis.Redis, index):
await TestBlPop.blpop_mult_keys(async_client, "list1{t}", "a")
await TestBlPop.blpop_mult_keys(async_client, "list2{t}", "b")