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
|
import hiredis
import pytest
testdata = [
# all_types
(('HSET', 'foo', 'key', 'value1', b'key_b', b'bytes str', 'key_mv', memoryview(b'bytes str'), b'key_i', 67, 'key_f', 3.14159265359),
b'*12\r\n$4\r\nHSET\r\n$3\r\nfoo\r\n$3\r\nkey\r\n$6\r\nvalue1\r\n$5\r\nkey_b\r\n$9\r\nbytes str\r\n$6\r\nkey_mv\r\n$9\r\nbytes str\r\n$5\r\nkey_i\r\n$2\r\n67\r\n$5\r\nkey_f\r\n$13\r\n3.14159265359\r\n'),
# spaces_in_cmd
(('COMMAND', 'GETKEYS', 'EVAL', 'return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}', 2, 'key1', 'key2', 'first', 'second'),
b'*9\r\n$7\r\nCOMMAND\r\n$7\r\nGETKEYS\r\n$4\r\nEVAL\r\n$40\r\nreturn {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}\r\n$1\r\n2\r\n$4\r\nkey1\r\n$4\r\nkey2\r\n$5\r\nfirst\r\n$6\r\nsecond\r\n'),
# null_chars
(('SET', 'a', bytes(b'\xaa\x00\xffU')),
b'*3\r\n$3\r\nSET\r\n$1\r\na\r\n$4\r\n\xaa\x00\xffU\r\n'),
# encoding
(('SET', 'a', "йדב문諺"),
b'*3\r\n$3\r\nSET\r\n$1\r\na\r\n$12\r\n\xD0\xB9\xD7\x93\xD7\x91\xEB\xAC\xB8\xE8\xAB\xBA\r\n'),
# big_int
(('SET', 'a', 2**128),
b'*3\r\n$3\r\nSET\r\n$1\r\na\r\n$39\r\n340282366920938463463374607431768211456\r\n'),
]
testdata_ids = [
"all_types",
"spaces_in_cmd",
"null_chars",
"encoding",
"big_int",
]
@pytest.mark.parametrize("cmd,expected_packed_cmd", testdata, ids=testdata_ids)
def test_basic(cmd, expected_packed_cmd):
packed_cmd = hiredis.pack_command(cmd)
assert packed_cmd == expected_packed_cmd
def test_wrong_type():
with pytest.raises(TypeError):
hiredis.pack_command(("HSET", "foo", True))
class Bar:
def __init__(self) -> None:
self.key = "key"
self.value = 36
with pytest.raises(TypeError):
hiredis.pack_command(("HSET", "foo", Bar()))
|
