From e1e2b188c6ba57f4dce5d2f4efafac321b9d9025 Mon Sep 17 00:00:00 2001
From: Ryan Leung <rleungx@gmail.com>
Date: Tue, 16 Apr 2024 17:51:32 +0800
Subject: [PATCH] make TestStores stable

Signed-off-by: Ryan Leung <rleungx@gmail.com>
---
 tests/integrations/mcs/scheduling/api_test.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/integrations/mcs/scheduling/api_test.go b/tests/integrations/mcs/scheduling/api_test.go
index f615e879e40..e9033e5016a 100644
--- a/tests/integrations/mcs/scheduling/api_test.go
+++ b/tests/integrations/mcs/scheduling/api_test.go
@@ -651,6 +651,9 @@ func (suite *apiTestSuite) checkStores(cluster *tests.TestCluster) {
 	tests.MustPutRegion(re, cluster, 3, 6, []byte("a"), []byte("b"))
 	for _, store := range stores {
 		tests.MustPutStore(re, cluster, store)
+		if store.GetId() == 6 {
+			cluster.GetLeaderServer().GetRaftCluster().GetBasicCluster().UpdateStoreStatus(6)
+		}
 	}
 	// Test /stores
 	apiServerAddr := cluster.GetLeaderServer().GetAddr()