Function TestTracker_FlushToDB_NewPeriod

coderd/boundaryusage/tracker_test.go:186–224

(t *testing.T)

}

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

	db, _ := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitShort)
	boundaryCtx := dbauthz.AsBoundaryUsageTracker(ctx)

	tracker := boundaryusage.NewTracker()
	replicaID := uuid.New()
	workspaceID := uuid.New()
	ownerID := uuid.New()

	tracker.Track(workspaceID, ownerID, 10, 5)

	// First flush.
	err := tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	// Simulate telemetry reset (new period).
	_, err = db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)

	// Track new data.
	workspace2 := uuid.New()
	owner2 := uuid.New()
	tracker.Track(workspace2, owner2, 3, 1)

	// Flushing again should detect new period and reset in-memory stats.
	err = tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	// The summary should only contain the new data after reset.
	summary, err := db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(1), summary.UniqueWorkspaces, "should only count new workspace")
	require.Equal(t, int64(1), summary.UniqueUsers, "should only count new user")
	require.Equal(t, int64(3), summary.AllowedRequests, "should only count new requests")
	require.Equal(t, int64(1), summary.DeniedRequests, "should only count new requests")
}

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