Function TestStatsCollector_backlog

coderd/workspaceapps/stats_test.go:347–402

(t *testing.T)

}

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

	rollupWindow := time.Minute
	flush := make(chan chan<- struct{}, 1)

	start := dbtime.Now().Truncate(time.Minute).UTC()
	var now atomic.Pointer[time.Time]
	now.Store(&start)

	reporter := &fakeReporter{}
	collector := workspaceapps.NewStatsCollector(workspaceapps.StatsCollectorOptions{
		Reporter:       reporter,
		ReportInterval: time.Hour,
		RollupWindow:   rollupWindow,

		Flush: flush,
		Now:   func() time.Time { return *now.Load() },
	})

	reporter.setError(xerrors.New("some error"))

	// The first collected stat is "rolled up" and moved into the
	// backlog during the first flush. On the second flush nothing is
	// rolled up due to being unable to report the backlog.
	for i := 0; i < 2; i++ {
		collector.Collect(workspaceapps.StatsReport{
			SessionID:        uuid.New(),
			SessionStartedAt: start,
			SessionEndedAt:   start.Add(10 * time.Second),
			Requests:         1,
		})
		start = start.Add(time.Minute)
		now.Store(&start)

		flushDone := make(chan struct{}, 1)
		flush <- flushDone
		<-flushDone
	}

	// Flush was performed 2 times, 2 reports should have failed.
	wantErrors := 2
	assert.Equal(t, wantErrors, reporter.errors())
	assert.Empty(t, reporter.stats())

	reporter.setError(nil)

	// Flush again, this time the backlog should be reported in addition
	// to the second collected stat being rolled up and reported.
	flushDone := make(chan struct{}, 1)
	flush <- flushDone
	<-flushDone

	assert.Equal(t, wantErrors, reporter.errors())
	assert.Len(t, reporter.stats(), 2)
}

func TestStatsCollector_Close(t *testing.T) {