Function TestUsageSummaryIncrementAndGet

internal/usage/usage_test.go:29–67

-- Store tests (DB required) --

(t *testing.T)

// -- Store tests (DB required) --

func TestUsageSummaryIncrementAndGet(t *testing.T) {
	pool := testutil.TestDB(t)
	store := usage.NewStore(pool)
	idStore := identity.NewStore(pool)

	ctx := context.Background()
	bucketDate := "2026-03-29"

	// Create a real user to satisfy FK constraint
	user, err := idStore.CreateOrGetUser(ctx, "billing-test@example.com", "Billing Test", "google-billing-test")
	if err != nil {
		t.Fatalf("CreateOrGetUser: %v", err)
	}

	// Increment inbound
	if err := store.IncrementUsageSummary(ctx, user.ID, bucketDate, "inbound"); err != nil {
		t.Fatalf("IncrementUsageSummary inbound: %v", err)
	}
	// Increment outbound twice
	for i := 0; i < 2; i++ {
		if err := store.IncrementUsageSummary(ctx, user.ID, bucketDate, "outbound"); err != nil {
			t.Fatalf("IncrementUsageSummary outbound: %v", err)
		}
	}

	sum, err := store.GetUsageSummary(ctx, user.ID, bucketDate)
	if err != nil {
		t.Fatalf("GetUsageSummary: %v", err)
	}
	if sum.InboundCount != 1 {
		t.Errorf("InboundCount = %d, want 1", sum.InboundCount)
	}
	if sum.OutboundCount != 2 {
		t.Errorf("OutboundCount = %d, want 2", sum.OutboundCount)
	}
	if sum.TotalCount != 3 {
		t.Errorf("TotalCount = %d, want 3", sum.TotalCount)
	}
}

func TestLiveUsageTracker_AlwaysAllows(t *testing.T) {
	pool := testutil.TestDB(t)