github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-01-05 15:37:24 +00:00
Code Issues Projects Releases Wiki Activity

RateLimiter: change CanAccept() to TryAccept()

Browse Source
This commit is contained in:
Hongchao Deng
2015-11-23 16:36:23 -08:00
parent 367f0e03dd
commit 13d152a873
6 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/util/throttle_test.go
View File

@@ -24,28 +24,28 @@ import (
func TestBasicThrottle(t *testing.T) {
r := NewTokenBucketRateLimiter(1, 3)
for i := 0; i < 3; i++ {
if !r.CanAccept() {
if !r.TryAccept() {
t.Error("unexpected false accept")
}
}
if r.CanAccept() {
if r.TryAccept() {
t.Error("unexpected true accept")
}
}
func TestIncrementThrottle(t *testing.T) {
r := NewTokenBucketRateLimiter(1, 1)
if !r.CanAccept() {
if !r.TryAccept() {
t.Error("unexpected false accept")
}
if r.CanAccept() {
if r.TryAccept() {
t.Error("unexpected true accept")
}
// Allow to refill
time.Sleep(2 * time.Second)
if !r.CanAccept() {
if !r.TryAccept() {
t.Error("unexpected false accept")
}
}