From dd8e53e38a8f4b9caf7dba0e4a57357a50d51f44 Mon Sep 17 00:00:00 2001 From: lhy1024 Date: Wed, 20 Mar 2024 17:15:54 +0800 Subject: [PATCH] tests/etcdutil: reduce TestRandomKillEtcd test time Signed-off-by: lhy1024 --- pkg/utils/etcdutil/etcdutil_test.go | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/pkg/utils/etcdutil/etcdutil_test.go b/pkg/utils/etcdutil/etcdutil_test.go index 479548902115..c704bbbf47f6 100644 --- a/pkg/utils/etcdutil/etcdutil_test.go +++ b/pkg/utils/etcdutil/etcdutil_test.go @@ -452,9 +452,26 @@ func (suite *loopWatcherTestSuite) TestLoadWithLimitChange() { re := suite.Require() re.NoError(failpoint.Enable("github.com/tikv/pd/pkg/utils/etcdutil/meetEtcdError", `return()`)) cache := make(map[string]struct{}) - for i := 0; i < int(maxLoadBatchSize)*2; i++ { - suite.put(re, fmt.Sprintf("TestLoadWithLimitChange%d", i), "") + + // create data + count := int(maxLoadBatchSize) * 2 + var wg sync.WaitGroup + tasks := make(chan int, count) + for w := 0; w < 16; w++ { + wg.Add(1) + go func() { + defer wg.Done() + for i := range tasks { + suite.put(re, fmt.Sprintf("TestLoadWithLimitChange%d", i), "") + } + }() + } + for i := 0; i < count; i++ { + tasks <- i } + close(tasks) + wg.Wait() + watcher := NewLoopWatcher( suite.ctx, &suite.wg,