summaryrefslogtreecommitdiff
path: root/tests/test_asyncio/test_commands.py
diff options
context:
space:
mode:
authordvora-h <67596500+dvora-h@users.noreply.github.com>2022-04-07 14:46:11 +0300
committerGitHub <noreply@github.com>2022-04-07 14:46:11 +0300
commitcb91eed751c226ebca5b1b8e8d62769fdee9bfbc (patch)
treea9e8decb34610a4c483db9c50849cca64e47afa6 /tests/test_asyncio/test_commands.py
parent20e4b873151804f596fa324c38f18fae175c2e63 (diff)
downloadredis-py-cb91eed751c226ebca5b1b8e8d62769fdee9bfbc.tar.gz
Async tests for redis commands, json, bloom, timeseries (#2087)
* test async commands * test async json commands * test async timeseries commands * test async bloom commands * linters * linters * skip on cluster * linters * linters * isort * flynt * remove print
Diffstat (limited to 'tests/test_asyncio/test_commands.py')
-rw-r--r--tests/test_asyncio/test_commands.py3049
1 files changed, 3049 insertions, 0 deletions
diff --git a/tests/test_asyncio/test_commands.py b/tests/test_asyncio/test_commands.py
index 12855c9..7822040 100644
--- a/tests/test_asyncio/test_commands.py
+++ b/tests/test_asyncio/test_commands.py
@@ -1,3 +1,3052 @@
"""
Tests async overrides of commands from their mixins
"""
+import binascii
+import datetime
+import re
+import time
+from string import ascii_letters
+
+import pytest
+
+import redis
+from redis import exceptions
+from redis.client import parse_info
+from tests.conftest import (
+ skip_if_server_version_gte,
+ skip_if_server_version_lt,
+ skip_unless_arch_bits,
+)
+
+REDIS_6_VERSION = "5.9.0"
+
+
+pytestmark = [pytest.mark.asyncio, pytest.mark.onlynoncluster]
+
+
+@pytest.fixture()
+async def slowlog(r: redis.Redis, event_loop):
+ current_config = await r.config_get()
+ old_slower_than_value = current_config["slowlog-log-slower-than"]
+ old_max_legnth_value = current_config["slowlog-max-len"]
+
+ await r.config_set("slowlog-log-slower-than", 0)
+ await r.config_set("slowlog-max-len", 128)
+
+ yield
+
+ await r.config_set("slowlog-log-slower-than", old_slower_than_value)
+ await r.config_set("slowlog-max-len", old_max_legnth_value)
+
+
+async def redis_server_time(client: redis.Redis):
+ seconds, milliseconds = await client.time()
+ timestamp = float(f"{seconds}.{milliseconds}")
+ return datetime.datetime.fromtimestamp(timestamp)
+
+
+async def get_stream_message(client: redis.Redis, stream: str, message_id: str):
+ """Fetch a stream message and format it as a (message_id, fields) pair"""
+ response = await client.xrange(stream, min=message_id, max=message_id)
+ assert len(response) == 1
+ return response[0]
+
+
+# RESPONSE CALLBACKS
+class TestResponseCallbacks:
+ """Tests for the response callback system"""
+
+ async def test_response_callbacks(self, r: redis.Redis):
+ assert r.response_callbacks == redis.Redis.RESPONSE_CALLBACKS
+ assert id(r.response_callbacks) != id(redis.Redis.RESPONSE_CALLBACKS)
+ r.set_response_callback("GET", lambda x: "static")
+ await r.set("a", "foo")
+ assert await r.get("a") == "static"
+
+ async def test_case_insensitive_command_names(self, r: redis.Redis):
+ assert r.response_callbacks["del"] == r.response_callbacks["DEL"]
+
+
+class TestRedisCommands:
+ async def test_command_on_invalid_key_type(self, r: redis.Redis):
+ await r.lpush("a", "1")
+ with pytest.raises(redis.ResponseError):
+ await r.get("a")
+
+ # SERVER INFORMATION
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_cat_no_category(self, r: redis.Redis):
+ categories = await r.acl_cat()
+ assert isinstance(categories, list)
+ assert "read" in categories
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_cat_with_category(self, r: redis.Redis):
+ commands = await r.acl_cat("read")
+ assert isinstance(commands, list)
+ assert "get" in commands
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_deluser(self, r: redis.Redis, request, event_loop):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ assert await r.acl_deluser(username) == 0
+ assert await r.acl_setuser(username, enabled=False, reset=True)
+ assert await r.acl_deluser(username) == 1
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_genpass(self, r: redis.Redis):
+ password = await r.acl_genpass()
+ assert isinstance(password, str)
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_getuser_setuser(self, r: redis.Redis, request, event_loop):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ # test enabled=False
+ assert await r.acl_setuser(username, enabled=False, reset=True)
+ assert await r.acl_getuser(username) == {
+ "categories": ["-@all"],
+ "commands": [],
+ "channels": [b"*"],
+ "enabled": False,
+ "flags": ["off", "allchannels", "sanitize-payload"],
+ "keys": [],
+ "passwords": [],
+ }
+
+ # test nopass=True
+ assert await r.acl_setuser(username, enabled=True, reset=True, nopass=True)
+ assert await r.acl_getuser(username) == {
+ "categories": ["-@all"],
+ "commands": [],
+ "channels": [b"*"],
+ "enabled": True,
+ "flags": ["on", "allchannels", "nopass", "sanitize-payload"],
+ "keys": [],
+ "passwords": [],
+ }
+
+ # test all args
+ assert await r.acl_setuser(
+ username,
+ enabled=True,
+ reset=True,
+ passwords=["+pass1", "+pass2"],
+ categories=["+set", "+@hash", "-geo"],
+ commands=["+get", "+mget", "-hset"],
+ keys=["cache:*", "objects:*"],
+ )
+ acl = await r.acl_getuser(username)
+ assert set(acl["categories"]) == {"-@all", "+@set", "+@hash"}
+ assert set(acl["commands"]) == {"+get", "+mget", "-hset"}
+ assert acl["enabled"] is True
+ assert acl["channels"] == [b"*"]
+ assert acl["flags"] == ["on", "allchannels", "sanitize-payload"]
+ assert set(acl["keys"]) == {b"cache:*", b"objects:*"}
+ assert len(acl["passwords"]) == 2
+
+ # test reset=False keeps existing ACL and applies new ACL on top
+ assert await r.acl_setuser(
+ username,
+ enabled=True,
+ reset=True,
+ passwords=["+pass1"],
+ categories=["+@set"],
+ commands=["+get"],
+ keys=["cache:*"],
+ )
+ assert await r.acl_setuser(
+ username,
+ enabled=True,
+ passwords=["+pass2"],
+ categories=["+@hash"],
+ commands=["+mget"],
+ keys=["objects:*"],
+ )
+ acl = await r.acl_getuser(username)
+ assert set(acl["categories"]) == {"-@all", "+@set", "+@hash"}
+ assert set(acl["commands"]) == {"+get", "+mget"}
+ assert acl["enabled"] is True
+ assert acl["channels"] == [b"*"]
+ assert acl["flags"] == ["on", "allchannels", "sanitize-payload"]
+ assert set(acl["keys"]) == {b"cache:*", b"objects:*"}
+ assert len(acl["passwords"]) == 2
+
+ # test removal of passwords
+ assert await r.acl_setuser(
+ username, enabled=True, reset=True, passwords=["+pass1", "+pass2"]
+ )
+ assert len((await r.acl_getuser(username))["passwords"]) == 2
+ assert await r.acl_setuser(username, enabled=True, passwords=["-pass2"])
+ assert len((await r.acl_getuser(username))["passwords"]) == 1
+
+ # Resets and tests that hashed passwords are set properly.
+ hashed_password = (
+ "5e884898da28047151d0e56f8dc629" "2773603d0d6aabbdd62a11ef721d1542d8"
+ )
+ assert await r.acl_setuser(
+ username, enabled=True, reset=True, hashed_passwords=["+" + hashed_password]
+ )
+ acl = await r.acl_getuser(username)
+ assert acl["passwords"] == [hashed_password]
+
+ # test removal of hashed passwords
+ assert await r.acl_setuser(
+ username,
+ enabled=True,
+ reset=True,
+ hashed_passwords=["+" + hashed_password],
+ passwords=["+pass1"],
+ )
+ assert len((await r.acl_getuser(username))["passwords"]) == 2
+ assert await r.acl_setuser(
+ username, enabled=True, hashed_passwords=["-" + hashed_password]
+ )
+ assert len((await r.acl_getuser(username))["passwords"]) == 1
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_list(self, r: redis.Redis, request, event_loop):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ assert await r.acl_setuser(username, enabled=False, reset=True)
+ users = await r.acl_list()
+ assert f"user {username} off sanitize-payload &* -@all" in users
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_log(self, r: redis.Redis, request, event_loop, create_redis):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+ await r.acl_setuser(
+ username,
+ enabled=True,
+ reset=True,
+ commands=["+get", "+set", "+select"],
+ keys=["cache:*"],
+ nopass=True,
+ )
+ await r.acl_log_reset()
+
+ user_client = await create_redis(username=username)
+
+ # Valid operation and key
+ assert await user_client.set("cache:0", 1)
+ assert await user_client.get("cache:0") == b"1"
+
+ # Invalid key
+ with pytest.raises(exceptions.NoPermissionError):
+ await user_client.get("violated_cache:0")
+
+ # Invalid operation
+ with pytest.raises(exceptions.NoPermissionError):
+ await user_client.hset("cache:0", "hkey", "hval")
+
+ assert isinstance(await r.acl_log(), list)
+ assert len(await r.acl_log()) == 2
+ assert len(await r.acl_log(count=1)) == 1
+ assert isinstance((await r.acl_log())[0], dict)
+ assert "client-info" in (await r.acl_log(count=1))[0]
+ assert await r.acl_log_reset()
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_setuser_categories_without_prefix_fails(
+ self, r: redis.Redis, request, event_loop
+ ):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ with pytest.raises(exceptions.DataError):
+ await r.acl_setuser(username, categories=["list"])
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_setuser_commands_without_prefix_fails(
+ self, r: redis.Redis, request, event_loop
+ ):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ with pytest.raises(exceptions.DataError):
+ await r.acl_setuser(username, commands=["get"])
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_setuser_add_passwords_and_nopass_fails(
+ self, r: redis.Redis, request, event_loop
+ ):
+ username = "redis-py-user"
+
+ def teardown():
+ coro = r.acl_deluser(username)
+ if event_loop.is_running():
+ event_loop.create_task(coro)
+ else:
+ event_loop.run_until_complete(coro)
+
+ request.addfinalizer(teardown)
+
+ with pytest.raises(exceptions.DataError):
+ await r.acl_setuser(username, passwords="+mypass", nopass=True)
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_users(self, r: redis.Redis):
+ users = await r.acl_users()
+ assert isinstance(users, list)
+ assert len(users) > 0
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_acl_whoami(self, r: redis.Redis):
+ username = await r.acl_whoami()
+ assert isinstance(username, str)
+
+ async def test_client_list(self, r: redis.Redis):
+ clients = await r.client_list()
+ assert isinstance(clients[0], dict)
+ assert "addr" in clients[0]
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_client_list_type(self, r: redis.Redis):
+ with pytest.raises(exceptions.RedisError):
+ await r.client_list(_type="not a client type")
+ for client_type in ["normal", "master", "replica", "pubsub"]:
+ clients = await r.client_list(_type=client_type)
+ assert isinstance(clients, list)
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_client_id(self, r: redis.Redis):
+ assert await r.client_id() > 0
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_client_unblock(self, r: redis.Redis):
+ myid = await r.client_id()
+ assert not await r.client_unblock(myid)
+ assert not await r.client_unblock(myid, error=True)
+ assert not await r.client_unblock(myid, error=False)
+
+ @skip_if_server_version_lt("2.6.9")
+ async def test_client_getname(self, r: redis.Redis):
+ assert await r.client_getname() is None
+
+ @skip_if_server_version_lt("2.6.9")
+ async def test_client_setname(self, r: redis.Redis):
+ assert await r.client_setname("redis_py_test")
+ assert await r.client_getname() == "redis_py_test"
+
+ @skip_if_server_version_lt("2.6.9")
+ async def test_client_kill(self, r: redis.Redis, r2):
+ await r.client_setname("redis-py-c1")
+ await r2.client_setname("redis-py-c2")
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 2
+
+ clients_by_name = {client.get("name"): client for client in clients}
+
+ client_addr = clients_by_name["redis-py-c2"].get("addr")
+ assert await r.client_kill(client_addr) is True
+
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 1
+ assert clients[0].get("name") == "redis-py-c1"
+
+ @skip_if_server_version_lt("2.8.12")
+ async def test_client_kill_filter_invalid_params(self, r: redis.Redis):
+ # empty
+ with pytest.raises(exceptions.DataError):
+ await r.client_kill_filter()
+
+ # invalid skipme
+ with pytest.raises(exceptions.DataError):
+ await r.client_kill_filter(skipme="yeah") # type: ignore
+
+ # invalid type
+ with pytest.raises(exceptions.DataError):
+ await r.client_kill_filter(_type="caster") # type: ignore
+
+ @skip_if_server_version_lt("2.8.12")
+ async def test_client_kill_filter_by_id(self, r: redis.Redis, r2):
+ await r.client_setname("redis-py-c1")
+ await r2.client_setname("redis-py-c2")
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 2
+
+ clients_by_name = {client.get("name"): client for client in clients}
+
+ client_2_id = clients_by_name["redis-py-c2"].get("id")
+ resp = await r.client_kill_filter(_id=client_2_id)
+ assert resp == 1
+
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 1
+ assert clients[0].get("name") == "redis-py-c1"
+
+ @skip_if_server_version_lt("2.8.12")
+ async def test_client_kill_filter_by_addr(self, r: redis.Redis, r2):
+ await r.client_setname("redis-py-c1")
+ await r2.client_setname("redis-py-c2")
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 2
+
+ clients_by_name = {client.get("name"): client for client in clients}
+
+ client_2_addr = clients_by_name["redis-py-c2"].get("addr")
+ resp = await r.client_kill_filter(addr=client_2_addr)
+ assert resp == 1
+
+ clients = [
+ client
+ for client in await r.client_list()
+ if client.get("name") in ["redis-py-c1", "redis-py-c2"]
+ ]
+ assert len(clients) == 1
+ assert clients[0].get("name") == "redis-py-c1"
+
+ @skip_if_server_version_lt("2.6.9")
+ async def test_client_list_after_client_setname(self, r: redis.Redis):
+ await r.client_setname("redis_py_test")
+ clients = await r.client_list()
+ # we don't know which client ours will be
+ assert "redis_py_test" in [c["name"] for c in clients]
+
+ @skip_if_server_version_lt("2.9.50")
+ async def test_client_pause(self, r: redis.Redis):
+ assert await r.client_pause(1)
+ assert await r.client_pause(timeout=1)
+ with pytest.raises(exceptions.RedisError):
+ await r.client_pause(timeout="not an integer")
+
+ async def test_config_get(self, r: redis.Redis):
+ data = await r.config_get()
+ assert "maxmemory" in data
+ assert data["maxmemory"].isdigit()
+
+ async def test_config_resetstat(self, r: redis.Redis):
+ await r.ping()
+ prior_commands_processed = int((await r.info())["total_commands_processed"])
+ assert prior_commands_processed >= 1
+ await r.config_resetstat()
+ reset_commands_processed = int((await r.info())["total_commands_processed"])
+ assert reset_commands_processed < prior_commands_processed
+
+ async def test_config_set(self, r: redis.Redis):
+ data = await r.config_get()
+ rdbname = data["dbfilename"]
+ try:
+ assert await r.config_set("dbfilename", "redis_py_test.rdb")
+ assert (await r.config_get())["dbfilename"] == "redis_py_test.rdb"
+ finally:
+ assert await r.config_set("dbfilename", rdbname)
+
+ async def test_dbsize(self, r: redis.Redis):
+ await r.set("a", "foo")
+ await r.set("b", "bar")
+ assert await r.dbsize() == 2
+
+ async def test_echo(self, r: redis.Redis):
+ assert await r.echo("foo bar") == b"foo bar"
+
+ async def test_info(self, r: redis.Redis):
+ await r.set("a", "foo")
+ await r.set("b", "bar")
+ info = await r.info()
+ assert isinstance(info, dict)
+ assert info["db9"]["keys"] == 2
+
+ async def test_lastsave(self, r: redis.Redis):
+ assert isinstance(await r.lastsave(), datetime.datetime)
+
+ async def test_object(self, r: redis.Redis):
+ await r.set("a", "foo")
+ assert isinstance(await r.object("refcount", "a"), int)
+ assert isinstance(await r.object("idletime", "a"), int)
+ assert await r.object("encoding", "a") in (b"raw", b"embstr")
+ assert await r.object("idletime", "invalid-key") is None
+
+ async def test_ping(self, r: redis.Redis):
+ assert await r.ping()
+
+ async def test_slowlog_get(self, r: redis.Redis, slowlog):
+ assert await r.slowlog_reset()
+ unicode_string = chr(3456) + "abcd" + chr(3421)
+ await r.get(unicode_string)
+ slowlog = await r.slowlog_get()
+ assert isinstance(slowlog, list)
+ commands = [log["command"] for log in slowlog]
+
+ get_command = b" ".join((b"GET", unicode_string.encode("utf-8")))
+ assert get_command in commands
+ assert b"SLOWLOG RESET" in commands
+ # the order should be ['GET <uni string>', 'SLOWLOG RESET'],
+ # but if other clients are executing commands at the same time, there
+ # could be commands, before, between, or after, so just check that
+ # the two we care about are in the appropriate order.
+ assert commands.index(get_command) < commands.index(b"SLOWLOG RESET")
+
+ # make sure other attributes are typed correctly
+ assert isinstance(slowlog[0]["start_time"], int)
+ assert isinstance(slowlog[0]["duration"], int)
+
+ async def test_slowlog_get_limit(self, r: redis.Redis, slowlog):
+ assert await r.slowlog_reset()
+ await r.get("foo")
+ slowlog = await r.slowlog_get(1)
+ assert isinstance(slowlog, list)
+ # only one command, based on the number we passed to slowlog_get()
+ assert len(slowlog) == 1
+
+ async def test_slowlog_length(self, r: redis.Redis, slowlog):
+ await r.get("foo")
+ assert isinstance(await r.slowlog_len(), int)
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_time(self, r: redis.Redis):
+ t = await r.time()
+ assert len(t) == 2
+ assert isinstance(t[0], int)
+ assert isinstance(t[1], int)
+
+ # BASIC KEY COMMANDS
+ async def test_append(self, r: redis.Redis):
+ assert await r.append("a", "a1") == 2
+ assert await r.get("a") == b"a1"
+ assert await r.append("a", "a2") == 4
+ assert await r.get("a") == b"a1a2"
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitcount(self, r: redis.Redis):
+ await r.setbit("a", 5, True)
+ assert await r.bitcount("a") == 1
+ await r.setbit("a", 6, True)
+ assert await r.bitcount("a") == 2
+ await r.setbit("a", 5, False)
+ assert await r.bitcount("a") == 1
+ await r.setbit("a", 9, True)
+ await r.setbit("a", 17, True)
+ await r.setbit("a", 25, True)
+ await r.setbit("a", 33, True)
+ assert await r.bitcount("a") == 5
+ assert await r.bitcount("a", 0, -1) == 5
+ assert await r.bitcount("a", 2, 3) == 2
+ assert await r.bitcount("a", 2, -1) == 3
+ assert await r.bitcount("a", -2, -1) == 2
+ assert await r.bitcount("a", 1, 1) == 1
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitop_not_empty_string(self, r: redis.Redis):
+ await r.set("a", "")
+ await r.bitop("not", "r", "a")
+ assert await r.get("r") is None
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitop_not(self, r: redis.Redis):
+ test_str = b"\xAA\x00\xFF\x55"
+ correct = ~0xAA00FF55 & 0xFFFFFFFF
+ await r.set("a", test_str)
+ await r.bitop("not", "r", "a")
+ assert int(binascii.hexlify(await r.get("r")), 16) == correct
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitop_not_in_place(self, r: redis.Redis):
+ test_str = b"\xAA\x00\xFF\x55"
+ correct = ~0xAA00FF55 & 0xFFFFFFFF
+ await r.set("a", test_str)
+ await r.bitop("not", "a", "a")
+ assert int(binascii.hexlify(await r.get("a")), 16) == correct
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitop_single_string(self, r: redis.Redis):
+ test_str = b"\x01\x02\xFF"
+ await r.set("a", test_str)
+ await r.bitop("and", "res1", "a")
+ await r.bitop("or", "res2", "a")
+ await r.bitop("xor", "res3", "a")
+ assert await r.get("res1") == test_str
+ assert await r.get("res2") == test_str
+ assert await r.get("res3") == test_str
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_bitop_string_operands(self, r: redis.Redis):
+ await r.set("a", b"\x01\x02\xFF\xFF")
+ await r.set("b", b"\x01\x02\xFF")
+ await r.bitop("and", "res1", "a", "b")
+ await r.bitop("or", "res2", "a", "b")
+ await r.bitop("xor", "res3", "a", "b")
+ assert int(binascii.hexlify(await r.get("res1")), 16) == 0x0102FF00
+ assert int(binascii.hexlify(await r.get("res2")), 16) == 0x0102FFFF
+ assert int(binascii.hexlify(await r.get("res3")), 16) == 0x000000FF
+
+ @skip_if_server_version_lt("2.8.7")
+ async def test_bitpos(self, r: redis.Redis):
+ key = "key:bitpos"
+ await r.set(key, b"\xff\xf0\x00")
+ assert await r.bitpos(key, 0) == 12
+ assert await r.bitpos(key, 0, 2, -1) == 16
+ assert await r.bitpos(key, 0, -2, -1) == 12
+ await r.set(key, b"\x00\xff\xf0")
+ assert await r.bitpos(key, 1, 0) == 8
+ assert await r.bitpos(key, 1, 1) == 8
+ await r.set(key, b"\x00\x00\x00")
+ assert await r.bitpos(key, 1) == -1
+
+ @skip_if_server_version_lt("2.8.7")
+ async def test_bitpos_wrong_arguments(self, r: redis.Redis):
+ key = "key:bitpos:wrong:args"
+ await r.set(key, b"\xff\xf0\x00")
+ with pytest.raises(exceptions.RedisError):
+ await r.bitpos(key, 0, end=1) == 12
+ with pytest.raises(exceptions.RedisError):
+ await r.bitpos(key, 7) == 12
+
+ async def test_decr(self, r: redis.Redis):
+ assert await r.decr("a") == -1
+ assert await r.get("a") == b"-1"
+ assert await r.decr("a") == -2
+ assert await r.get("a") == b"-2"
+ assert await r.decr("a", amount=5) == -7
+ assert await r.get("a") == b"-7"
+
+ async def test_decrby(self, r: redis.Redis):
+ assert await r.decrby("a", amount=2) == -2
+ assert await r.decrby("a", amount=3) == -5
+ assert await r.get("a") == b"-5"
+
+ async def test_delete(self, r: redis.Redis):
+ assert await r.delete("a") == 0
+ await r.set("a", "foo")
+ assert await r.delete("a") == 1
+
+ async def test_delete_with_multiple_keys(self, r: redis.Redis):
+ await r.set("a", "foo")
+ await r.set("b", "bar")
+ assert await r.delete("a", "b") == 2
+ assert await r.get("a") is None
+ assert await r.get("b") is None
+
+ async def test_delitem(self, r: redis.Redis):
+ await r.set("a", "foo")
+ await r.delete("a")
+ assert await r.get("a") is None
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_unlink(self, r: redis.Redis):
+ assert await r.unlink("a") == 0
+ await r.set("a", "foo")
+ assert await r.unlink("a") == 1
+ assert await r.get("a") is None
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_unlink_with_multiple_keys(self, r: redis.Redis):
+ await r.set("a", "foo")
+ await r.set("b", "bar")
+ assert await r.unlink("a", "b") == 2
+ assert await r.get("a") is None
+ assert await r.get("b") is None
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_dump_and_restore(self, r: redis.Redis):
+ await r.set("a", "foo")
+ dumped = await r.dump("a")
+ await r.delete("a")
+ await r.restore("a", 0, dumped)
+ assert await r.get("a") == b"foo"
+
+ @skip_if_server_version_lt("3.0.0")
+ async def test_dump_and_restore_and_replace(self, r: redis.Redis):
+ await r.set("a", "bar")
+ dumped = await r.dump("a")
+ with pytest.raises(redis.ResponseError):
+ await r.restore("a", 0, dumped)
+
+ await r.restore("a", 0, dumped, replace=True)
+ assert await r.get("a") == b"bar"
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_dump_and_restore_absttl(self, r: redis.Redis):
+ await r.set("a", "foo")
+ dumped = await r.dump("a")
+ await r.delete("a")
+ ttl = int(
+ (await redis_server_time(r) + datetime.timedelta(minutes=1)).timestamp()
+ * 1000
+ )
+ await r.restore("a", ttl, dumped, absttl=True)
+ assert await r.get("a") == b"foo"
+ assert 0 < await r.ttl("a") <= 61
+
+ async def test_exists(self, r: redis.Redis):
+ assert await r.exists("a") == 0
+ await r.set("a", "foo")
+ await r.set("b", "bar")
+ assert await r.exists("a") == 1
+ assert await r.exists("a", "b") == 2
+
+ async def test_exists_contains(self, r: redis.Redis):
+ assert not await r.exists("a")
+ await r.set("a", "foo")
+ assert await r.exists("a")
+
+ async def test_expire(self, r: redis.Redis):
+ assert not await r.expire("a", 10)
+ await r.set("a", "foo")
+ assert await r.expire("a", 10)
+ assert 0 < await r.ttl("a") <= 10
+ assert await r.persist("a")
+ assert await r.ttl("a") == -1
+
+ async def test_expireat_datetime(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ await r.set("a", "foo")
+ assert await r.expireat("a", expire_at)
+ assert 0 < await r.ttl("a") <= 61
+
+ async def test_expireat_no_key(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ assert not await r.expireat("a", expire_at)
+
+ async def test_expireat_unixtime(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ await r.set("a", "foo")
+ expire_at_seconds = int(time.mktime(expire_at.timetuple()))
+ assert await r.expireat("a", expire_at_seconds)
+ assert 0 < await r.ttl("a") <= 61
+
+ async def test_get_and_set(self, r: redis.Redis):
+ # get and set can't be tested independently of each other
+ assert await r.get("a") is None
+ byte_string = b"value"
+ integer = 5
+ unicode_string = chr(3456) + "abcd" + chr(3421)
+ assert await r.set("byte_string", byte_string)
+ assert await r.set("integer", 5)
+ assert await r.set("unicode_string", unicode_string)
+ assert await r.get("byte_string") == byte_string
+ assert await r.get("integer") == str(integer).encode()
+ assert (await r.get("unicode_string")).decode("utf-8") == unicode_string
+
+ async def test_get_set_bit(self, r: redis.Redis):
+ # no value
+ assert not await r.getbit("a", 5)
+ # set bit 5
+ assert not await r.setbit("a", 5, True)
+ assert await r.getbit("a", 5)
+ # unset bit 4
+ assert not await r.setbit("a", 4, False)
+ assert not await r.getbit("a", 4)
+ # set bit 4
+ assert not await r.setbit("a", 4, True)
+ assert await r.getbit("a", 4)
+ # set bit 5 again
+ assert await r.setbit("a", 5, True)
+ assert await r.getbit("a", 5)
+
+ async def test_getrange(self, r: redis.Redis):
+ await r.set("a", "foo")
+ assert await r.getrange("a", 0, 0) == b"f"
+ assert await r.getrange("a", 0, 2) == b"foo"
+ assert await r.getrange("a", 3, 4) == b""
+
+ async def test_getset(self, r: redis.Redis):
+ assert await r.getset("a", "foo") is None
+ assert await r.getset("a", "bar") == b"foo"
+ assert await r.get("a") == b"bar"
+
+ async def test_incr(self, r: redis.Redis):
+ assert await r.incr("a") == 1
+ assert await r.get("a") == b"1"
+ assert await r.incr("a") == 2
+ assert await r.get("a") == b"2"
+ assert await r.incr("a", amount=5) == 7
+ assert await r.get("a") == b"7"
+
+ async def test_incrby(self, r: redis.Redis):
+ assert await r.incrby("a") == 1
+ assert await r.incrby("a", 4) == 5
+ assert await r.get("a") == b"5"
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_incrbyfloat(self, r: redis.Redis):
+ assert await r.incrbyfloat("a") == 1.0
+ assert await r.get("a") == b"1"
+ assert await r.incrbyfloat("a", 1.1) == 2.1
+ assert float(await r.get("a")) == float(2.1)
+
+ async def test_keys(self, r: redis.Redis):
+ assert await r.keys() == []
+ keys_with_underscores = {b"test_a", b"test_b"}
+ keys = keys_with_underscores.union({b"testc"})
+ for key in keys:
+ await r.set(key, 1)
+ assert set(await r.keys(pattern="test_*")) == keys_with_underscores
+ assert set(await r.keys(pattern="test*")) == keys
+
+ async def test_mget(self, r: redis.Redis):
+ assert await r.mget([]) == []
+ assert await r.mget(["a", "b"]) == [None, None]
+ await r.set("a", "1")
+ await r.set("b", "2")
+ await r.set("c", "3")
+ assert await r.mget("a", "other", "b", "c") == [b"1", None, b"2", b"3"]
+
+ async def test_mset(self, r: redis.Redis):
+ d = {"a": b"1", "b": b"2", "c": b"3"}
+ assert await r.mset(d)
+ for k, v in d.items():
+ assert await r.get(k) == v
+
+ async def test_msetnx(self, r: redis.Redis):
+ d = {"a": b"1", "b": b"2", "c": b"3"}
+ assert await r.msetnx(d)
+ d2 = {"a": b"x", "d": b"4"}
+ assert not await r.msetnx(d2)
+ for k, v in d.items():
+ assert await r.get(k) == v
+ assert await r.get("d") is None
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_pexpire(self, r: redis.Redis):
+ assert not await r.pexpire("a", 60000)
+ await r.set("a", "foo")
+ assert await r.pexpire("a", 60000)
+ assert 0 < await r.pttl("a") <= 60000
+ assert await r.persist("a")
+ assert await r.pttl("a") == -1
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_pexpireat_datetime(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ await r.set("a", "foo")
+ assert await r.pexpireat("a", expire_at)
+ assert 0 < await r.pttl("a") <= 61000
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_pexpireat_no_key(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ assert not await r.pexpireat("a", expire_at)
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_pexpireat_unixtime(self, r: redis.Redis):
+ expire_at = await redis_server_time(r) + datetime.timedelta(minutes=1)
+ await r.set("a", "foo")
+ expire_at_seconds = int(time.mktime(expire_at.timetuple())) * 1000
+ assert await r.pexpireat("a", expire_at_seconds)
+ assert 0 < await r.pttl("a") <= 61000
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_psetex(self, r: redis.Redis):
+ assert await r.psetex("a", 1000, "value")
+ assert await r.get("a") == b"value"
+ assert 0 < await r.pttl("a") <= 1000
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_psetex_timedelta(self, r: redis.Redis):
+ expire_at = datetime.timedelta(milliseconds=1000)
+ assert await r.psetex("a", expire_at, "value")
+ assert await r.get("a") == b"value"
+ assert 0 < await r.pttl("a") <= 1000
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_pttl(self, r: redis.Redis):
+ assert not await r.pexpire("a", 10000)
+ await r.set("a", "1")
+ assert await r.pexpire("a", 10000)
+ assert 0 < await r.pttl("a") <= 10000
+ assert await r.persist("a")
+ assert await r.pttl("a") == -1
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_pttl_no_key(self, r: redis.Redis):
+ """PTTL on servers 2.8 and after return -2 when the key doesn't exist"""
+ assert await r.pttl("a") == -2
+
+ async def test_randomkey(self, r: redis.Redis):
+ assert await r.randomkey() is None
+ for key in ("a", "b", "c"):
+ await r.set(key, 1)
+ assert await r.randomkey() in (b"a", b"b", b"c")
+
+ async def test_rename(self, r: redis.Redis):
+ await r.set("a", "1")
+ assert await r.rename("a", "b")
+ assert await r.get("a") is None
+ assert await r.get("b") == b"1"
+
+ async def test_renamenx(self, r: redis.Redis):
+ await r.set("a", "1")
+ await r.set("b", "2")
+ assert not await r.renamenx("a", "b")
+ assert await r.get("a") == b"1"
+ assert await r.get("b") == b"2"
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_nx(self, r: redis.Redis):
+ assert await r.set("a", "1", nx=True)
+ assert not await r.set("a", "2", nx=True)
+ assert await r.get("a") == b"1"
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_xx(self, r: redis.Redis):
+ assert not await r.set("a", "1", xx=True)
+ assert await r.get("a") is None
+ await r.set("a", "bar")
+ assert await r.set("a", "2", xx=True)
+ assert await r.get("a") == b"2"
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_px(self, r: redis.Redis):
+ assert await r.set("a", "1", px=10000)
+ assert await r.get("a") == b"1"
+ assert 0 < await r.pttl("a") <= 10000
+ assert 0 < await r.ttl("a") <= 10
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_px_timedelta(self, r: redis.Redis):
+ expire_at = datetime.timedelta(milliseconds=1000)
+ assert await r.set("a", "1", px=expire_at)
+ assert 0 < await r.pttl("a") <= 1000
+ assert 0 < await r.ttl("a") <= 1
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_ex(self, r: redis.Redis):
+ assert await r.set("a", "1", ex=10)
+ assert 0 < await r.ttl("a") <= 10
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_ex_timedelta(self, r: redis.Redis):
+ expire_at = datetime.timedelta(seconds=60)
+ assert await r.set("a", "1", ex=expire_at)
+ assert 0 < await r.ttl("a") <= 60
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_set_multipleoptions(self, r: redis.Redis):
+ await r.set("a", "val")
+ assert await r.set("a", "1", xx=True, px=10000)
+ assert 0 < await r.ttl("a") <= 10
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_set_keepttl(self, r: redis.Redis):
+ await r.set("a", "val")
+ assert await r.set("a", "1", xx=True, px=10000)
+ assert 0 < await r.ttl("a") <= 10
+ await r.set("a", "2", keepttl=True)
+ assert await r.get("a") == b"2"
+ assert 0 < await r.ttl("a") <= 10
+
+ async def test_setex(self, r: redis.Redis):
+ assert await r.setex("a", 60, "1")
+ assert await r.get("a") == b"1"
+ assert 0 < await r.ttl("a") <= 60
+
+ async def test_setnx(self, r: redis.Redis):
+ assert await r.setnx("a", "1")
+ assert await r.get("a") == b"1"
+ assert not await r.setnx("a", "2")
+ assert await r.get("a") == b"1"
+
+ async def test_setrange(self, r: redis.Redis):
+ assert await r.setrange("a", 5, "foo") == 8
+ assert await r.get("a") == b"\0\0\0\0\0foo"
+ await r.set("a", "abcdefghijh")
+ assert await r.setrange("a", 6, "12345") == 11
+ assert await r.get("a") == b"abcdef12345"
+
+ async def test_strlen(self, r: redis.Redis):
+ await r.set("a", "foo")
+ assert await r.strlen("a") == 3
+
+ async def test_substr(self, r: redis.Redis):
+ await r.set("a", "0123456789")
+ assert await r.substr("a", 0) == b"0123456789"
+ assert await r.substr("a", 2) == b"23456789"
+ assert await r.substr("a", 3, 5) == b"345"
+ assert await r.substr("a", 3, -2) == b"345678"
+
+ async def test_ttl(self, r: redis.Redis):
+ await r.set("a", "1")
+ assert await r.expire("a", 10)
+ assert 0 < await r.ttl("a") <= 10
+ assert await r.persist("a")
+ assert await r.ttl("a") == -1
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_ttl_nokey(self, r: redis.Redis):
+ """TTL on servers 2.8 and after return -2 when the key doesn't exist"""
+ assert await r.ttl("a") == -2
+
+ async def test_type(self, r: redis.Redis):
+ assert await r.type("a") == b"none"
+ await r.set("a", "1")
+ assert await r.type("a") == b"string"
+ await r.delete("a")
+ await r.lpush("a", "1")
+ assert await r.type("a") == b"list"
+ await r.delete("a")
+ await r.sadd("a", "1")
+ assert await r.type("a") == b"set"
+ await r.delete("a")
+ await r.zadd("a", {"1": 1})
+ assert await r.type("a") == b"zset"
+
+ # LIST COMMANDS
+ async def test_blpop(self, r: redis.Redis):
+ await r.rpush("a", "1", "2")
+ await r.rpush("b", "3", "4")
+ assert await r.blpop(["b", "a"], timeout=1) == (b"b", b"3")
+ assert await r.blpop(["b", "a"], timeout=1) == (b"b", b"4")
+ assert await r.blpop(["b", "a"], timeout=1) == (b"a", b"1")
+ assert await r.blpop(["b", "a"], timeout=1) == (b"a", b"2")
+ assert await r.blpop(["b", "a"], timeout=1) is None
+ await r.rpush("c", "1")
+ assert await r.blpop("c", timeout=1) == (b"c", b"1")
+
+ async def test_brpop(self, r: redis.Redis):
+ await r.rpush("a", "1", "2")
+ await r.rpush("b", "3", "4")
+ assert await r.brpop(["b", "a"], timeout=1) == (b"b", b"4")
+ assert await r.brpop(["b", "a"], timeout=1) == (b"b", b"3")
+ assert await r.brpop(["b", "a"], timeout=1) == (b"a", b"2")
+ assert await r.brpop(["b", "a"], timeout=1) == (b"a", b"1")
+ assert await r.brpop(["b", "a"], timeout=1) is None
+ await r.rpush("c", "1")
+ assert await r.brpop("c", timeout=1) == (b"c", b"1")
+
+ async def test_brpoplpush(self, r: redis.Redis):
+ await r.rpush("a", "1", "2")
+ await r.rpush("b", "3", "4")
+ assert await r.brpoplpush("a", "b") == b"2"
+ assert await r.brpoplpush("a", "b") == b"1"
+ assert await r.brpoplpush("a", "b", timeout=1) is None
+ assert await r.lrange("a", 0, -1) == []
+ assert await r.lrange("b", 0, -1) == [b"1", b"2", b"3", b"4"]
+
+ async def test_brpoplpush_empty_string(self, r: redis.Redis):
+ await r.rpush("a", "")
+ assert await r.brpoplpush("a", "b") == b""
+
+ async def test_lindex(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.lindex("a", "0") == b"1"
+ assert await r.lindex("a", "1") == b"2"
+ assert await r.lindex("a", "2") == b"3"
+
+ async def test_linsert(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.linsert("a", "after", "2", "2.5") == 4
+ assert await r.lrange("a", 0, -1) == [b"1", b"2", b"2.5", b"3"]
+ assert await r.linsert("a", "before", "2", "1.5") == 5
+ assert await r.lrange("a", 0, -1) == [b"1", b"1.5", b"2", b"2.5", b"3"]
+
+ async def test_llen(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.llen("a") == 3
+
+ async def test_lpop(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.lpop("a") == b"1"
+ assert await r.lpop("a") == b"2"
+ assert await r.lpop("a") == b"3"
+ assert await r.lpop("a") is None
+
+ async def test_lpush(self, r: redis.Redis):
+ assert await r.lpush("a", "1") == 1
+ assert await r.lpush("a", "2") == 2
+ assert await r.lpush("a", "3", "4") == 4
+ assert await r.lrange("a", 0, -1) == [b"4", b"3", b"2", b"1"]
+
+ async def test_lpushx(self, r: redis.Redis):
+ assert await r.lpushx("a", "1") == 0
+ assert await r.lrange("a", 0, -1) == []
+ await r.rpush("a", "1", "2", "3")
+ assert await r.lpushx("a", "4") == 4
+ assert await r.lrange("a", 0, -1) == [b"4", b"1", b"2", b"3"]
+
+ async def test_lrange(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3", "4", "5")
+ assert await r.lrange("a", 0, 2) == [b"1", b"2", b"3"]
+ assert await r.lrange("a", 2, 10) == [b"3", b"4", b"5"]
+ assert await r.lrange("a", 0, -1) == [b"1", b"2", b"3", b"4", b"5"]
+
+ async def test_lrem(self, r: redis.Redis):
+ await r.rpush("a", "Z", "b", "Z", "Z", "c", "Z", "Z")
+ # remove the first 'Z' item
+ assert await r.lrem("a", 1, "Z") == 1
+ assert await r.lrange("a", 0, -1) == [b"b", b"Z", b"Z", b"c", b"Z", b"Z"]
+ # remove the last 2 'Z' items
+ assert await r.lrem("a", -2, "Z") == 2
+ assert await r.lrange("a", 0, -1) == [b"b", b"Z", b"Z", b"c"]
+ # remove all 'Z' items
+ assert await r.lrem("a", 0, "Z") == 2
+ assert await r.lrange("a", 0, -1) == [b"b", b"c"]
+
+ async def test_lset(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.lrange("a", 0, -1) == [b"1", b"2", b"3"]
+ assert await r.lset("a", 1, "4")
+ assert await r.lrange("a", 0, 2) == [b"1", b"4", b"3"]
+
+ async def test_ltrim(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.ltrim("a", 0, 1)
+ assert await r.lrange("a", 0, -1) == [b"1", b"2"]
+
+ async def test_rpop(self, r: redis.Redis):
+ await r.rpush("a", "1", "2", "3")
+ assert await r.rpop("a") == b"3"
+ assert await r.rpop("a") == b"2"
+ assert await r.rpop("a") == b"1"
+ assert await r.rpop("a") is None
+
+ async def test_rpoplpush(self, r: redis.Redis):
+ await r.rpush("a", "a1", "a2", "a3")
+ await r.rpush("b", "b1", "b2", "b3")
+ assert await r.rpoplpush("a", "b") == b"a3"
+ assert await r.lrange("a", 0, -1) == [b"a1", b"a2"]
+ assert await r.lrange("b", 0, -1) == [b"a3", b"b1", b"b2", b"b3"]
+
+ async def test_rpush(self, r: redis.Redis):
+ assert await r.rpush("a", "1") == 1
+ assert await r.rpush("a", "2") == 2
+ assert await r.rpush("a", "3", "4") == 4
+ assert await r.lrange("a", 0, -1) == [b"1", b"2", b"3", b"4"]
+
+ @skip_if_server_version_lt("6.0.6")
+ async def test_lpos(self, r: redis.Redis):
+ assert await r.rpush("a", "a", "b", "c", "1", "2", "3", "c", "c") == 8
+ assert await r.lpos("a", "a") == 0
+ assert await r.lpos("a", "c") == 2
+
+ assert await r.lpos("a", "c", rank=1) == 2
+ assert await r.lpos("a", "c", rank=2) == 6
+ assert await r.lpos("a", "c", rank=4) is None
+ assert await r.lpos("a", "c", rank=-1) == 7
+ assert await r.lpos("a", "c", rank=-2) == 6
+
+ assert await r.lpos("a", "c", count=0) == [2, 6, 7]
+ assert await r.lpos("a", "c", count=1) == [2]
+ assert await r.lpos("a", "c", count=2) == [2, 6]
+ assert await r.lpos("a", "c", count=100) == [2, 6, 7]
+
+ assert await r.lpos("a", "c", count=0, rank=2) == [6, 7]
+ assert await r.lpos("a", "c", count=2, rank=-1) == [7, 6]
+
+ assert await r.lpos("axxx", "c", count=0, rank=2) == []
+ assert await r.lpos("axxx", "c") is None
+
+ assert await r.lpos("a", "x", count=2) == []
+ assert await r.lpos("a", "x") is None
+
+ assert await r.lpos("a", "a", count=0, maxlen=1) == [0]
+ assert await r.lpos("a", "c", count=0, maxlen=1) == []
+ assert await r.lpos("a", "c", count=0, maxlen=3) == [2]
+ assert await r.lpos("a", "c", count=0, maxlen=3, rank=-1) == [7, 6]
+ assert await r.lpos("a", "c", count=0, maxlen=7, rank=2) == [6]
+
+ async def test_rpushx(self, r: redis.Redis):
+ assert await r.rpushx("a", "b") == 0
+ assert await r.lrange("a", 0, -1) == []
+ await r.rpush("a", "1", "2", "3")
+ assert await r.rpushx("a", "4") == 4
+ assert await r.lrange("a", 0, -1) == [b"1", b"2", b"3", b"4"]
+
+ # SCAN COMMANDS
+ @skip_if_server_version_lt("2.8.0")
+ async def test_scan(self, r: redis.Redis):
+ await r.set("a", 1)
+ await r.set("b", 2)
+ await r.set("c", 3)
+ cursor, keys = await r.scan()
+ assert cursor == 0
+ assert set(keys) == {b"a", b"b", b"c"}
+ _, keys = await r.scan(match="a")
+ assert set(keys) == {b"a"}
+
+ @skip_if_server_version_lt(REDIS_6_VERSION)
+ async def test_scan_type(self, r: redis.Redis):
+ await r.sadd("a-set", 1)
+ await r.hset("a-hash", "foo", 2)
+ await r.lpush("a-list", "aux", 3)
+ _, keys = await r.scan(match="a*", _type="SET")
+ assert set(keys) == {b"a-set"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_scan_iter(self, r: redis.Redis):
+ await r.set("a", 1)
+ await r.set("b", 2)
+ await r.set("c", 3)
+ keys = [k async for k in r.scan_iter()]
+ assert set(keys) == {b"a", b"b", b"c"}
+ keys = [k async for k in r.scan_iter(match="a")]
+ assert set(keys) == {b"a"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_sscan(self, r: redis.Redis):
+ await r.sadd("a", 1, 2, 3)
+ cursor, members = await r.sscan("a")
+ assert cursor == 0
+ assert set(members) == {b"1", b"2", b"3"}
+ _, members = await r.sscan("a", match=b"1")
+ assert set(members) == {b"1"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_sscan_iter(self, r: redis.Redis):
+ await r.sadd("a", 1, 2, 3)
+ members = [k async for k in r.sscan_iter("a")]
+ assert set(members) == {b"1", b"2", b"3"}
+ members = [k async for k in r.sscan_iter("a", match=b"1")]
+ assert set(members) == {b"1"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_hscan(self, r: redis.Redis):
+ await r.hset("a", mapping={"a": 1, "b": 2, "c": 3})
+ cursor, dic = await r.hscan("a")
+ assert cursor == 0
+ assert dic == {b"a": b"1", b"b": b"2", b"c": b"3"}
+ _, dic = await r.hscan("a", match="a")
+ assert dic == {b"a": b"1"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_hscan_iter(self, r: redis.Redis):
+ await r.hset("a", mapping={"a": 1, "b": 2, "c": 3})
+ dic = {k: v async for k, v in r.hscan_iter("a")}
+ assert dic == {b"a": b"1", b"b": b"2", b"c": b"3"}
+ dic = {k: v async for k, v in r.hscan_iter("a", match="a")}
+ assert dic == {b"a": b"1"}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_zscan(self, r: redis.Redis):
+ await r.zadd("a", {"a": 1, "b": 2, "c": 3})
+ cursor, pairs = await r.zscan("a")
+ assert cursor == 0
+ assert set(pairs) == {(b"a", 1), (b"b", 2), (b"c", 3)}
+ _, pairs = await r.zscan("a", match="a")
+ assert set(pairs) == {(b"a", 1)}
+
+ @skip_if_server_version_lt("2.8.0")
+ async def test_zscan_iter(self, r: redis.Redis):
+ await r.zadd("a", {"a": 1, "b": 2, "c": 3})
+ pairs = [k async for k in r.zscan_iter("a")]
+ assert set(pairs) == {(b"a", 1), (b"b", 2), (b"c", 3)}
+ pairs = [k async for k in r.zscan_iter("a", match="a")]
+ assert set(pairs) == {(b"a", 1)}
+
+ # SET COMMANDS
+ async def test_sadd(self, r: redis.Redis):
+ members = {b"1", b"2", b"3"}
+ await r.sadd("a", *members)
+ assert await r.smembers("a") == members
+
+ async def test_scard(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.scard("a") == 3
+
+ async def test_sdiff(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.sdiff("a", "b") == {b"1", b"2", b"3"}
+ await r.sadd("b", "2", "3")
+ assert await r.sdiff("a", "b") == {b"1"}
+
+ async def test_sdiffstore(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.sdiffstore("c", "a", "b") == 3
+ assert await r.smembers("c") == {b"1", b"2", b"3"}
+ await r.sadd("b", "2", "3")
+ assert await r.sdiffstore("c", "a", "b") == 1
+ assert await r.smembers("c") == {b"1"}
+
+ async def test_sinter(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.sinter("a", "b") == set()
+ await r.sadd("b", "2", "3")
+ assert await r.sinter("a", "b") == {b"2", b"3"}
+
+ async def test_sinterstore(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.sinterstore("c", "a", "b") == 0
+ assert await r.smembers("c") == set()
+ await r.sadd("b", "2", "3")
+ assert await r.sinterstore("c", "a", "b") == 2
+ assert await r.smembers("c") == {b"2", b"3"}
+
+ async def test_sismember(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.sismember("a", "1")
+ assert await r.sismember("a", "2")
+ assert await r.sismember("a", "3")
+ assert not await r.sismember("a", "4")
+
+ async def test_smembers(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3")
+ assert await r.smembers("a") == {b"1", b"2", b"3"}
+
+ async def test_smove(self, r: redis.Redis):
+ await r.sadd("a", "a1", "a2")
+ await r.sadd("b", "b1", "b2")
+ assert await r.smove("a", "b", "a1")
+ assert await r.smembers("a") == {b"a2"}
+ assert await r.smembers("b") == {b"b1", b"b2", b"a1"}
+
+ async def test_spop(self, r: redis.Redis):
+ s = [b"1", b"2", b"3"]
+ await r.sadd("a", *s)
+ value = await r.spop("a")
+ assert value in s
+ assert await r.smembers("a") == set(s) - {value}
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_spop_multi_value(self, r: redis.Redis):
+ s = [b"1", b"2", b"3"]
+ await r.sadd("a", *s)
+ values = await r.spop("a", 2)
+ assert len(values) == 2
+
+ for value in values:
+ assert value in s
+
+ assert await r.spop("a", 1) == list(set(s) - set(values))
+
+ async def test_srandmember(self, r: redis.Redis):
+ s = [b"1", b"2", b"3"]
+ await r.sadd("a", *s)
+ assert await r.srandmember("a") in s
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_srandmember_multi_value(self, r: redis.Redis):
+ s = [b"1", b"2", b"3"]
+ await r.sadd("a", *s)
+ randoms = await r.srandmember("a", number=2)
+ assert len(randoms) == 2
+ assert set(randoms).intersection(s) == set(randoms)
+
+ async def test_srem(self, r: redis.Redis):
+ await r.sadd("a", "1", "2", "3", "4")
+ assert await r.srem("a", "5") == 0
+ assert await r.srem("a", "2", "4") == 2
+ assert await r.smembers("a") == {b"1", b"3"}
+
+ async def test_sunion(self, r: redis.Redis):
+ await r.sadd("a", "1", "2")
+ await r.sadd("b", "2", "3")
+ assert await r.sunion("a", "b") == {b"1", b"2", b"3"}
+
+ async def test_sunionstore(self, r: redis.Redis):
+ await r.sadd("a", "1", "2")
+ await r.sadd("b", "2", "3")
+ assert await r.sunionstore("c", "a", "b") == 3
+ assert await r.smembers("c") == {b"1", b"2", b"3"}
+
+ # SORTED SET COMMANDS
+ async def test_zadd(self, r: redis.Redis):
+ mapping = {"a1": 1.0, "a2": 2.0, "a3": 3.0}
+ await r.zadd("a", mapping)
+ assert await r.zrange("a", 0, -1, withscores=True) == [
+ (b"a1", 1.0),
+ (b"a2", 2.0),
+ (b"a3", 3.0),
+ ]
+
+ # error cases
+ with pytest.raises(exceptions.DataError):
+ await r.zadd("a", {})
+
+ # cannot use both nx and xx options
+ with pytest.raises(exceptions.DataError):
+ await r.zadd("a", mapping, nx=True, xx=True)
+
+ # cannot use the incr options with more than one value
+ with pytest.raises(exceptions.DataError):
+ await r.zadd("a", mapping, incr=True)
+
+ async def test_zadd_nx(self, r: redis.Redis):
+ assert await r.zadd("a", {"a1": 1}) == 1
+ assert await r.zadd("a", {"a1": 99, "a2": 2}, nx=True) == 1
+ assert await r.zrange("a", 0, -1, withscores=True) == [
+ (b"a1", 1.0),
+ (b"a2", 2.0),
+ ]
+
+ async def test_zadd_xx(self, r: redis.Redis):
+ assert await r.zadd("a", {"a1": 1}) == 1
+ assert await r.zadd("a", {"a1": 99, "a2": 2}, xx=True) == 0
+ assert await r.zrange("a", 0, -1, withscores=True) == [(b"a1", 99.0)]
+
+ async def test_zadd_ch(self, r: redis.Redis):
+ assert await r.zadd("a", {"a1": 1}) == 1
+ assert await r.zadd("a", {"a1": 99, "a2": 2}, ch=True) == 2
+ assert await r.zrange("a", 0, -1, withscores=True) == [
+ (b"a2", 2.0),
+ (b"a1", 99.0),
+ ]
+
+ async def test_zadd_incr(self, r: redis.Redis):
+ assert await r.zadd("a", {"a1": 1}) == 1
+ assert await r.zadd("a", {"a1": 4.5}, incr=True) == 5.5
+
+ async def test_zadd_incr_with_xx(self, r: redis.Redis):
+ # this asks zadd to incr 'a1' only if it exists, but it clearly
+ # doesn't. Redis returns a null value in this case and so should
+ # redis-py
+ assert await r.zadd("a", {"a1": 1}, xx=True, incr=True) is None
+
+ async def test_zcard(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zcard("a") == 3
+
+ async def test_zcount(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zcount("a", "-inf", "+inf") == 3
+ assert await r.zcount("a", 1, 2) == 2
+ assert await r.zcount("a", "(" + str(1), 2) == 1
+ assert await r.zcount("a", 1, "(" + str(2)) == 1
+ assert await r.zcount("a", 10, 20) == 0
+
+ async def test_zincrby(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zincrby("a", 1, "a2") == 3.0
+ assert await r.zincrby("a", 5, "a3") == 8.0
+ assert await r.zscore("a", "a2") == 3.0
+ assert await r.zscore("a", "a3") == 8.0
+
+ @skip_if_server_version_lt("2.8.9")
+ async def test_zlexcount(self, r: redis.Redis):
+ await r.zadd("a", {"a": 0, "b": 0, "c": 0, "d": 0, "e": 0, "f": 0, "g": 0})
+ assert await r.zlexcount("a", "-", "+") == 7
+ assert await r.zlexcount("a", "[b", "[f") == 5
+
+ async def test_zinterstore_sum(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zinterstore("d", ["a", "b", "c"]) == 2
+ assert await r.zrange("d", 0, -1, withscores=True) == [(b"a3", 8), (b"a1", 9)]
+
+ async def test_zinterstore_max(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zinterstore("d", ["a", "b", "c"], aggregate="MAX") == 2
+ assert await r.zrange("d", 0, -1, withscores=True) == [(b"a3", 5), (b"a1", 6)]
+
+ async def test_zinterstore_min(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ await r.zadd("b", {"a1": 2, "a2": 3, "a3": 5})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zinterstore("d", ["a", "b", "c"], aggregate="MIN") == 2
+ assert await r.zrange("d", 0, -1, withscores=True) == [(b"a1", 1), (b"a3", 3)]
+
+ async def test_zinterstore_with_weight(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zinterstore("d", {"a": 1, "b": 2, "c": 3}) == 2
+ assert await r.zrange("d", 0, -1, withscores=True) == [(b"a3", 20), (b"a1", 23)]
+
+ @skip_if_server_version_lt("4.9.0")
+ async def test_zpopmax(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zpopmax("a") == [(b"a3", 3)]
+
+ # with count
+ assert await r.zpopmax("a", count=2) == [(b"a2", 2), (b"a1", 1)]
+
+ @skip_if_server_version_lt("4.9.0")
+ async def test_zpopmin(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zpopmin("a") == [(b"a1", 1)]
+
+ # with count
+ assert await r.zpopmin("a", count=2) == [(b"a2", 2), (b"a3", 3)]
+
+ @skip_if_server_version_lt("4.9.0")
+ async def test_bzpopmax(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2})
+ await r.zadd("b", {"b1": 10, "b2": 20})
+ assert await r.bzpopmax(["b", "a"], timeout=1) == (b"b", b"b2", 20)
+ assert await r.bzpopmax(["b", "a"], timeout=1) == (b"b", b"b1", 10)
+ assert await r.bzpopmax(["b", "a"], timeout=1) == (b"a", b"a2", 2)
+ assert await r.bzpopmax(["b", "a"], timeout=1) == (b"a", b"a1", 1)
+ assert await r.bzpopmax(["b", "a"], timeout=1) is None
+ await r.zadd("c", {"c1": 100})
+ assert await r.bzpopmax("c", timeout=1) == (b"c", b"c1", 100)
+
+ @skip_if_server_version_lt("4.9.0")
+ async def test_bzpopmin(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2})
+ await r.zadd("b", {"b1": 10, "b2": 20})
+ assert await r.bzpopmin(["b", "a"], timeout=1) == (b"b", b"b1", 10)
+ assert await r.bzpopmin(["b", "a"], timeout=1) == (b"b", b"b2", 20)
+ assert await r.bzpopmin(["b", "a"], timeout=1) == (b"a", b"a1", 1)
+ assert await r.bzpopmin(["b", "a"], timeout=1) == (b"a", b"a2", 2)
+ assert await r.bzpopmin(["b", "a"], timeout=1) is None
+ await r.zadd("c", {"c1": 100})
+ assert await r.bzpopmin("c", timeout=1) == (b"c", b"c1", 100)
+
+ async def test_zrange(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zrange("a", 0, 1) == [b"a1", b"a2"]
+ assert await r.zrange("a", 1, 2) == [b"a2", b"a3"]
+
+ # withscores
+ assert await r.zrange("a", 0, 1, withscores=True) == [
+ (b"a1", 1.0),
+ (b"a2", 2.0),
+ ]
+ assert await r.zrange("a", 1, 2, withscores=True) == [
+ (b"a2", 2.0),
+ (b"a3", 3.0),
+ ]
+
+ # custom score function
+ assert await r.zrange("a", 0, 1, withscores=True, score_cast_func=int) == [
+ (b"a1", 1),
+ (b"a2", 2),
+ ]
+
+ @skip_if_server_version_lt("2.8.9")
+ async def test_zrangebylex(self, r: redis.Redis):
+ await r.zadd("a", {"a": 0, "b": 0, "c": 0, "d": 0, "e": 0, "f": 0, "g": 0})
+ assert await r.zrangebylex("a", "-", "[c") == [b"a", b"b", b"c"]
+ assert await r.zrangebylex("a", "-", "(c") == [b"a", b"b"]
+ assert await r.zrangebylex("a", "[aaa", "(g") == [b"b", b"c", b"d", b"e", b"f"]
+ assert await r.zrangebylex("a", "[f", "+") == [b"f", b"g"]
+ assert await r.zrangebylex("a", "-", "+", start=3, num=2) == [b"d", b"e"]
+
+ @skip_if_server_version_lt("2.9.9")
+ async def test_zrevrangebylex(self, r: redis.Redis):
+ await r.zadd("a", {"a": 0, "b": 0, "c": 0, "d": 0, "e": 0, "f": 0, "g": 0})
+ assert await r.zrevrangebylex("a", "[c", "-") == [b"c", b"b", b"a"]
+ assert await r.zrevrangebylex("a", "(c", "-") == [b"b", b"a"]
+ assert await r.zrevrangebylex("a", "(g", "[aaa") == [
+ b"f",
+ b"e",
+ b"d",
+ b"c",
+ b"b",
+ ]
+ assert await r.zrevrangebylex("a", "+", "[f") == [b"g", b"f"]
+ assert await r.zrevrangebylex("a", "+", "-", start=3, num=2) == [b"d", b"c"]
+
+ async def test_zrangebyscore(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zrangebyscore("a", 2, 4) == [b"a2", b"a3", b"a4"]
+
+ # slicing with start/num
+ assert await r.zrangebyscore("a", 2, 4, start=1, num=2) == [b"a3", b"a4"]
+
+ # withscores
+ assert await r.zrangebyscore("a", 2, 4, withscores=True) == [
+ (b"a2", 2.0),
+ (b"a3", 3.0),
+ (b"a4", 4.0),
+ ]
+
+ # custom score function
+ assert await r.zrangebyscore(
+ "a", 2, 4, withscores=True, score_cast_func=int
+ ) == [(b"a2", 2), (b"a3", 3), (b"a4", 4)]
+
+ async def test_zrank(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zrank("a", "a1") == 0
+ assert await r.zrank("a", "a2") == 1
+ assert await r.zrank("a", "a6") is None
+
+ async def test_zrem(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zrem("a", "a2") == 1
+ assert await r.zrange("a", 0, -1) == [b"a1", b"a3"]
+ assert await r.zrem("a", "b") == 0
+ assert await r.zrange("a", 0, -1) == [b"a1", b"a3"]
+
+ async def test_zrem_multiple_keys(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zrem("a", "a1", "a2") == 2
+ assert await r.zrange("a", 0, 5) == [b"a3"]
+
+ @skip_if_server_version_lt("2.8.9")
+ async def test_zremrangebylex(self, r: redis.Redis):
+ await r.zadd("a", {"a": 0, "b": 0, "c": 0, "d": 0, "e": 0, "f": 0, "g": 0})
+ assert await r.zremrangebylex("a", "-", "[c") == 3
+ assert await r.zrange("a", 0, -1) == [b"d", b"e", b"f", b"g"]
+ assert await r.zremrangebylex("a", "[f", "+") == 2
+ assert await r.zrange("a", 0, -1) == [b"d", b"e"]
+ assert await r.zremrangebylex("a", "[h", "+") == 0
+ assert await r.zrange("a", 0, -1) == [b"d", b"e"]
+
+ async def test_zremrangebyrank(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zremrangebyrank("a", 1, 3) == 3
+ assert await r.zrange("a", 0, 5) == [b"a1", b"a5"]
+
+ async def test_zremrangebyscore(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zremrangebyscore("a", 2, 4) == 3
+ assert await r.zrange("a", 0, -1) == [b"a1", b"a5"]
+ assert await r.zremrangebyscore("a", 2, 4) == 0
+ assert await r.zrange("a", 0, -1) == [b"a1", b"a5"]
+
+ async def test_zrevrange(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zrevrange("a", 0, 1) == [b"a3", b"a2"]
+ assert await r.zrevrange("a", 1, 2) == [b"a2", b"a1"]
+
+ # withscores
+ assert await r.zrevrange("a", 0, 1, withscores=True) == [
+ (b"a3", 3.0),
+ (b"a2", 2.0),
+ ]
+ assert await r.zrevrange("a", 1, 2, withscores=True) == [
+ (b"a2", 2.0),
+ (b"a1", 1.0),
+ ]
+
+ # custom score function
+ assert await r.zrevrange("a", 0, 1, withscores=True, score_cast_func=int) == [
+ (b"a3", 3.0),
+ (b"a2", 2.0),
+ ]
+
+ async def test_zrevrangebyscore(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zrevrangebyscore("a", 4, 2) == [b"a4", b"a3", b"a2"]
+
+ # slicing with start/num
+ assert await r.zrevrangebyscore("a", 4, 2, start=1, num=2) == [b"a3", b"a2"]
+
+ # withscores
+ assert await r.zrevrangebyscore("a", 4, 2, withscores=True) == [
+ (b"a4", 4.0),
+ (b"a3", 3.0),
+ (b"a2", 2.0),
+ ]
+
+ # custom score function
+ assert await r.zrevrangebyscore(
+ "a", 4, 2, withscores=True, score_cast_func=int
+ ) == [(b"a4", 4), (b"a3", 3), (b"a2", 2)]
+
+ async def test_zrevrank(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3, "a4": 4, "a5": 5})
+ assert await r.zrevrank("a", "a1") == 4
+ assert await r.zrevrank("a", "a2") == 3
+ assert await r.zrevrank("a", "a6") is None
+
+ async def test_zscore(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ assert await r.zscore("a", "a1") == 1.0
+ assert await r.zscore("a", "a2") == 2.0
+ assert await r.zscore("a", "a4") is None
+
+ async def test_zunionstore_sum(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zunionstore("d", ["a", "b", "c"]) == 4
+ assert await r.zrange("d", 0, -1, withscores=True) == [
+ (b"a2", 3),
+ (b"a4", 4),
+ (b"a3", 8),
+ (b"a1", 9),
+ ]
+
+ async def test_zunionstore_max(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zunionstore("d", ["a", "b", "c"], aggregate="MAX") == 4
+ assert await r.zrange("d", 0, -1, withscores=True) == [
+ (b"a2", 2),
+ (b"a4", 4),
+ (b"a3", 5),
+ (b"a1", 6),
+ ]
+
+ async def test_zunionstore_min(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 2, "a3": 3})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 4})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zunionstore("d", ["a", "b", "c"], aggregate="MIN") == 4
+ assert await r.zrange("d", 0, -1, withscores=True) == [
+ (b"a1", 1),
+ (b"a2", 2),
+ (b"a3", 3),
+ (b"a4", 4),
+ ]
+
+ async def test_zunionstore_with_weight(self, r: redis.Redis):
+ await r.zadd("a", {"a1": 1, "a2": 1, "a3": 1})
+ await r.zadd("b", {"a1": 2, "a2": 2, "a3": 2})
+ await r.zadd("c", {"a1": 6, "a3": 5, "a4": 4})
+ assert await r.zunionstore("d", {"a": 1, "b": 2, "c": 3}) == 4
+ assert await r.zrange("d", 0, -1, withscores=True) == [
+ (b"a2", 5),
+ (b"a4", 12),
+ (b"a3", 20),
+ (b"a1", 23),
+ ]
+
+ # HYPERLOGLOG TESTS
+ @skip_if_server_version_lt("2.8.9")
+ async def test_pfadd(self, r: redis.Redis):
+ members = {b"1", b"2", b"3"}
+ assert await r.pfadd("a", *members) == 1
+ assert await r.pfadd("a", *members) == 0
+ assert await r.pfcount("a") == len(members)
+
+ @skip_if_server_version_lt("2.8.9")
+ async def test_pfcount(self, r: redis.Redis):
+ members = {b"1", b"2", b"3"}
+ await r.pfadd("a", *members)
+ assert await r.pfcount("a") == len(members)
+ members_b = {b"2", b"3", b"4"}
+ await r.pfadd("b", *members_b)
+ assert await r.pfcount("b") == len(members_b)
+ assert await r.pfcount("a", "b") == len(members_b.union(members))
+
+ @skip_if_server_version_lt("2.8.9")
+ async def test_pfmerge(self, r: redis.Redis):
+ mema = {b"1", b"2", b"3"}
+ memb = {b"2", b"3", b"4"}
+ memc = {b"5", b"6", b"7"}
+ await r.pfadd("a", *mema)
+ await r.pfadd("b", *memb)
+ await r.pfadd("c", *memc)
+ await r.pfmerge("d", "c", "a")
+ assert await r.pfcount("d") == 6
+ await r.pfmerge("d", "b")
+ assert await r.pfcount("d") == 7
+
+ # HASH COMMANDS
+ async def test_hget_and_hset(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": 1, "2": 2, "3": 3})
+ assert await r.hget("a", "1") == b"1"
+ assert await r.hget("a", "2") == b"2"
+ assert await r.hget("a", "3") == b"3"
+
+ # field was updated, redis returns 0
+ assert await r.hset("a", "2", 5) == 0
+ assert await r.hget("a", "2") == b"5"
+
+ # field is new, redis returns 1
+ assert await r.hset("a", "4", 4) == 1
+ assert await r.hget("a", "4") == b"4"
+
+ # key inside of hash that doesn't exist returns null value
+ assert await r.hget("a", "b") is None
+
+ # keys with bool(key) == False
+ assert await r.hset("a", 0, 10) == 1
+ assert await r.hset("a", "", 10) == 1
+
+ async def test_hset_with_multi_key_values(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": 1, "2": 2, "3": 3})
+ assert await r.hget("a", "1") == b"1"
+ assert await r.hget("a", "2") == b"2"
+ assert await r.hget("a", "3") == b"3"
+
+ await r.hset("b", "foo", "bar", mapping={"1": 1, "2": 2})
+ assert await r.hget("b", "1") == b"1"
+ assert await r.hget("b", "2") == b"2"
+ assert await r.hget("b", "foo") == b"bar"
+
+ async def test_hset_without_data(self, r: redis.Redis):
+ with pytest.raises(exceptions.DataError):
+ await r.hset("x")
+
+ async def test_hdel(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": 1, "2": 2, "3": 3})
+ assert await r.hdel("a", "2") == 1
+ assert await r.hget("a", "2") is None
+ assert await r.hdel("a", "1", "3") == 2
+ assert await r.hlen("a") == 0
+
+ async def test_hexists(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": 1, "2": 2, "3": 3})
+ assert await r.hexists("a", "1")
+ assert not await r.hexists("a", "4")
+
+ async def test_hgetall(self, r: redis.Redis):
+ h = {b"a1": b"1", b"a2": b"2", b"a3": b"3"}
+ await r.hset("a", mapping=h)
+ assert await r.hgetall("a") == h
+
+ async def test_hincrby(self, r: redis.Redis):
+ assert await r.hincrby("a", "1") == 1
+ assert await r.hincrby("a", "1", amount=2) == 3
+ assert await r.hincrby("a", "1", amount=-2) == 1
+
+ @skip_if_server_version_lt("2.6.0")
+ async def test_hincrbyfloat(self, r: redis.Redis):
+ assert await r.hincrbyfloat("a", "1") == 1.0
+ assert await r.hincrbyfloat("a", "1") == 2.0
+ assert await r.hincrbyfloat("a", "1", 1.2) == 3.2
+
+ async def test_hkeys(self, r: redis.Redis):
+ h = {b"a1": b"1", b"a2": b"2", b"a3": b"3"}
+ await r.hset("a", mapping=h)
+ local_keys = list(h.keys())
+ remote_keys = await r.hkeys("a")
+ assert sorted(local_keys) == sorted(remote_keys)
+
+ async def test_hlen(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": 1, "2": 2, "3": 3})
+ assert await r.hlen("a") == 3
+
+ async def test_hmget(self, r: redis.Redis):
+ assert await r.hset("a", mapping={"a": 1, "b": 2, "c": 3})
+ assert await r.hmget("a", "a", "b", "c") == [b"1", b"2", b"3"]
+
+ async def test_hmset(self, r: redis.Redis):
+ warning_message = (
+ r"^Redis\.hmset\(\) is deprecated\. " r"Use Redis\.hset\(\) instead\.$"
+ )
+ h = {b"a": b"1", b"b": b"2", b"c": b"3"}
+ with pytest.warns(DeprecationWarning, match=warning_message):
+ assert await r.hmset("a", h)
+ assert await r.hgetall("a") == h
+
+ async def test_hsetnx(self, r: redis.Redis):
+ # Initially set the hash field
+ assert await r.hsetnx("a", "1", 1)
+ assert await r.hget("a", "1") == b"1"
+ assert not await r.hsetnx("a", "1", 2)
+ assert await r.hget("a", "1") == b"1"
+
+ async def test_hvals(self, r: redis.Redis):
+ h = {b"a1": b"1", b"a2": b"2", b"a3": b"3"}
+ await r.hset("a", mapping=h)
+ local_vals = list(h.values())
+ remote_vals = await r.hvals("a")
+ assert sorted(local_vals) == sorted(remote_vals)
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_hstrlen(self, r: redis.Redis):
+ await r.hset("a", mapping={"1": "22", "2": "333"})
+ assert await r.hstrlen("a", "1") == 2
+ assert await r.hstrlen("a", "2") == 3
+
+ # SORT
+ async def test_sort_basic(self, r: redis.Redis):
+ await r.rpush("a", "3", "2", "1", "4")
+ assert await r.sort("a") == [b"1", b"2", b"3", b"4"]
+
+ async def test_sort_limited(self, r: redis.Redis):
+ await r.rpush("a", "3", "2", "1", "4")
+ assert await r.sort("a", start=1, num=2) == [b"2", b"3"]
+
+ async def test_sort_by(self, r: redis.Redis):
+ await r.set("score:1", 8)
+ await r.set("score:2", 3)
+ await r.set("score:3", 5)
+ await r.rpush("a", "3", "2", "1")
+ assert await r.sort("a", by="score:*") == [b"2", b"3", b"1"]
+
+ async def test_sort_get(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", get="user:*") == [b"u1", b"u2", b"u3"]
+
+ async def test_sort_get_multi(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", get=("user:*", "#")) == [
+ b"u1",
+ b"1",
+ b"u2",
+ b"2",
+ b"u3",
+ b"3",
+ ]
+
+ async def test_sort_get_groups_two(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", get=("user:*", "#"), groups=True) == [
+ (b"u1", b"1"),
+ (b"u2", b"2"),
+ (b"u3", b"3"),
+ ]
+
+ async def test_sort_groups_string_get(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ with pytest.raises(exceptions.DataError):
+ await r.sort("a", get="user:*", groups=True)
+
+ async def test_sort_groups_just_one_get(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ with pytest.raises(exceptions.DataError):
+ await r.sort("a", get=["user:*"], groups=True)
+
+ async def test_sort_groups_no_get(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.rpush("a", "2", "3", "1")
+ with pytest.raises(exceptions.DataError):
+ await r.sort("a", groups=True)
+
+ async def test_sort_groups_three_gets(self, r: redis.Redis):
+ await r.set("user:1", "u1")
+ await r.set("user:2", "u2")
+ await r.set("user:3", "u3")
+ await r.set("door:1", "d1")
+ await r.set("door:2", "d2")
+ await r.set("door:3", "d3")
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", get=("user:*", "door:*", "#"), groups=True) == [
+ (b"u1", b"d1", b"1"),
+ (b"u2", b"d2", b"2"),
+ (b"u3", b"d3", b"3"),
+ ]
+
+ async def test_sort_desc(self, r: redis.Redis):
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", desc=True) == [b"3", b"2", b"1"]
+
+ async def test_sort_alpha(self, r: redis.Redis):
+ await r.rpush("a", "e", "c", "b", "d", "a")
+ assert await r.sort("a", alpha=True) == [b"a", b"b", b"c", b"d", b"e"]
+
+ async def test_sort_store(self, r: redis.Redis):
+ await r.rpush("a", "2", "3", "1")
+ assert await r.sort("a", store="sorted_values") == 3
+ assert await r.lrange("sorted_values", 0, -1) == [b"1", b"2", b"3"]
+
+ async def test_sort_all_options(self, r: redis.Redis):
+ await r.set("user:1:username", "zeus")
+ await r.set("user:2:username", "titan")
+ await r.set("user:3:username", "hermes")
+ await r.set("user:4:username", "hercules")
+ await r.set("user:5:username", "apollo")
+ await r.set("user:6:username", "athena")
+ await r.set("user:7:username", "hades")
+ await r.set("user:8:username", "dionysus")
+
+ await r.set("user:1:favorite_drink", "yuengling")
+ await r.set("user:2:favorite_drink", "rum")
+ await r.set("user:3:favorite_drink", "vodka")
+ await r.set("user:4:favorite_drink", "milk")
+ await r.set("user:5:favorite_drink", "pinot noir")
+ await r.set("user:6:favorite_drink", "water")
+ await r.set("user:7:favorite_drink", "gin")
+ await r.set("user:8:favorite_drink", "apple juice")
+
+ await r.rpush("gods", "5", "8", "3", "1", "2", "7", "6", "4")
+ num = await r.sort(
+ "gods",
+ start=2,
+ num=4,
+ by="user:*:username",
+ get="user:*:favorite_drink",
+ desc=True,
+ alpha=True,
+ store="sorted",
+ )
+ assert num == 4
+ assert await r.lrange("sorted", 0, 10) == [
+ b"vodka",
+ b"milk",
+ b"gin",
+ b"apple juice",
+ ]
+
+ async def test_sort_issue_924(self, r: redis.Redis):
+ # Tests for issue https://github.com/andymccurdy/redis-py/issues/924
+ await r.execute_command("SADD", "issue#924", 1)
+ await r.execute_command("SORT", "issue#924")
+
+ async def test_cluster_addslots(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("ADDSLOTS", 1) is True
+
+ async def test_cluster_count_failure_reports(self, mock_cluster_resp_int):
+ assert isinstance(
+ await mock_cluster_resp_int.cluster("COUNT-FAILURE-REPORTS", "node"), int
+ )
+
+ async def test_cluster_countkeysinslot(self, mock_cluster_resp_int):
+ assert isinstance(
+ await mock_cluster_resp_int.cluster("COUNTKEYSINSLOT", 2), int
+ )
+
+ async def test_cluster_delslots(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("DELSLOTS", 1) is True
+
+ async def test_cluster_failover(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("FAILOVER", 1) is True
+
+ async def test_cluster_forget(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("FORGET", 1) is True
+
+ async def test_cluster_info(self, mock_cluster_resp_info):
+ assert isinstance(await mock_cluster_resp_info.cluster("info"), dict)
+
+ async def test_cluster_keyslot(self, mock_cluster_resp_int):
+ assert isinstance(await mock_cluster_resp_int.cluster("keyslot", "asdf"), int)
+
+ async def test_cluster_meet(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("meet", "ip", "port", 1) is True
+
+ async def test_cluster_nodes(self, mock_cluster_resp_nodes):
+ assert isinstance(await mock_cluster_resp_nodes.cluster("nodes"), dict)
+
+ async def test_cluster_replicate(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("replicate", "nodeid") is True
+
+ async def test_cluster_reset(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("reset", "hard") is True
+
+ async def test_cluster_saveconfig(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.cluster("saveconfig") is True
+
+ async def test_cluster_setslot(self, mock_cluster_resp_ok):
+ assert (
+ await mock_cluster_resp_ok.cluster("setslot", 1, "IMPORTING", "nodeid")
+ is True
+ )
+
+ async def test_cluster_slaves(self, mock_cluster_resp_slaves):
+ assert isinstance(
+ await mock_cluster_resp_slaves.cluster("slaves", "nodeid"), dict
+ )
+
+ @skip_if_server_version_lt("3.0.0")
+ async def test_readwrite(self, r: redis.Redis):
+ assert await r.readwrite()
+
+ @skip_if_server_version_lt("3.0.0")
+ async def test_readonly_invalid_cluster_state(self, r: redis.Redis):
+ with pytest.raises(exceptions.RedisError):
+ await r.readonly()
+
+ @skip_if_server_version_lt("3.0.0")
+ async def test_readonly(self, mock_cluster_resp_ok):
+ assert await mock_cluster_resp_ok.readonly() is True
+
+ # GEO COMMANDS
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geoadd(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ assert await r.geoadd("barcelona", values) == 2
+ assert await r.zcard("barcelona") == 2
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geoadd_invalid_params(self, r: redis.Redis):
+ with pytest.raises(exceptions.RedisError):
+ await r.geoadd("barcelona", (1, 2))
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geodist(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ assert await r.geoadd("barcelona", values) == 2
+ assert await r.geodist("barcelona", "place1", "place2") == 3067.4157
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geodist_units(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.geodist("barcelona", "place1", "place2", "km") == 3.0674
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geodist_missing_one_member(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1")
+ await r.geoadd("barcelona", values)
+ assert await r.geodist("barcelona", "place1", "missing_member", "km") is None
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geodist_invalid_units(self, r: redis.Redis):
+ with pytest.raises(exceptions.RedisError):
+ assert await r.geodist("x", "y", "z", "inches")
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geohash(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.geohash("barcelona", "place1", "place2", "place3") == [
+ "sp3e9yg3kd0",
+ "sp3e9cbc3t0",
+ None,
+ ]
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_geopos(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ # redis uses 52 bits precision, hereby small errors may be introduced.
+ assert await r.geopos("barcelona", "place1", "place2") == [
+ (2.19093829393386841, 41.43379028184083523),
+ (2.18737632036209106, 41.40634178640635099),
+ ]
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_geopos_no_value(self, r: redis.Redis):
+ assert await r.geopos("barcelona", "place1", "place2") == [None, None]
+
+ @skip_if_server_version_lt("3.2.0")
+ @skip_if_server_version_gte("4.0.0")
+ async def test_old_geopos_no_value(self, r: redis.Redis):
+ assert await r.geopos("barcelona", "place1", "place2") == []
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ b"\x80place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadius("barcelona", 2.191, 41.433, 1000) == [b"place1"]
+ assert await r.georadius("barcelona", 2.187, 41.406, 1000) == [b"\x80place2"]
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_no_values(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadius("barcelona", 1, 2, 1000) == []
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_units(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadius("barcelona", 2.191, 41.433, 1, unit="km") == [
+ b"place1"
+ ]
+
+ @skip_unless_arch_bits(64)
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_with(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+
+ # test a bunch of combinations to test the parse response
+ # function.
+ assert await r.georadius(
+ "barcelona",
+ 2.191,
+ 41.433,
+ 1,
+ unit="km",
+ withdist=True,
+ withcoord=True,
+ withhash=True,
+ ) == [
+ [
+ b"place1",
+ 0.0881,
+ 3471609698139488,
+ (2.19093829393386841, 41.43379028184083523),
+ ]
+ ]
+
+ assert await r.georadius(
+ "barcelona", 2.191, 41.433, 1, unit="km", withdist=True, withcoord=True
+ ) == [[b"place1", 0.0881, (2.19093829393386841, 41.43379028184083523)]]
+
+ assert await r.georadius(
+ "barcelona", 2.191, 41.433, 1, unit="km", withhash=True, withcoord=True
+ ) == [
+ [b"place1", 3471609698139488, (2.19093829393386841, 41.43379028184083523)]
+ ]
+
+ # test no values.
+ assert (
+ await r.georadius(
+ "barcelona",
+ 2,
+ 1,
+ 1,
+ unit="km",
+ withdist=True,
+ withcoord=True,
+ withhash=True,
+ )
+ == []
+ )
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_count(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadius("barcelona", 2.191, 41.433, 3000, count=1) == [
+ b"place1"
+ ]
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_sort(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadius("barcelona", 2.191, 41.433, 3000, sort="ASC") == [
+ b"place1",
+ b"place2",
+ ]
+ assert await r.georadius("barcelona", 2.191, 41.433, 3000, sort="DESC") == [
+ b"place2",
+ b"place1",
+ ]
+
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_store(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ await r.georadius("barcelona", 2.191, 41.433, 1000, store="places_barcelona")
+ assert await r.zrange("places_barcelona", 0, -1) == [b"place1"]
+
+ @skip_unless_arch_bits(64)
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadius_store_dist(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ "place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ await r.georadius(
+ "barcelona", 2.191, 41.433, 1000, store_dist="places_barcelona"
+ )
+ # instead of save the geo score, the distance is saved.
+ assert await r.zscore("places_barcelona", "place1") == 88.05060698409301
+
+ @skip_unless_arch_bits(64)
+ @skip_if_server_version_lt("3.2.0")
+ async def test_georadiusmember(self, r: redis.Redis):
+ values = (2.1909389952632, 41.433791470673, "place1") + (
+ 2.1873744593677,
+ 41.406342043777,
+ b"\x80place2",
+ )
+
+ await r.geoadd("barcelona", values)
+ assert await r.georadiusbymember("barcelona", "place1", 4000) == [
+ b"\x80place2",
+ b"place1",
+ ]
+ assert await r.georadiusbymember("barcelona", "place1", 10) == [b"place1"]
+
+ assert await r.georadiusbymember(
+ "barcelona", "place1", 4000, withdist=True, withcoord=True, withhash=True
+ ) == [
+ [
+ b"\x80place2",
+ 3067.4157,
+ 3471609625421029,
+ (2.187376320362091, 41.40634178640635),
+ ],
+ [
+ b"place1",
+ 0.0,
+ 3471609698139488,
+ (2.1909382939338684, 41.433790281840835),
+ ],
+ ]
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xack(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer = "consumer"
+ # xack on a stream that doesn't exist
+ assert await r.xack(stream, group, "0-0") == 0
+
+ m1 = await r.xadd(stream, {"one": "one"})
+ m2 = await r.xadd(stream, {"two": "two"})
+ m3 = await r.xadd(stream, {"three": "three"})
+
+ # xack on a group that doesn't exist
+ assert await r.xack(stream, group, m1) == 0
+
+ await r.xgroup_create(stream, group, 0)
+ await r.xreadgroup(group, consumer, streams={stream: ">"})
+ # xack returns the number of ack'd elements
+ assert await r.xack(stream, group, m1) == 1
+ assert await r.xack(stream, group, m2, m3) == 2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xadd(self, r: redis.Redis):
+ stream = "stream"
+ message_id = await r.xadd(stream, {"foo": "bar"})
+ assert re.match(rb"[0-9]+\-[0-9]+", message_id)
+
+ # explicit message id
+ message_id = b"9999999999999999999-0"
+ assert message_id == await r.xadd(stream, {"foo": "bar"}, id=message_id)
+
+ # with maxlen, the list evicts the first message
+ await r.xadd(stream, {"foo": "bar"}, maxlen=2, approximate=False)
+ assert await r.xlen(stream) == 2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xclaim(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer1 = "consumer1"
+ consumer2 = "consumer2"
+
+ message_id = await r.xadd(stream, {"john": "wick"})
+ message = await get_stream_message(r, stream, message_id)
+ await r.xgroup_create(stream, group, 0)
+
+ # trying to claim a message that isn't already pending doesn't
+ # do anything
+ response = await r.xclaim(
+ stream, group, consumer2, min_idle_time=0, message_ids=(message_id,)
+ )
+ assert response == []
+
+ # read the group as consumer1 to initially claim the messages
+ await r.xreadgroup(group, consumer1, streams={stream: ">"})
+
+ # claim the message as consumer2
+ response = await r.xclaim(
+ stream, group, consumer2, min_idle_time=0, message_ids=(message_id,)
+ )
+ assert response[0] == message
+
+ # reclaim the message as consumer1, but use the justid argument
+ # which only returns message ids
+ assert (
+ await r.xclaim(
+ stream,
+ group,
+ consumer1,
+ min_idle_time=0,
+ message_ids=(message_id,),
+ justid=True,
+ )
+ == [message_id]
+ )
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xclaim_trimmed(self, r: redis.Redis):
+ # xclaim should not raise an exception if the item is not there
+ stream = "stream"
+ group = "group"
+
+ await r.xgroup_create(stream, group, id="$", mkstream=True)
+
+ # add a couple of new items
+ sid1 = await r.xadd(stream, {"item": 0})
+ sid2 = await r.xadd(stream, {"item": 0})
+
+ # read them from consumer1
+ await r.xreadgroup(group, "consumer1", {stream: ">"})
+
+ # add a 3rd and trim the stream down to 2 items
+ await r.xadd(stream, {"item": 3}, maxlen=2, approximate=False)
+
+ # xclaim them from consumer2
+ # the item that is still in the stream should be returned
+ item = await r.xclaim(stream, group, "consumer2", 0, [sid1, sid2])
+ assert len(item) == 2
+ assert item[0] == (None, None)
+ assert item[1][0] == sid2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xdel(self, r: redis.Redis):
+ stream = "stream"
+
+ # deleting from an empty stream doesn't do anything
+ assert await r.xdel(stream, 1) == 0
+
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ m3 = await r.xadd(stream, {"foo": "bar"})
+
+ # xdel returns the number of deleted elements
+ assert await r.xdel(stream, m1) == 1
+ assert await r.xdel(stream, m2, m3) == 2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xgroup_create(self, r: redis.Redis):
+ # tests xgroup_create and xinfo_groups
+ stream = "stream"
+ group = "group"
+ await r.xadd(stream, {"foo": "bar"})
+
+ # no group is setup yet, no info to obtain
+ assert await r.xinfo_groups(stream) == []
+
+ assert await r.xgroup_create(stream, group, 0)
+ expected = [
+ {
+ "name": group.encode(),
+ "consumers": 0,
+ "pending": 0,
+ "last-delivered-id": b"0-0",
+ }
+ ]
+ assert await r.xinfo_groups(stream) == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xgroup_create_mkstream(self, r: redis.Redis):
+ # tests xgroup_create and xinfo_groups
+ stream = "stream"
+ group = "group"
+
+ # an error is raised if a group is created on a stream that
+ # doesn't already exist
+ with pytest.raises(exceptions.ResponseError):
+ await r.xgroup_create(stream, group, 0)
+
+ # however, with mkstream=True, the underlying stream is created
+ # automatically
+ assert await r.xgroup_create(stream, group, 0, mkstream=True)
+ expected = [
+ {
+ "name": group.encode(),
+ "consumers": 0,
+ "pending": 0,
+ "last-delivered-id": b"0-0",
+ }
+ ]
+ assert await r.xinfo_groups(stream) == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xgroup_delconsumer(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer = "consumer"
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xgroup_create(stream, group, 0)
+
+ # a consumer that hasn't yet read any messages doesn't do anything
+ assert await r.xgroup_delconsumer(stream, group, consumer) == 0
+
+ # read all messages from the group
+ await r.xreadgroup(group, consumer, streams={stream: ">"})
+
+ # deleting the consumer should return 2 pending messages
+ assert await r.xgroup_delconsumer(stream, group, consumer) == 2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xgroup_destroy(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ await r.xadd(stream, {"foo": "bar"})
+
+ # destroying a nonexistent group returns False
+ assert not await r.xgroup_destroy(stream, group)
+
+ await r.xgroup_create(stream, group, 0)
+ assert await r.xgroup_destroy(stream, group)
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xgroup_setid(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ message_id = await r.xadd(stream, {"foo": "bar"})
+
+ await r.xgroup_create(stream, group, 0)
+ # advance the last_delivered_id to the message_id
+ await r.xgroup_setid(stream, group, message_id)
+ expected = [
+ {
+ "name": group.encode(),
+ "consumers": 0,
+ "pending": 0,
+ "last-delivered-id": message_id,
+ }
+ ]
+ assert await r.xinfo_groups(stream) == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xinfo_consumers(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer1 = "consumer1"
+ consumer2 = "consumer2"
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+
+ await r.xgroup_create(stream, group, 0)
+ await r.xreadgroup(group, consumer1, streams={stream: ">"}, count=1)
+ await r.xreadgroup(group, consumer2, streams={stream: ">"})
+ info = await r.xinfo_consumers(stream, group)
+ assert len(info) == 2
+ expected = [
+ {"name": consumer1.encode(), "pending": 1},
+ {"name": consumer2.encode(), "pending": 2},
+ ]
+
+ # we can't determine the idle time, so just make sure it's an int
+ assert isinstance(info[0].pop("idle"), int)
+ assert isinstance(info[1].pop("idle"), int)
+ assert info == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xinfo_stream(self, r: redis.Redis):
+ stream = "stream"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ info = await r.xinfo_stream(stream)
+
+ assert info["length"] == 2
+ assert info["first-entry"] == await get_stream_message(r, stream, m1)
+ assert info["last-entry"] == await get_stream_message(r, stream, m2)
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xlen(self, r: redis.Redis):
+ stream = "stream"
+ assert await r.xlen(stream) == 0
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+ assert await r.xlen(stream) == 2
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xpending(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer1 = "consumer1"
+ consumer2 = "consumer2"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ await r.xgroup_create(stream, group, 0)
+
+ # xpending on a group that has no consumers yet
+ expected = {"pending": 0, "min": None, "max": None, "consumers": []}
+ assert await r.xpending(stream, group) == expected
+
+ # read 1 message from the group with each consumer
+ await r.xreadgroup(group, consumer1, streams={stream: ">"}, count=1)
+ await r.xreadgroup(group, consumer2, streams={stream: ">"}, count=1)
+
+ expected = {
+ "pending": 2,
+ "min": m1,
+ "max": m2,
+ "consumers": [
+ {"name": consumer1.encode(), "pending": 1},
+ {"name": consumer2.encode(), "pending": 1},
+ ],
+ }
+ assert await r.xpending(stream, group) == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xpending_range(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer1 = "consumer1"
+ consumer2 = "consumer2"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ await r.xgroup_create(stream, group, 0)
+
+ # xpending range on a group that has no consumers yet
+ assert await r.xpending_range(stream, group, min="-", max="+", count=5) == []
+
+ # read 1 message from the group with each consumer
+ await r.xreadgroup(group, consumer1, streams={stream: ">"}, count=1)
+ await r.xreadgroup(group, consumer2, streams={stream: ">"}, count=1)
+
+ response = await r.xpending_range(stream, group, min="-", max="+", count=5)
+ assert len(response) == 2
+ assert response[0]["message_id"] == m1
+ assert response[0]["consumer"] == consumer1.encode()
+ assert response[1]["message_id"] == m2
+ assert response[1]["consumer"] == consumer2.encode()
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xrange(self, r: redis.Redis):
+ stream = "stream"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ m3 = await r.xadd(stream, {"foo": "bar"})
+ m4 = await r.xadd(stream, {"foo": "bar"})
+
+ def get_ids(results):
+ return [result[0] for result in results]
+
+ results = await r.xrange(stream, min=m1)
+ assert get_ids(results) == [m1, m2, m3, m4]
+
+ results = await r.xrange(stream, min=m2, max=m3)
+ assert get_ids(results) == [m2, m3]
+
+ results = await r.xrange(stream, max=m3)
+ assert get_ids(results) == [m1, m2, m3]
+
+ results = await r.xrange(stream, max=m2, count=1)
+ assert get_ids(results) == [m1]
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xread(self, r: redis.Redis):
+ stream = "stream"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"bing": "baz"})
+
+ expected = [
+ [
+ stream.encode(),
+ [
+ await get_stream_message(r, stream, m1),
+ await get_stream_message(r, stream, m2),
+ ],
+ ]
+ ]
+ # xread starting at 0 returns both messages
+ assert await r.xread(streams={stream: 0}) == expected
+
+ expected = [
+ [
+ stream.encode(),
+ [
+ await get_stream_message(r, stream, m1),
+ ],
+ ]
+ ]
+ # xread starting at 0 and count=1 returns only the first message
+ assert await r.xread(streams={stream: 0}, count=1) == expected
+
+ expected = [
+ [
+ stream.encode(),
+ [
+ await get_stream_message(r, stream, m2),
+ ],
+ ]
+ ]
+ # xread starting at m1 returns only the second message
+ assert await r.xread(streams={stream: m1}) == expected
+
+ # xread starting at the last message returns an empty list
+ assert await r.xread(streams={stream: m2}) == []
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xreadgroup(self, r: redis.Redis):
+ stream = "stream"
+ group = "group"
+ consumer = "consumer"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"bing": "baz"})
+ await r.xgroup_create(stream, group, 0)
+
+ expected = [
+ [
+ stream.encode(),
+ [
+ await get_stream_message(r, stream, m1),
+ await get_stream_message(r, stream, m2),
+ ],
+ ]
+ ]
+ # xread starting at 0 returns both messages
+ assert await r.xreadgroup(group, consumer, streams={stream: ">"}) == expected
+
+ await r.xgroup_destroy(stream, group)
+ await r.xgroup_create(stream, group, 0)
+
+ expected = [
+ [
+ stream.encode(),
+ [
+ await get_stream_message(r, stream, m1),
+ ],
+ ]
+ ]
+ # xread with count=1 returns only the first message
+ assert (
+ await r.xreadgroup(group, consumer, streams={stream: ">"}, count=1)
+ == expected
+ )
+
+ await r.xgroup_destroy(stream, group)
+
+ # create the group using $ as the last id meaning subsequent reads
+ # will only find messages added after this
+ await r.xgroup_create(stream, group, "$")
+
+ expected = []
+ # xread starting after the last message returns an empty message list
+ assert await r.xreadgroup(group, consumer, streams={stream: ">"}) == expected
+
+ # xreadgroup with noack does not have any items in the PEL
+ await r.xgroup_destroy(stream, group)
+ await r.xgroup_create(stream, group, "0")
+ assert (
+ len(
+ (
+ await r.xreadgroup(
+ group, consumer, streams={stream: ">"}, noack=True
+ )
+ )[0][1]
+ )
+ == 2
+ )
+ # now there should be nothing pending
+ assert (
+ len((await r.xreadgroup(group, consumer, streams={stream: "0"}))[0][1]) == 0
+ )
+
+ await r.xgroup_destroy(stream, group)
+ await r.xgroup_create(stream, group, "0")
+ # delete all the messages in the stream
+ expected = [
+ [
+ stream.encode(),
+ [
+ (m1, {}),
+ (m2, {}),
+ ],
+ ]
+ ]
+ await r.xreadgroup(group, consumer, streams={stream: ">"})
+ await r.xtrim(stream, 0)
+ assert await r.xreadgroup(group, consumer, streams={stream: "0"}) == expected
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xrevrange(self, r: redis.Redis):
+ stream = "stream"
+ m1 = await r.xadd(stream, {"foo": "bar"})
+ m2 = await r.xadd(stream, {"foo": "bar"})
+ m3 = await r.xadd(stream, {"foo": "bar"})
+ m4 = await r.xadd(stream, {"foo": "bar"})
+
+ def get_ids(results):
+ return [result[0] for result in results]
+
+ results = await r.xrevrange(stream, max=m4)
+ assert get_ids(results) == [m4, m3, m2, m1]
+
+ results = await r.xrevrange(stream, max=m3, min=m2)
+ assert get_ids(results) == [m3, m2]
+
+ results = await r.xrevrange(stream, min=m3)
+ assert get_ids(results) == [m4, m3]
+
+ results = await r.xrevrange(stream, min=m2, count=1)
+ assert get_ids(results) == [m4]
+
+ @skip_if_server_version_lt("5.0.0")
+ async def test_xtrim(self, r: redis.Redis):
+ stream = "stream"
+
+ # trimming an empty key doesn't do anything
+ assert await r.xtrim(stream, 1000) == 0
+
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+ await r.xadd(stream, {"foo": "bar"})
+
+ # trimming an amount large than the number of messages
+ # doesn't do anything
+ assert await r.xtrim(stream, 5, approximate=False) == 0
+
+ # 1 message is trimmed
+ assert await r.xtrim(stream, 3, approximate=False) == 1
+
+ async def test_bitfield_operations(self, r: redis.Redis):
+ # comments show affected bits
+ await r.execute_command("SELECT", 10)
+ bf = r.bitfield("a")
+ resp = await (
+ bf.set("u8", 8, 255) # 00000000 11111111
+ .get("u8", 0) # 00000000
+ .get("u4", 8) # 1111
+ .get("u4", 12) # 1111
+ .get("u4", 13) # 111 0
+ .execute()
+ )
+ assert resp == [0, 0, 15, 15, 14]
+
+ # .set() returns the previous value...
+ resp = await (
+ bf.set("u8", 4, 1) # 0000 0001
+ .get("u16", 0) # 00000000 00011111
+ .set("u16", 0, 0) # 00000000 00000000
+ .execute()
+ )
+ assert resp == [15, 31, 31]
+
+ # incrby adds to the value
+ resp = await (
+ bf.incrby("u8", 8, 254) # 00000000 11111110
+ .incrby("u8", 8, 1) # 00000000 11111111
+ .get("u16", 0) # 00000000 11111111
+ .execute()
+ )
+ assert resp == [254, 255, 255]
+
+ # Verify overflow protection works as a method:
+ await r.delete("a")
+ resp = await (
+ bf.set("u8", 8, 254) # 00000000 11111110
+ .overflow("fail")
+ .incrby("u8", 8, 2) # incrby 2 would overflow, None returned
+ .incrby("u8", 8, 1) # 00000000 11111111
+ .incrby("u8", 8, 1) # incrby 1 would overflow, None returned
+ .get("u16", 0) # 00000000 11111111
+ .execute()
+ )
+ assert resp == [0, None, 255, None, 255]
+
+ # Verify overflow protection works as arg to incrby:
+ await r.delete("a")
+ resp = await (
+ bf.set("u8", 8, 255) # 00000000 11111111
+ .incrby("u8", 8, 1) # 00000000 00000000 wrap default
+ .set("u8", 8, 255) # 00000000 11111111
+ .incrby("u8", 8, 1, "FAIL") # 00000000 11111111 fail
+ .incrby("u8", 8, 1) # 00000000 11111111 still fail
+ .get("u16", 0) # 00000000 11111111
+ .execute()
+ )
+ assert resp == [0, 0, 0, None, None, 255]
+
+ # test default default_overflow
+ await r.delete("a")
+ bf = r.bitfield("a", default_overflow="FAIL")
+ resp = await (
+ bf.set("u8", 8, 255) # 00000000 11111111
+ .incrby("u8", 8, 1) # 00000000 11111111 fail default
+ .get("u16", 0) # 00000000 11111111
+ .execute()
+ )
+ assert resp == [0, None, 255]
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_memory_stats(self, r: redis.Redis):
+ # put a key into the current db to make sure that "db.<current-db>"
+ # has data
+ await r.set("foo", "bar")
+ stats = await r.memory_stats()
+ assert isinstance(stats, dict)
+ for key, value in stats.items():
+ if key.startswith("db."):
+ assert isinstance(value, dict)
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_memory_usage(self, r: redis.Redis):
+ await r.set("foo", "bar")
+ assert isinstance(await r.memory_usage("foo"), int)
+
+ @skip_if_server_version_lt("4.0.0")
+ async def test_module_list(self, r: redis.Redis):
+ assert isinstance(await r.module_list(), list)
+ assert not await r.module_list()
+
+
+class TestBinarySave:
+ async def test_binary_get_set(self, r: redis.Redis):
+ assert await r.set(" foo bar ", "123")
+ assert await r.get(" foo bar ") == b"123"
+
+ assert await r.set(" foo\r\nbar\r\n ", "456")
+ assert await r.get(" foo\r\nbar\r\n ") == b"456"
+
+ assert await r.set(" \r\n\t\x07\x13 ", "789")
+ assert await r.get(" \r\n\t\x07\x13 ") == b"789"
+
+ assert sorted(await r.keys("*")) == [
+ b" \r\n\t\x07\x13 ",
+ b" foo\r\nbar\r\n ",
+ b" foo bar ",
+ ]
+
+ assert await r.delete(" foo bar ")
+ assert await r.delete(" foo\r\nbar\r\n ")
+ assert await r.delete(" \r\n\t\x07\x13 ")
+
+ async def test_binary_lists(self, r: redis.Redis):
+ mapping = {
+ b"foo bar": [b"1", b"2", b"3"],
+ b"foo\r\nbar\r\n": [b"4", b"5", b"6"],
+ b"foo\tbar\x07": [b"7", b"8", b"9"],
+ }
+ # fill in lists
+ for key, value in mapping.items():
+ await r.rpush(key, *value)
+
+ # check that KEYS returns all the keys as they are
+ assert sorted(await r.keys("*")) == sorted(mapping.keys())
+
+ # check that it is possible to get list content by key name
+ for key, value in mapping.items():
+ assert await r.lrange(key, 0, -1) == value
+
+ async def test_22_info(self, r: redis.Redis):
+ """
+ Older Redis versions contained 'allocation_stats' in INFO that
+ was the cause of a number of bugs when parsing.
+ """
+ info = (
+ "allocation_stats:6=1,7=1,8=7141,9=180,10=92,11=116,12=5330,"
+ "13=123,14=3091,15=11048,16=225842,17=1784,18=814,19=12020,"
+ "20=2530,21=645,22=15113,23=8695,24=142860,25=318,26=3303,"
+ "27=20561,28=54042,29=37390,30=1884,31=18071,32=31367,33=160,"
+ "34=169,35=201,36=10155,37=1045,38=15078,39=22985,40=12523,"
+ "41=15588,42=265,43=1287,44=142,45=382,46=945,47=426,48=171,"
+ "49=56,50=516,51=43,52=41,53=46,54=54,55=75,56=647,57=332,"
+ "58=32,59=39,60=48,61=35,62=62,63=32,64=221,65=26,66=30,"
+ "67=36,68=41,69=44,70=26,71=144,72=169,73=24,74=37,75=25,"
+ "76=42,77=21,78=126,79=374,80=27,81=40,82=43,83=47,84=46,"
+ "85=114,86=34,87=37,88=7240,89=34,90=38,91=18,92=99,93=20,"
+ "94=18,95=17,96=15,97=22,98=18,99=69,100=17,101=22,102=15,"
+ "103=29,104=39,105=30,106=70,107=22,108=21,109=26,110=52,"
+ "111=45,112=33,113=67,114=41,115=44,116=48,117=53,118=54,"
+ "119=51,120=75,121=44,122=57,123=44,124=66,125=56,126=52,"
+ "127=81,128=108,129=70,130=50,131=51,132=53,133=45,134=62,"
+ "135=12,136=13,137=7,138=15,139=21,140=11,141=20,142=6,143=7,"
+ "144=11,145=6,146=16,147=19,148=1112,149=1,151=83,154=1,"
+ "155=1,156=1,157=1,160=1,161=1,162=2,166=1,169=1,170=1,171=2,"
+ "172=1,174=1,176=2,177=9,178=34,179=73,180=30,181=1,185=3,"
+ "187=1,188=1,189=1,192=1,196=1,198=1,200=1,201=1,204=1,205=1,"
+ "207=1,208=1,209=1,214=2,215=31,216=78,217=28,218=5,219=2,"
+ "220=1,222=1,225=1,227=1,234=1,242=1,250=1,252=1,253=1,"
+ ">=256=203"
+ )
+ parsed = parse_info(info)
+ assert "allocation_stats" in parsed
+ assert "6" in parsed["allocation_stats"]
+ assert ">=256" in parsed["allocation_stats"]
+
+ async def test_large_responses(self, r: redis.Redis):
+ """The PythonParser has some special cases for return values > 1MB"""
+ # load up 5MB of data into a key
+ data = "".join([ascii_letters] * (5000000 // len(ascii_letters)))
+ await r.set("a", data)
+ assert await r.get("a") == data.encode()
+
+ async def test_floating_point_encoding(self, r: redis.Redis):
+ """
+ High precision floating point values sent to the server should keep
+ precision.
+ """
+ timestamp = 1349673917.939762
+ await r.zadd("a", {"a1": timestamp})
+ assert await r.zscore("a", "a1") == timestamp