1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
|
from datetime import datetime
from time import sleep
import pytest
import redis
from redis.exceptions import ResponseError
from fakeredis._commands import SUPPORTED_COMMANDS
from test.testtools import fake_only
@pytest.mark.unsupported_server_types("dragonfly")
def test_swapdb(r, create_connection):
r1 = create_connection(3)
r.set("foo", "abc")
r.set("bar", "xyz")
r1.set("foo", "foo")
r1.set("baz", "baz")
assert r.swapdb(2, 3)
assert r.get("foo") == b"foo"
assert r.get("bar") is None
assert r.get("baz") == b"baz"
assert r1.get("foo") == b"abc"
assert r1.get("bar") == b"xyz"
assert r1.get("baz") is None
@pytest.mark.unsupported_server_types("dragonfly")
def test_swapdb_same_db(r: redis.Redis):
assert r.swapdb(1, 1)
def test_save(r: redis.Redis):
assert r.save()
@pytest.mark.unsupported_server_types("dragonfly")
def test_bgsave(r: redis.Redis):
assert r.bgsave()
with pytest.raises(ResponseError):
r.execute_command("BGSAVE", "SCHEDULE", "FOO")
with pytest.raises(ResponseError):
r.execute_command("BGSAVE", "FOO")
def test_lastsave(r: redis.Redis):
assert isinstance(r.lastsave(), datetime)
@fake_only
def test_command(r: redis.Redis):
commands_dict = r.command()
one_word_commands = {cmd for cmd in SUPPORTED_COMMANDS if " " not in cmd and SUPPORTED_COMMANDS[cmd].server_types}
server_unsupported_commands = one_word_commands - set(commands_dict.keys()) - {"hgetex", "hsetex", "hgetdel"}
for command in server_unsupported_commands:
assert "redis" not in SUPPORTED_COMMANDS[command].server_types, (
f"Command {command} is not supported by fakeredis"
)
@fake_only
def test_command_count(r: redis.Redis):
assert r.command_count() >= len(
[cmd for (cmd, cmd_info) in SUPPORTED_COMMANDS.items() if " " not in cmd and "redis" in cmd_info.server_types]
)
@pytest.mark.unsupported_server_types("dragonfly")
@pytest.mark.slow
def test_bgsave_timestamp_update(r: redis.Redis):
early_timestamp = r.lastsave()
sleep(1)
assert r.bgsave()
sleep(1)
late_timestamp = r.lastsave()
assert early_timestamp < late_timestamp
@pytest.mark.slow
def test_save_timestamp_update(r: redis.Redis):
early_timestamp = r.lastsave()
sleep(1)
assert r.save()
late_timestamp = r.lastsave()
assert early_timestamp < late_timestamp
def test_dbsize(r: redis.Redis):
assert r.dbsize() == 0
r.set("foo", "bar")
r.set("bar", "foo")
assert r.dbsize() == 2
def test_flushdb_redispy4(r: redis.Redis):
r.set("foo", "bar")
assert r.keys() == [b"foo"]
assert r.flushdb() is True
assert r.keys() == []
|