Skip to content

Commit a9b8c23

Browse files
authored
Merge pull request #1211 from supercaracal/remove-flaky-and-redundant-cases-for-pubsub-in-cluster-mode
Remove flaky and redundant test cases for Pub/Sub in cluster mode
2 parents c92e403 + 92dc3fd commit a9b8c23

File tree

1 file changed

+12
-39
lines changed

1 file changed

+12
-39
lines changed

cluster/test/commands_on_pub_sub_test.rb

Lines changed: 12 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,9 @@ def test_publish_subscribe_unsubscribe_pubsub
1313

1414
thread = Thread.new do
1515
redis.subscribe('channel1', 'channel2') do |on|
16-
on.subscribe { |_c, t| sub_cnt = t }
17-
on.unsubscribe { |_c, t| sub_cnt = t }
16+
on.subscribe { sub_cnt += 1 }
1817
on.message do |c, msg|
1918
messages[c] = msg
20-
# FIXME: blocking occurs when `unsubscribe` method was called with channel arguments
2119
redis.unsubscribe if messages.size == 2
2220
end
2321
end
@@ -27,28 +25,19 @@ def test_publish_subscribe_unsubscribe_pubsub
2725

2826
publisher = build_another_client
2927

30-
assert_equal %w[channel1 channel2], publisher.pubsub(:channels)
31-
assert_equal %w[channel1 channel2], publisher.pubsub(:channels, 'cha*')
32-
assert_equal [], publisher.pubsub(:channels, 'chachacha*')
33-
assert_equal({}, publisher.pubsub(:numsub))
28+
assert_equal %w[channel1 channel2], publisher.pubsub(:channels, 'channel*')
3429
assert_equal({ 'channel1' => 1, 'channel2' => 1, 'channel3' => 0 },
3530
publisher.pubsub(:numsub, 'channel1', 'channel2', 'channel3'))
36-
assert_equal 0, publisher.pubsub(:numpat)
3731

3832
publisher.publish('channel1', 'one')
3933
publisher.publish('channel2', 'two')
34+
publisher.publish('channel3', 'three')
4035

4136
thread.join
4237

43-
assert_equal({ 'channel1' => 'one', 'channel2' => 'two' }, messages.sort.to_h)
44-
45-
assert_equal [], publisher.pubsub(:channels)
46-
assert_equal [], publisher.pubsub(:channels, 'cha*')
47-
assert_equal [], publisher.pubsub(:channels, 'chachacha*')
48-
assert_equal({}, publisher.pubsub(:numsub))
49-
assert_equal({ 'channel1' => 0, 'channel2' => 0, 'channel3' => 0 },
50-
publisher.pubsub(:numsub, 'channel1', 'channel2', 'channel3'))
51-
assert_equal 0, publisher.pubsub(:numpat)
38+
assert_equal(2, messages.size)
39+
assert_equal('one', messages['channel1'])
40+
assert_equal('two', messages['channel2'])
5241
end
5342

5443
def test_publish_psubscribe_punsubscribe_pubsub
@@ -57,11 +46,9 @@ def test_publish_psubscribe_punsubscribe_pubsub
5746

5847
thread = Thread.new do
5948
redis.psubscribe('guc*', 'her*') do |on|
60-
on.psubscribe { |_c, t| sub_cnt = t }
61-
on.punsubscribe { |_c, t| sub_cnt = t }
62-
on.pmessage do |_ptn, chn, msg|
63-
messages[chn] = msg
64-
# FIXME: blocking occurs when `unsubscribe` method was called with channel arguments
49+
on.psubscribe { sub_cnt += 1 }
50+
on.pmessage do |_ptn, c, msg|
51+
messages[c] = msg
6552
redis.punsubscribe if messages.size == 2
6653
end
6754
end
@@ -71,13 +58,6 @@ def test_publish_psubscribe_punsubscribe_pubsub
7158

7259
publisher = build_another_client
7360

74-
assert_equal [], publisher.pubsub(:channels)
75-
assert_equal [], publisher.pubsub(:channels, 'bur*')
76-
assert_equal [], publisher.pubsub(:channels, 'guc*')
77-
assert_equal [], publisher.pubsub(:channels, 'her*')
78-
assert_equal({}, publisher.pubsub(:numsub))
79-
assert_equal({ 'burberry1' => 0, 'gucci2' => 0, 'hermes3' => 0 },
80-
publisher.pubsub(:numsub, 'burberry1', 'gucci2', 'hermes3'))
8161
assert_equal 2, publisher.pubsub(:numpat)
8262

8363
publisher.publish('burberry1', 'one')
@@ -86,15 +66,8 @@ def test_publish_psubscribe_punsubscribe_pubsub
8666

8767
thread.join
8868

89-
assert_equal({ 'gucci2' => 'two', 'hermes3' => 'three' }, messages.sort.to_h)
90-
91-
assert_equal [], publisher.pubsub(:channels)
92-
assert_equal [], publisher.pubsub(:channels, 'bur*')
93-
assert_equal [], publisher.pubsub(:channels, 'guc*')
94-
assert_equal [], publisher.pubsub(:channels, 'her*')
95-
assert_equal({}, publisher.pubsub(:numsub))
96-
assert_equal({ 'burberry1' => 0, 'gucci2' => 0, 'hermes3' => 0 },
97-
publisher.pubsub(:numsub, 'burberry1', 'gucci2', 'hermes3'))
98-
assert_equal 0, publisher.pubsub(:numpat)
69+
assert_equal(2, messages.size)
70+
assert_equal('two', messages['gucci2'])
71+
assert_equal('three', messages['hermes3'])
9972
end
10073
end

0 commit comments

Comments
 (0)