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
|
from .base import (
BaseTest,
commands,
values,
st,
keys,
common_commands,
counts,
zero_or_more,
expires_seconds,
expires_ms,
ints,
int_as_bytes,
)
optional_bitcount_range = st.just(()) | st.tuples(int_as_bytes, int_as_bytes)
str_len = st.integers(min_value=-3, max_value=3) | st.integers(min_value=-3000, max_value=3000)
string_commands = (
commands(st.just("append"), keys, values)
| commands(st.just("bitcount"), keys, optional_bitcount_range)
| commands(st.sampled_from(["incr", "decr"]), keys)
| commands(st.sampled_from(["incrby", "decrby"]), keys, values)
| commands(st.just("get"), keys)
| commands(st.just("getbit"), keys, counts)
| commands(st.just("setbit"), keys, counts, st.integers(min_value=0, max_value=1) | ints)
| commands(st.sampled_from(["substr", "getrange"]), keys, str_len, counts)
| commands(st.just("getset"), keys, values)
| commands(st.just("mget"), st.lists(keys))
| commands(st.sampled_from(["mset", "msetnx"]), st.lists(st.tuples(keys, values)))
| commands(st.just("set"), keys, values, *zero_or_more("nx", "xx", "keepttl"))
| commands(st.just("setex"), keys, expires_seconds, values)
| commands(st.just("psetex"), keys, expires_ms, values)
| commands(st.just("setnx"), keys, values)
| commands(st.just("setrange"), keys, str_len, values)
| commands(st.just("strlen"), keys)
)
class TestString(BaseTest):
create_command_strategy = commands(st.just("set"), keys, values)
command_strategy = string_commands | common_commands
|