Function TestUpsertBoundaryUsageStats_Insert

coderd/boundaryusage/tracker_test.go:246–272

(t *testing.T)

}

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

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

	replicaID := uuid.New()

	// Set different values for delta vs cumulative to verify INSERT uses delta.
	newPeriod, err := db.UpsertBoundaryUsageStats(ctx, database.UpsertBoundaryUsageStatsParams{
		ReplicaID:             replicaID,
		UniqueWorkspacesDelta: 5,
		UniqueUsersDelta:      3,
		UniqueWorkspacesCount: 999, // should be ignored on INSERT
		UniqueUsersCount:      999, // should be ignored on INSERT
		AllowedRequests:       100,
		DeniedRequests:        10,
	})
	require.NoError(t, err)
	require.True(t, newPeriod, "should return true for insert")

	// Verify INSERT used the delta values, not cumulative.
	summary, err := db.GetAndResetBoundaryUsageSummary(ctx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(5), summary.UniqueWorkspaces)
	require.Equal(t, int64(3), summary.UniqueUsers)
}

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