summaryrefslogtreecommitdiff
path: root/tests/unit/moduleapi/blockonkeys.tcl
blob: cb99ab1c98ddc8ab0369b5d0956a5202a6b14d54 (plain)
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
set testmodule [file normalize tests/modules/blockonkeys.so]

start_server {tags {"modules"}} {
    r module load $testmodule

    test {Module client blocked on keys (no metadata): No block} {
        r del k
        r fsl.push k 33
        r fsl.push k 34
        r fsl.bpop2 k 0
    } {34 33}

    test {Module client blocked on keys (no metadata): Timeout} {
        r del k
        set rd [redis_deferring_client]
        r fsl.push k 33
        $rd fsl.bpop2 k 1
        assert_equal {Request timedout} [$rd read]
    }

    test {Module client blocked on keys (no metadata): Blocked, case 1} {
        r del k
        set rd [redis_deferring_client]
        r fsl.push k 33
        $rd fsl.bpop2 k 0
        r fsl.push k 34
        assert_equal {34 33} [$rd read]
    }

    test {Module client blocked on keys (no metadata): Blocked, case 2} {
        r del k
        set rd [redis_deferring_client]
        r fsl.push k 33
        r fsl.push k 34
        $rd fsl.bpop2 k 0
        assert_equal {34 33} [$rd read]
    }

    test {Module client blocked on keys (with metadata): No block} {
        r del k
        r fsl.push k 34
        r fsl.bpopgt k 30 0
    } {34}

    test {Module client blocked on keys (with metadata): Timeout} {
        r del k
        set rd [redis_deferring_client]
        r fsl.push k 33
        $rd fsl.bpopgt k 35 1
        assert_equal {Request timedout} [$rd read]
    }

    test {Module client blocked on keys (with metadata): Blocked, case 1} {
        r del k
        set rd [redis_deferring_client]
        r fsl.push k 33
        $rd fsl.bpopgt k 33 0
        r fsl.push k 34
        assert_equal {34} [$rd read]
    }

    test {Module client blocked on keys (with metadata): Blocked, case 2} {
        r del k
        set rd [redis_deferring_client]
        $rd fsl.bpopgt k 35 0
        r fsl.push k 33
        r fsl.push k 34
        r fsl.push k 35
        r fsl.push k 36
        assert_equal {36} [$rd read]
    }

    test {Module client blocked on keys does not wake up on wrong type} {
        r del k
        set rd [redis_deferring_client]
        $rd fsl.bpop2 k 0
        r lpush k 12
        r lpush k 13
        r lpush k 14
        r del k
        r fsl.push k 33
        r fsl.push k 34
        assert_equal {34 33} [$rd read]
    }
}