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
|
# frozen_string_literal: true
require_relative '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 = {}
wire = Wire.new do
redis.subscribe('channel1', 'channel2') do |on|
on.subscribe { |_c, t| sub_cnt = t }
on.unsubscribe { |_c, t| sub_cnt = t }
on.message do |c, msg|
messages[c] = msg
# FIXME: blocking occurs when `unsubscribe` method was called with channel arguments
redis.unsubscribe if messages.size == 2
end
end
end
Wire.pass until sub_cnt == 2
publisher = build_another_client
assert_equal %w[channel1 channel2], publisher.pubsub(:channels)
assert_equal %w[channel1 channel2], publisher.pubsub(:channels, 'cha*')
assert_equal [], publisher.pubsub(:channels, 'chachacha*')
assert_equal({}, publisher.pubsub(:numsub))
assert_equal({ 'channel1' => 1, 'channel2' => 1, 'channel3' => 0 },
publisher.pubsub(:numsub, 'channel1', 'channel2', 'channel3'))
assert_equal 0, publisher.pubsub(:numpat)
publisher.publish('channel1', 'one')
publisher.publish('channel2', 'two')
wire.join
assert_equal({ 'channel1' => 'one', 'channel2' => 'two' }, messages.sort.to_h)
assert_equal [], publisher.pubsub(:channels)
assert_equal [], publisher.pubsub(:channels, 'cha*')
assert_equal [], publisher.pubsub(:channels, 'chachacha*')
assert_equal({}, publisher.pubsub(:numsub))
assert_equal({ 'channel1' => 0, 'channel2' => 0, 'channel3' => 0 },
publisher.pubsub(:numsub, 'channel1', 'channel2', 'channel3'))
assert_equal 0, publisher.pubsub(:numpat)
end
def test_publish_psubscribe_punsubscribe_pubsub
sub_cnt = 0
messages = {}
wire = Wire.new do
redis.psubscribe('guc*', 'her*') do |on|
on.psubscribe { |_c, t| sub_cnt = t }
on.punsubscribe { |_c, t| sub_cnt = t }
on.pmessage do |_ptn, chn, msg|
messages[chn] = msg
# FIXME: blocking occurs when `unsubscribe` method was called with channel arguments
redis.punsubscribe if messages.size == 2
end
end
end
Wire.pass until sub_cnt == 2
publisher = build_another_client
assert_equal [], publisher.pubsub(:channels)
assert_equal [], publisher.pubsub(:channels, 'bur*')
assert_equal [], publisher.pubsub(:channels, 'guc*')
assert_equal [], publisher.pubsub(:channels, 'her*')
assert_equal({}, publisher.pubsub(:numsub))
assert_equal({ 'burberry1' => 0, 'gucci2' => 0, 'hermes3' => 0 }, publisher.pubsub(:numsub, 'burberry1', 'gucci2', 'hermes3'))
assert_equal 2, publisher.pubsub(:numpat)
publisher.publish('burberry1', 'one')
publisher.publish('gucci2', 'two')
publisher.publish('hermes3', 'three')
wire.join
assert_equal({ 'gucci2' => 'two', 'hermes3' => 'three' }, messages.sort.to_h)
assert_equal [], publisher.pubsub(:channels)
assert_equal [], publisher.pubsub(:channels, 'bur*')
assert_equal [], publisher.pubsub(:channels, 'guc*')
assert_equal [], publisher.pubsub(:channels, 'her*')
assert_equal({}, publisher.pubsub(:numsub))
assert_equal({ 'burberry1' => 0, 'gucci2' => 0, 'hermes3' => 0 }, publisher.pubsub(:numsub, 'burberry1', 'gucci2', 'hermes3'))
assert_equal 0, publisher.pubsub(:numpat)
end
end
|