27 lines
690 B
Go
27 lines
690 B
Go
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")
|
|
}
|
|
}
|