mirror of
https://github.com/coredns/coredns.git
synced 2025-08-06 22:37:03 +02:00
38 lines
735 B
Go
38 lines
735 B
Go
package freq
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestFreqUpdate(t *testing.T) {
|
|
now := time.Now().UTC()
|
|
f := New(now)
|
|
window := 1 * time.Minute
|
|
|
|
f.Update(window, time.Now().UTC())
|
|
f.Update(window, time.Now().UTC())
|
|
f.Update(window, time.Now().UTC())
|
|
hitsCheck(t, f, 3)
|
|
|
|
f.Reset(now, 0)
|
|
history := time.Now().UTC().Add(-3 * time.Minute)
|
|
f.Update(window, history)
|
|
hitsCheck(t, f, 1)
|
|
}
|
|
|
|
func TestReset(t *testing.T) {
|
|
f := New(time.Now().UTC())
|
|
f.Update(1*time.Minute, time.Now().UTC())
|
|
hitsCheck(t, f, 1)
|
|
f.Reset(time.Now().UTC(), 0)
|
|
hitsCheck(t, f, 0)
|
|
}
|
|
|
|
func hitsCheck(t *testing.T, f *Freq, expected int) {
|
|
t.Helper()
|
|
if x := f.Hits(); x != expected {
|
|
t.Fatalf("Expected hits to be %d, got %d", expected, x)
|
|
}
|
|
}
|