forked from tikv/pd
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
This is an automated cherry-pick of tikv#8757
close tikv#8756 Signed-off-by: ti-chi-bot <[email protected]>
- Loading branch information
1 parent
10e4889
commit 9f97680
Showing
6 changed files
with
513 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,185 @@ | ||
| // Copyright 2024 TiKV Authors | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| package realcluster | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "os/exec" | ||
| "path/filepath" | ||
| "strings" | ||
| "testing" | ||
| "time" | ||
|
|
||
| "github.com/pingcap/log" | ||
| "github.com/stretchr/testify/require" | ||
| "github.com/stretchr/testify/suite" | ||
| "go.uber.org/zap" | ||
| ) | ||
|
|
||
| type clusterSuite struct { | ||
| suite.Suite | ||
|
|
||
| clusterCnt int | ||
| suiteName string | ||
| ms bool | ||
| } | ||
|
|
||
| var ( | ||
| playgroundLogDir = "/tmp/real_cluster/playground" | ||
| tiupBin = os.Getenv("HOME") + "/.tiup/bin/tiup" | ||
| ) | ||
|
|
||
| // SetupSuite will run before the tests in the suite are run. | ||
| func (s *clusterSuite) SetupSuite() { | ||
| t := s.T() | ||
|
|
||
| // Clean the data dir. It is the default data dir of TiUP. | ||
| dataDir := filepath.Join(os.Getenv("HOME"), ".tiup", "data", "pd_real_cluster_test_"+s.suiteName+"_*") | ||
| matches, err := filepath.Glob(dataDir) | ||
| require.NoError(t, err) | ||
|
|
||
| for _, match := range matches { | ||
| require.NoError(t, runCommand("rm", "-rf", match)) | ||
| } | ||
| s.startCluster(t) | ||
| t.Cleanup(func() { | ||
| s.stopCluster(t) | ||
| }) | ||
| } | ||
|
|
||
| // TearDownSuite will run after all the tests in the suite have been run. | ||
| func (s *clusterSuite) TearDownSuite() { | ||
| // Even if the cluster deployment fails, we still need to destroy the cluster. | ||
| // If the cluster does not fail to deploy, the cluster will be destroyed in | ||
| // the cleanup function. And these code will not work. | ||
| s.clusterCnt++ | ||
| s.stopCluster(s.T()) | ||
| } | ||
|
|
||
| func (s *clusterSuite) startCluster(t *testing.T) { | ||
| log.Info("start to deploy a cluster", zap.Bool("ms", s.ms)) | ||
|
|
||
| tag := s.tag() | ||
| deployTiupPlayground(t, tag, s.ms) | ||
| waitTiupReady(t, tag) | ||
| s.clusterCnt++ | ||
| } | ||
|
|
||
| func (s *clusterSuite) stopCluster(t *testing.T) { | ||
| s.clusterCnt-- | ||
|
|
||
| log.Info("start to destroy a cluster", zap.String("tag", s.tag()), zap.Bool("ms", s.ms)) | ||
| destroy(t, s.tag()) | ||
| time.Sleep(5 * time.Second) | ||
| } | ||
|
|
||
| func (s *clusterSuite) tag() string { | ||
| return fmt.Sprintf("pd_real_cluster_test_%s_%d", s.suiteName, s.clusterCnt) | ||
| } | ||
|
|
||
| func (s *clusterSuite) restart() { | ||
| tag := s.tag() | ||
| log.Info("start to restart", zap.String("tag", tag)) | ||
| s.stopCluster(s.T()) | ||
| s.startCluster(s.T()) | ||
| log.Info("TiUP restart success") | ||
| } | ||
|
|
||
| func destroy(t *testing.T, tag string) { | ||
| cmdStr := fmt.Sprintf("ps -ef | grep %s | awk '{print $2}'", tag) | ||
| cmd := exec.Command("sh", "-c", cmdStr) | ||
| bytes, err := cmd.Output() | ||
| require.NoError(t, err) | ||
| pids := string(bytes) | ||
| pidArr := strings.Split(pids, "\n") | ||
| for _, pid := range pidArr { | ||
| // nolint:errcheck | ||
| runCommand("sh", "-c", "kill -9 "+pid) | ||
| } | ||
| log.Info("destroy success", zap.String("tag", tag)) | ||
| } | ||
|
|
||
| func deployTiupPlayground(t *testing.T, tag string, ms bool) { | ||
| curPath, err := os.Getwd() | ||
| require.NoError(t, err) | ||
| require.NoError(t, os.Chdir("../../..")) | ||
|
|
||
| if !fileExists("third_bin") || !fileExists("third_bin/tikv-server") || !fileExists("third_bin/tidb-server") || !fileExists("third_bin/tiflash") { | ||
| log.Info("downloading binaries...") | ||
| log.Info("this may take a few minutes, you can also download them manually and put them in the bin directory.") | ||
| require.NoError(t, runCommand("sh", | ||
| "./tests/integrations/realcluster/download_integration_test_binaries.sh")) | ||
| } | ||
| if !fileExists("bin") || !fileExists("bin/pd-server") { | ||
| log.Info("complie pd binaries...") | ||
| require.NoError(t, runCommand("make", "pd-server")) | ||
| } | ||
|
|
||
| if !fileExists(playgroundLogDir) { | ||
| require.NoError(t, os.MkdirAll(playgroundLogDir, 0755)) | ||
| } | ||
|
|
||
| // nolint:errcheck | ||
| go func() { | ||
| if ms { | ||
| runCommand("sh", "-c", | ||
| tiupBin+` playground nightly --pd.mode ms --kv 3 --tiflash 1 --db 1 --pd 3 --tso 1 --scheduling 1 \ | ||
| --without-monitor --tag `+tag+` \ | ||
| --pd.binpath ./bin/pd-server \ | ||
| --kv.binpath ./third_bin/tikv-server \ | ||
| --db.binpath ./third_bin/tidb-server \ | ||
| --tiflash.binpath ./third_bin/tiflash \ | ||
| --tso.binpath ./bin/pd-server \ | ||
| --scheduling.binpath ./bin/pd-server \ | ||
| --pd.config ./tests/integrations/realcluster/pd.toml \ | ||
| > `+filepath.Join(playgroundLogDir, tag+".log")+` 2>&1 & `) | ||
| } else { | ||
| runCommand("sh", "-c", | ||
| tiupBin+` playground nightly --kv 3 --tiflash 1 --db 1 --pd 3 \ | ||
| --without-monitor --tag `+tag+` \ | ||
| --pd.binpath ./bin/pd-server \ | ||
| --kv.binpath ./third_bin/tikv-server \ | ||
| --db.binpath ./third_bin/tidb-server \ | ||
| --tiflash.binpath ./third_bin/tiflash \ | ||
| --pd.config ./tests/integrations/realcluster/pd.toml \ | ||
| > `+filepath.Join(playgroundLogDir, tag+".log")+` 2>&1 & `) | ||
| } | ||
| }() | ||
|
|
||
| // Avoid to change the dir before execute `tiup playground`. | ||
| time.Sleep(10 * time.Second) | ||
| require.NoError(t, os.Chdir(curPath)) | ||
| } | ||
|
|
||
| func waitTiupReady(t *testing.T, tag string) { | ||
| const ( | ||
| interval = 5 | ||
| maxTimes = 20 | ||
| ) | ||
| log.Info("start to wait TiUP ready", zap.String("tag", tag)) | ||
| for i := range maxTimes { | ||
| err := runCommand(tiupBin, "playground", "display", "--tag", tag) | ||
| if err == nil { | ||
| log.Info("TiUP is ready", zap.String("tag", tag)) | ||
| return | ||
| } | ||
|
|
||
| log.Info("TiUP is not ready, will retry", zap.Int("retry times", i), | ||
| zap.String("tag", tag), zap.Error(err)) | ||
| time.Sleep(time.Duration(interval) * time.Second) | ||
| } | ||
| require.Failf(t, "TiUP is not ready", "tag: %s", tag) | ||
| } |
Oops, something went wrong.