fix rate limit ut (#24096)

Signed-off-by: N Wei Liu <wei.liu@zilliz.com>

fix rate limit ut (#24096)
Signed-off-by: N Wei Liu <wei.liu@zilliz.com>
43699154 · wei liu · GitHub · 41cb0947 · 43699154
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

internal/proxy/multi_rate_limiter_test.go internal/proxy/multi_rate_limiter_test.go +3 -3

未找到文件。
--- a/internal/proxy/multi_rate_limiter_test.go
+++ b/internal/proxy/multi_rate_limiter_test.go
@@ -73,9 +73,9 @@ func TestMultiRateLimiter(t *testing.T) {
 		bak := Params.QuotaConfig.QuotaAndLimitsEnabled
 		paramtable.Get().Save(Params.QuotaConfig.QuotaAndLimitsEnabled.Key, "true")
 		multiLimiter := NewMultiRateLimiter()
-		multiLimiter.collectionLimiters[1] = newRateLimiter()
-		multiLimiter.collectionLimiters[2] = newRateLimiter()
-		multiLimiter.collectionLimiters[3] = newRateLimiter()
+		multiLimiter.collectionLimiters[1] = newRateLimiter(false)
+		multiLimiter.collectionLimiters[2] = newRateLimiter(false)
+		multiLimiter.collectionLimiters[3] = newRateLimiter(false)
 		for _, rt := range internalpb.RateType_value {
 			if IsDDLRequest(internalpb.RateType(rt)) {
 				multiLimiter.globalDDLLimiter.limiters.Insert(internalpb.RateType(rt), ratelimitutil.NewLimiter(ratelimitutil.Limit(5), 1))