make TestDefaultKeyspaceGroup stable

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Nov 24, 2023 · 5a85b97 · 5a85b97
1 parent eafc1ed
commit 5a85b97
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/integrations/mcs/keyspace/tso_keyspace_group_test.go b/tests/integrations/mcs/keyspace/tso_keyspace_group_test.go
@@ -301,7 +301,10 @@ func (suite *keyspaceGroupTestSuite) TestDefaultKeyspaceGroup() {
 		return code == http.StatusOK && kg != nil
 	}, testutil.WithWaitFor(time.Second*1))
 	suite.Equal(utils.DefaultKeyspaceGroupID, kg.ID)
-	suite.Len(kg.Members, utils.DefaultKeyspaceGroupReplicaCount)
+	// the allocNodesToAllKeyspaceGroups loop will run every 100ms.
+	testutil.Eventually(suite.Require(), func() bool {
+		return len(kg.Members) == utils.DefaultKeyspaceGroupReplicaCount
+	})
 	for _, member := range kg.Members {
 		suite.Contains(nodes, member.Address)
 	}