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
100
101
102
103
104
105
106
107
108
|
# frozen_string_literal: true
require "helper"
# ruby -w -Itest test/cluster_commands_on_pub_sub_test.rb
# @see https://redis.io/commands#pubsub
class TestClusterCommandsOnPubSub < Minitest::Test
include Helper::Cluster
def test_publish_subscribe_unsubscribe_pubsub
sub_cnt = 0
messages = {}
thread = Thread.new do
redis.subscribe('channel1', 'channel2') do |on|
on.subscribe { sub_cnt += 1 }
on.message do |c, msg|
messages[c] = msg
redis.unsubscribe if messages.size == 2
end
end
end
Thread.pass until sub_cnt == 2
publisher = build_another_client
assert_equal %w[channel1 channel2], publisher.pubsub(:channels, 'channel*')
assert_equal({ 'channel1' => 1, 'channel2' => 1, 'channel3' => 0 },
publisher.pubsub(:numsub, 'channel1', 'channel2', 'channel3'))
publisher.publish('channel1', 'one')
publisher.publish('channel2', 'two')
publisher.publish('channel3', 'three')
thread.join
assert_equal(2, messages.size)
assert_equal('one', messages['channel1'])
assert_equal('two', messages['channel2'])
end
def test_publish_psubscribe_punsubscribe_pubsub
sub_cnt = 0
messages = {}
thread = Thread.new do
redis.psubscribe('guc*', 'her*') do |on|
on.psubscribe { sub_cnt += 1 }
on.pmessage do |_ptn, c, msg|
messages[c] = msg
redis.punsubscribe if messages.size == 2
end
end
end
Thread.pass until sub_cnt == 2
publisher = build_another_client
assert_equal 2, publisher.pubsub(:numpat)
publisher.publish('burberry1', 'one')
publisher.publish('gucci2', 'two')
publisher.publish('hermes3', 'three')
thread.join
assert_equal(2, messages.size)
assert_equal('two', messages['gucci2'])
assert_equal('three', messages['hermes3'])
end
def test_spublish_ssubscribe_sunsubscribe_pubsub
omit_version('7.0.0')
sub_cnt = 0
messages = {}
thread = Thread.new do
redis.ssubscribe('channel1', 'channel2') do |on|
on.ssubscribe { sub_cnt += 1 }
on.smessage do |c, msg|
messages[c] = msg
redis.sunsubscribe if messages.size == 2
end
end
end
Thread.pass until sub_cnt == 2
publisher = build_another_client
assert_equal %w[channel1 channel2], publisher.pubsub(:shardchannels, 'channel*')
assert_equal({ 'channel1' => 1, 'channel2' => 1, 'channel3' => 0 },
publisher.pubsub(:shardnumsub, 'channel1', 'channel2', 'channel3'))
publisher.spublish('channel1', 'one')
publisher.spublish('channel2', 'two')
publisher.spublish('channel3', 'three')
thread.join
assert_equal(2, messages.size)
assert_equal('one', messages['channel1'])
assert_equal('two', messages['channel2'])
end
end
|
