Add more test to verify primary path is used correctly

Signed-off-by: Bin Shi <[email protected]>

tests/integrations/mcs/tso/keyspace_group_manager_test.go

@@ -194,16 +194,24 @@ func (suite *tsoKeyspaceGroupManagerTestSuite) TestKeyspacesServedByNonDefaultKe
 						re.NotNil(am)
 
 						// Make sure every keyspace group is using the right timestamp path
-						// for loading/saving timestamp from/to etcd.
-						var timestampPath string
+						// for loading/saving timestamp from/to etcd and the right primary path
+						// for leader election.
+						var (
+							timestampPath string
+							primaryPath   string
+						)
 						clusterID := strconv.FormatUint(suite.pdLeaderServer.GetClusterID(), 10)
 						if param.keyspaceGroupID == mcsutils.DefaultKeyspaceGroupID {
 							timestampPath = fmt.Sprintf("/pd/%s/timestamp", clusterID)
+							primaryPath = fmt.Sprintf("/ms/%s/tso/00000/primary", clusterID)
 						} else {
 							timestampPath = fmt.Sprintf("/ms/%s/tso/%05d/gta/timestamp",
 								clusterID, param.keyspaceGroupID)
+							primaryPath = fmt.Sprintf("/ms/%s/tso/%s/election/%05d/primary",
+								clusterID, mcsutils.KeyspaceGroupsKey, param.keyspaceGroupID)
 						}
-						re.Equal(timestampPath, am.GetTimestampPath(""))
+						re.Equal(timestampPath, am.GetTimestampPath(tsopkg.GlobalDCLocation))
+						re.Equal(primaryPath, am.GetMember().GetLeaderPath())
 
 						served = true
 					}