80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package repository_test
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
managedaccess "billing/internal/access"
|
|
"billing/internal/repository"
|
|
)
|
|
|
|
func TestBusinessEventsPersistKeyQuotaAndAutomaticResetActions(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "usage.db")
|
|
store, err := repository.OpenSQLiteUsage(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := store.BootstrapManagedKey(context.Background(), "default", "000000"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
now := time.Date(2026, 8, 19, 1, 0, 0, 0, time.UTC)
|
|
next := now.Add(time.Hour)
|
|
if _, err := store.UpdateBilling(context.Background(), "key_default", managedaccess.BillingSettings{
|
|
QuotaMicros: 20_000_000, ResetPeriod: managedaccess.ResetDaily, NextResetAt: &next, MaxConcurrency: 4,
|
|
}, now); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := store.BillingState(context.Background(), "key_default", next.Add(time.Minute)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := store.ResetBilling(context.Background(), "key_default", next.Add(2*time.Minute)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
events, err := store.ListBusinessEvents(context.Background(), 100)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
joined := businessEventText(events)
|
|
for _, expected := range []string{
|
|
"系统初始化用户 `default` 的 Key",
|
|
"管理员调整用户 `default` 的额度：$0 → $20",
|
|
"系统执行用户 `default` 的额度自然重置",
|
|
"管理员手动重置用户 `default` 的额度",
|
|
} {
|
|
if !strings.Contains(joined, expected) {
|
|
t.Fatalf("missing %q in %s", expected, joined)
|
|
}
|
|
}
|
|
for _, event := range events {
|
|
if event.Status != repository.BusinessEventSucceeded {
|
|
t.Fatalf("unexpected status: %+v", event)
|
|
}
|
|
}
|
|
if err := store.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
reopened, err := repository.OpenSQLiteUsage(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer reopened.Close()
|
|
persisted, err := reopened.ListBusinessEvents(context.Background(), 100)
|
|
if err != nil || len(persisted) != len(events) {
|
|
t.Fatalf("persisted events=%d want=%d err=%v", len(persisted), len(events), err)
|
|
}
|
|
}
|
|
|
|
func businessEventText(events []repository.BusinessEvent) string {
|
|
values := make([]string, 0, len(events))
|
|
for _, event := range events {
|
|
values = append(values, event.Event)
|
|
}
|
|
return strings.Join(values, "\n")
|
|
}
|