File: test_string.py

package info (click to toggle)
python-fakeredis 2.29.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,772 kB
  • sloc: python: 19,002; sh: 8; makefile: 5
file content (42 lines) | stat: -rw-r--r-- 1,581 bytes parent folder | download
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