Function TestLiveUsageTracker_SystemClassNotMetered

internal/usage/usage_test.go:101–159

TestLiveUsageTracker_SystemClassNotMetered is the load-bearing assertion for the metering gate: a system-class account (synthetic probe traffic) writes ZERO usage_events and ZERO usage_summaries, so it never accrues quota. A standard account in the same test writes both (regression guard that the ga

(t *testing.T)

// standard account in the same test writes both (regression guard that the gate
// only short-circuits non-standard classes).
func TestLiveUsageTracker_SystemClassNotMetered(t *testing.T) {
	pool := testutil.TestDB(t)
	store := usage.NewStore(pool)
	idStore := identity.NewStore(pool)
	tracker := usage.NewUsageTracker(store)
	ctx := context.Background()

	countEvents := func(userID string) int {
		var n int
		if err := pool.QueryRow(ctx,
			`SELECT COUNT(*) FROM usage_events WHERE user_id = $1`, userID).Scan(&n); err != nil {
			t.Fatalf("count usage_events: %v", err)
		}
		return n
	}

	// system-class account: gate short-circuits, no rows written.
	sysUser, err := idStore.CreateOrGetUser(ctx, "probe-system@example.com", "Probe System", "google-probe-system")
	if err != nil {
		t.Fatalf("CreateOrGetUser system: %v", err)
	}
	if _, err := pool.Exec(ctx, `UPDATE users SET account_class = 'system' WHERE id = $1`, sysUser.ID); err != nil {
		t.Fatalf("set system class: %v", err)
	}

	for _, dir := range []string{"inbound", "outbound"} {
		allowed, err := tracker.RecordAndCheck(ctx, sysUser.ID, "agent-sys", "probe.example.com", dir)
		if err != nil || !allowed {
			t.Errorf("system %s RecordAndCheck: allowed=%v err=%v", dir, allowed, err)
		}
	}
	if n := countEvents(sysUser.ID); n != 0 {
		t.Errorf("system account wrote %d usage_events, want 0", n)
	}
	if _, err := store.GetUsageSummary(ctx, sysUser.ID, usage.CurrentDate()); !errors.Is(err, pgx.ErrNoRows) {
		t.Errorf("system account usage_summaries lookup err = %v, want ErrNoRows (no summary written)", err)
	}

	// standard-class account (regression): gate does NOT short-circuit.
	stdUser, err := idStore.CreateOrGetUser(ctx, "probe-standard@example.com", "Probe Standard", "google-probe-standard")
	if err != nil {
		t.Fatalf("CreateOrGetUser standard: %v", err)
	}
	for _, dir := range []string{"inbound", "outbound"} {
		if _, err := tracker.RecordAndCheck(ctx, stdUser.ID, "agent-std", "real.example.com", dir); err != nil {
			t.Fatalf("standard %s RecordAndCheck: %v", dir, err)
		}
	}
	if n := countEvents(stdUser.ID); n != 2 {
		t.Errorf("standard account wrote %d usage_events, want 2", n)
	}
	sum, err := store.GetUsageSummary(ctx, stdUser.ID, usage.CurrentDate())
	if err != nil {
		t.Fatalf("standard GetUsageSummary: %v", err)
	}
	if sum.TotalCount != 2 {
		t.Errorf("standard account summary TotalCount = %d, want 2", sum.TotalCount)
	}