add more check

Signed-off-by: lhy1024 <admin@liudos.us>
tikv · Jun 16, 2023 · dbe3a6f · dbe3a6f
1 parent 25c930a
commit dbe3a6f
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/tests/integrations/mcs/tso/keyspace_group_manager_test.go b/tests/integrations/mcs/tso/keyspace_group_manager_test.go
@@ -536,8 +536,7 @@ func TestTwiceSplitKeyspaceGroup(t *testing.T) {
 	testutil.Eventually(re, func() bool {
 		_, _, err = client_b.(pd.Client).GetTS(ctx)
 		re.NoError(err)
-		kg, err := kgm.GetKeyspaceGroupByID(mcsutils.DefaultKeyspaceGroupID)
-		re.NoError(err)
+		kg := handlersutil.MustLoadKeyspaceGroupByID(re, leaderServer, 0)
 		return !kg.IsSplitting()
 	})
 	client_b.(pd.Client).Close()
@@ -552,11 +551,21 @@ func TestTwiceSplitKeyspaceGroup(t *testing.T) {
 	testutil.Eventually(re, func() bool {
 		_, _, err = client_a.(pd.Client).GetTS(ctx)
 		re.NoError(err)
-		kg, err := kgm.GetKeyspaceGroupByID(mcsutils.DefaultKeyspaceGroupID)
-		re.NoError(err)
+		kg := handlersutil.MustLoadKeyspaceGroupByID(re, leaderServer, 0)
 		return !kg.IsSplitting()
 	})
 	client_a.(pd.Client).Close()
 
+	// Check the keyspace group 0 is split to 1 and 2.
+	kg0 := handlersutil.MustLoadKeyspaceGroupByID(re, leaderServer, 0)
+	kg1 := handlersutil.MustLoadKeyspaceGroupByID(re, leaderServer, 1)
+	kg2 := handlersutil.MustLoadKeyspaceGroupByID(re, leaderServer, 2)
+	re.Equal([]uint32{0}, kg0.Keyspaces)
+	re.Equal([]uint32{2}, kg1.Keyspaces)
+	re.Equal([]uint32{1}, kg2.Keyspaces)
+	re.False(kg0.IsSplitting())
+	re.False(kg1.IsSplitting())
+	re.False(kg2.IsSplitting())
+
 	re.NoError(failpoint.Disable("github.com/tikv/pd/pkg/keyspace/acceleratedAllocNodes"))
 }