feat: 后台增加限流配置页面 #2655

TencentBlueKing · Oct 24, 2024 · 3bff4fd · 3bff4fd
1 parent c10696c
commit 3bff4fd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...st/kotlin/com/tencent/bkrepo/common/ratelimiter/service/AbstractRateLimiterServiceTest.kt b/...st/kotlin/com/tencent/bkrepo/common/ratelimiter/service/AbstractRateLimiterServiceTest.kt
@@ -43,6 +43,7 @@ import com.tencent.bkrepo.common.ratelimiter.enums.LimitDimension
 import com.tencent.bkrepo.common.ratelimiter.enums.WorkScope
 import com.tencent.bkrepo.common.ratelimiter.exception.InvalidResourceException
 import com.tencent.bkrepo.common.ratelimiter.metrics.RateLimiterMetrics
+import com.tencent.bkrepo.common.ratelimiter.repository.RateLimitRepository
 import com.tencent.bkrepo.common.ratelimiter.rule.common.ResInfo
 import com.tencent.bkrepo.common.ratelimiter.rule.common.ResourceLimit
 import com.tencent.bkrepo.common.ratelimiter.service.user.RateLimiterConfigService
@@ -68,13 +69,17 @@ open class AbstractRateLimiterServiceTest : DistributedTest() {
     @MockBean
     private lateinit var meterRegistry: MeterRegistry
 
+    @MockBean
+    private lateinit var rateLimitRepository: RateLimitRepository
+
     fun init() {
         request = MockHttpServletRequest()
         val scheduler = ThreadPoolTaskScheduler()
         scheduler.initialize()
         taskScheduler = scheduler
         rateLimiterProperties = RateLimiterProperties()
         rateLimiterMetrics = RateLimiterMetrics(meterRegistry)
+        rateLimiterConfigService = RateLimiterConfigService(rateLimitRepository)
     }