pdutil: fix unstable test TestPauseSchedulersByKeyRange (#35949)

close pingcap/tidb#35947
This commit is contained in:
Yujie Xia
2022-07-05 16:13:01 +08:00
committed by GitHub
parent d8dbab23cc
commit ff7c1200ec

View File

@ -13,6 +13,7 @@ import (
"net/http/httptest"
"net/url"
"strings"
"sync"
"testing"
"time"
@ -239,11 +240,21 @@ func TestPauseSchedulersByKeyRange(t *testing.T) {
labelExpires := make(map[string]time.Time)
var (
mu sync.Mutex
deleted bool
)
httpSrv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
mu.Lock()
defer mu.Unlock()
if deleted {
return
}
if r.Method == http.MethodDelete {
ruleID := strings.TrimPrefix(r.URL.Path, "/"+regionLabelPrefix+"/")
print(ruleID)
delete(labelExpires, ruleID)
deleted = true
return
}
var labelRule LabelRule