mirror of
https://github.com/asticode/go-astikit.git
synced 2026-04-22 23:07:17 +08:00
Quentin Renard
27 lines
409 B
Go
27 lines
409 B
Go
package astikit
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLimiter(t *testing.T) {
|
|
var l = NewLimiter()
|
|
defer l.Close()
|
|
l.Add("test", 2, time.Second)
|
|
b, ok := l.Bucket("test")
|
|
if !ok {
|
|
t.Error("no bucket found")
|
|
}
|
|
defer b.Close()
|
|
if !b.Inc() {
|
|
t.Errorf("got false, expected true")
|
|
}
|
|
if !b.Inc() {
|
|
t.Errorf("got false, expected true")
|
|
}
|
|
if b.Inc() {
|
|
t.Errorf("got true, expected false")
|
|
}
|
|
}
|