package calls

import (
	"testing"
	"time"
)

func TestBudgetSlidesItsWindow(t *testing.T) {
	b := newBudget(2, time.Minute)
	now := time.Date(2026, 9, 8, 10, 0, 0, 0, time.UTC)
	if !b.allow("k", now) || !b.allow("k", now) {
		t.Fatal("first two refused")
	}
	if b.allow("k", now) {
		t.Fatal("third allowed")
	}
	if !b.allow("other", now) {
		t.Fatal("other key refused")
	}
	if !b.allow("k", now.Add(61*time.Second)) {
		t.Fatal("refused after the window slid")
	}
}