package ai

import "testing"

func TestQuotaStatusRemaining(t *testing.T) {
	status := QuotaStatus{
		RequestsUsedToday: 40,
		RequestsLimit:     100,
		TokensUsedMonth:   100_000,
		TokensLimit:       500_000,
	}
	status.RequestsRemaining = status.RequestsLimit - status.RequestsUsedToday
	status.TokensRemaining = status.TokensLimit - status.TokensUsedMonth
	if status.RequestsRemaining != 60 {
		t.Fatalf("requests remaining = %d", status.RequestsRemaining)
	}
	if status.TokensRemaining != 400_000 {
		t.Fatalf("tokens remaining = %d", status.TokensRemaining)
	}
}

func TestErrQuotaExceeded(t *testing.T) {
	if ErrQuotaExceeded.Error() == "" {
		t.Fatal("expected error message")
	}
}