From f7708e72c05de361551c332190e51fe1808407a8 Mon Sep 17 00:00:00 2001 From: zphrs <z@zephiris.dev> Date: Thu, 13 Mar 2025 23:20:59 -0700 Subject: [PATCH] fixed shuffle --- tests/shuffle/basic_shuffle.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py index c23cda6..68c98ed 100644 --- a/tests/shuffle/basic_shuffle.py +++ b/tests/shuffle/basic_shuffle.py @@ -72,7 +72,8 @@ def basic_shuffle(conductor: ClusterConductor, dir, log: Logger): # Remove shard 3, causing a shuffle. Move Node 2 to shard 1 so the keys should still exist, and be shuffled conductor.remove_shard("shard3") - conductor.add_node_to_shard("shard1", conductor.get_nodes([2])) + conductor.add_node_to_shard("shard1", conductor.get_node(2)) + fx.broadcast_view(conductor.get_shard_view()) r = c.get_all(0, timeout=10) # should get all of shard 1's keys assert r.ok, f"expected ok for get, got {r.status_code}" -- GitLab