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