mirrors/dragonfly
1
0
Fork 0
mirror of https://github.com/dragonflydb/dragonfly.git synced 2025-05-10 18:05:44 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fix(test): Execute SSUBSCRIBE command directly in client (#4763)

Browse source 
Using `pubsub.ssubscribe` function doesn't wait for any response from
server. Fixed by switching to execute SSUBSCRIBE commands in client directly.

Signed-off-by: mkaruza <mario@dragonflydb.io>
This commit is contained in:
mkaruza 2025-03-14 16:01:30 +01:00 • committed by GitHub
parent 8d6a18465b
commit 5facf12836
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/dragonfly/cluster_test.py
View file

@ -3000,14 +3000,10 @@ async def test_cluster_sharded_pubsub_shard_commands(df_factory: DflyInstanceFac
await push_config(json.dumps(generate_config(nodes_info)), [node.client for node in nodes_info])
node_a = ClusterNode("localhost", nodes[0].port)
node_b = ClusterNode("localhost", nodes[1].port)
consumer_client = RedisCluster(startup_nodes=[node_a, node_b])
consumer = consumer_client.pubsub()
consumer.ssubscribe("pubsub-shard-channel")
consumer.ssubscribe("shard-channel")
# We are executing SSUBSCRIBE commands and wait for them to be sure that
# channels are created
message = await c_nodes[0].execute_command("SSUBSCRIBE pubsub-shard-channel")
message = await c_nodes[0].execute_command("SSUBSCRIBE shard-channel")
message = await c_nodes[0].execute_command("PUBSUB SHARDCHANNELS")
message.sort()