Re-enable and fix remove-node test (#5499)

buildbuddy-io · Dec 11, 2023 · 30e62ed · 30e62ed
1 parent 025d8fa
commit 30e62ed
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/enterprise/server/raft/store/store_test.go b/enterprise/server/raft/store/store_test.go
@@ -224,6 +224,9 @@ func TestAddNodeToCluster(t *testing.T) {
 	})
 	require.NoError(t, err)
 
+	stores := []*TestingStore{s1, s2, s3, s4}
+	waitForRangeLease(t, stores, 2)
+
 	rd := s1.GetRange(1)
 	_, err = s1.AddReplica(ctx, &rfpb.AddReplicaRequest{
 		Range: rd,
@@ -241,7 +244,6 @@ func TestAddNodeToCluster(t *testing.T) {
 }
 
 func TestRemoveNodeFromCluster(t *testing.T) {
-	t.Skip()
 	sf := newStoreFactory(t)
 	s1, nh1 := sf.NewStore(t)
 	s2, nh2 := sf.NewStore(t)
@@ -257,6 +259,9 @@ func TestRemoveNodeFromCluster(t *testing.T) {
 	})
 	require.NoError(t, err)
 
+	stores := []*TestingStore{s1, s2, s3, s4}
+	waitForRangeLease(t, stores, 2)
+
 	rd := s1.GetRange(1)
 	_, err = s1.RemoveReplica(ctx, &rfpb.RemoveReplicaRequest{
 		Range:     rd,