Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: Ryan Leung <[email protected]>
  • Loading branch information
rleungx committed Nov 4, 2024
1 parent 1524a05 commit 5f7e44a
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions tests/integrations/tso/client_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ import (
"github.com/tikv/pd/pkg/utils/tempurl"
"github.com/tikv/pd/pkg/utils/tsoutil"
"github.com/tikv/pd/server/apiv2/handlers"
"github.com/tikv/pd/server/config"
"github.com/tikv/pd/tests"
"github.com/tikv/pd/tests/integrations/mcs"
handlersutil "github.com/tikv/pd/tests/server/apiv2/handlers"
Expand Down Expand Up @@ -91,7 +92,9 @@ func (suite *tsoClientTestSuite) SetupSuite() {
if suite.legacy {
suite.cluster, err = tests.NewTestCluster(suite.ctx, serverCount)
} else {
suite.cluster, err = tests.NewTestAPICluster(suite.ctx, serverCount)
suite.cluster, err = tests.NewTestAPICluster(suite.ctx, serverCount, func(conf *config.Config, _ string) {
conf.MicroService.EnableTSODynamicSwitching = false
})
}
re.NoError(err)
err = suite.cluster.RunInitialServers()
Expand All @@ -104,11 +107,6 @@ func (suite *tsoClientTestSuite) SetupSuite() {
suite.keyspaceIDs = make([]uint32, 0)

if !suite.legacy {
opt := suite.pdLeaderServer.GetServer().GetPersistOptions()
cfg := opt.GetMicroServiceConfig()
cfg.EnableTSODynamicSwitching = false
opt.SetMicroServiceConfig(cfg)

suite.tsoCluster, err = tests.NewTestTSOCluster(suite.ctx, 3, suite.backendEndpoints)
re.NoError(err)

Expand Down

0 comments on commit 5f7e44a

Please sign in to comment.