mirror of
https://github.com/lukaszraczylo/graphql-monitoring-proxy.git
synced 2026-11-04 08:04:10 +00:00
lukaszraczylo
ab63f924a9
The refill loop dropped the fractional part of each tick. At 15 tokens per second and 100 ms ticks, it added 10 tokens per second instead of 15. The clock now advances only by the time the added tokens pay for.
48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Pins theorems/Theorems/RetryRefill.lean: total_added(15 tps, 100ms ticks, 10s) = 150.
|
|
func TestRefillStep_KeepsFractionalTokens(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
tps float64
|
|
want int64
|
|
}{
|
|
{"15 tps", 15, 150},
|
|
{"25 tps", 25, 250},
|
|
{"5 tps (sub-token per tick)", 5, 50},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
tick := 100 * time.Millisecond
|
|
var last, total int64
|
|
now := int64(tick)
|
|
for i := 0; i < 100; i++ {
|
|
added, newLast := refillStep(tt.tps, last, now)
|
|
total += added
|
|
last = newLast
|
|
now += int64(tick)
|
|
}
|
|
assert.Equal(t, tt.want, total)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRefillStep_NoElapsedNoChange(t *testing.T) {
|
|
added, last := refillStep(10, 1000, 1000)
|
|
assert.Zero(t, added)
|
|
assert.Equal(t, int64(1000), last)
|
|
}
|
|
|
|
func TestRefillStep_ZeroRateNeverAdds(t *testing.T) {
|
|
added, last := refillStep(0, 0, int64(time.Hour))
|
|
assert.Zero(t, added)
|
|
assert.Zero(t, last)
|
|
}
|