Function TestUpsertBoundaryUsageStats_Update

coderd/boundaryusage/tracker_test.go:274–310

(t *testing.T)

}

func TestUpsertBoundaryUsageStats_Update(t *testing.T) {
	t.Parallel()

	db, _ := dbtestutil.NewDB(t)
	ctx := dbauthz.AsBoundaryUsageTracker(context.Background())

	replicaID := uuid.New()

	// First insert uses delta fields.
	_, err := db.UpsertBoundaryUsageStats(ctx, database.UpsertBoundaryUsageStatsParams{
		ReplicaID:             replicaID,
		UniqueWorkspacesDelta: 5,
		UniqueUsersDelta:      3,
		AllowedRequests:       100,
		DeniedRequests:        10,
	})
	require.NoError(t, err)

	// Second upsert (update). Set different delta vs cumulative to verify UPDATE uses cumulative.
	newPeriod, err := db.UpsertBoundaryUsageStats(ctx, database.UpsertBoundaryUsageStatsParams{
		ReplicaID:             replicaID,
		UniqueWorkspacesCount: 8, // cumulative, should be used
		UniqueUsersCount:      5, // cumulative, should be used
		AllowedRequests:       200,
		DeniedRequests:        20,
	})
	require.NoError(t, err)
	require.False(t, newPeriod, "should return false for update")

	// Verify UPDATE used cumulative values.
	summary, err := db.GetAndResetBoundaryUsageSummary(ctx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(8), summary.UniqueWorkspaces)
	require.Equal(t, int64(5), summary.UniqueUsers)
	require.Equal(t, int64(100+200), summary.AllowedRequests)
	require.Equal(t, int64(10+20), summary.DeniedRequests)
}

func TestGetAndResetBoundaryUsageSummary_MultipleReplicas(t *testing.T) {
	t.Parallel()