Function TestReconciliationStats

enterprise/coderd/prebuilds/reconcile_test.go:2955–3007

(t *testing.T)

}

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

	// Setup
	clock := quartz.NewReal()
	db, ps := dbtestutil.NewDB(t)
	client, _, _ := coderdtest.NewWithAPI(t, &coderdtest.Options{
		Database: db,
		Pubsub:   ps,
		Clock:    clock,
	})
	fakeEnqueuer := newFakeEnqueuer()
	registry := prometheus.NewRegistry()
	cache := files.New(registry, &coderdtest.FakeAuthorizer{})
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: false}).Leveled(slog.LevelDebug)
	reconciler := prebuilds.NewStoreReconciler(
		db, ps, cache, codersdk.PrebuildsConfig{}, logger,
		clock,
		registry,
		fakeEnqueuer,
		newNoopUsageCheckerPtr(),
		noop.NewTracerProvider(),
		10,
		nil,
	)
	owner := coderdtest.CreateFirstUser(t, client)

	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
	defer cancel()

	// Create a template version with a preset
	dbfake.TemplateVersion(t, db).Seed(database.TemplateVersion{
		OrganizationID: owner.OrganizationID,
		CreatedBy:      owner.UserID,
	}).Preset(database.TemplateVersionPreset{
		DesiredInstances: sql.NullInt32{
			Int32: 1,
			Valid: true,
		},
	}).Do()

	// Verify that ReconcileAll tracks and returns elapsed time
	start := time.Now()
	stats, err := reconciler.ReconcileAll(ctx)
	actualElapsed := time.Since(start)
	require.NoError(t, err)
	require.Greater(t, stats.Elapsed, time.Duration(0))

	// Verify stats.Elapsed matches actual execution time
	require.InDelta(t, actualElapsed.Milliseconds(), stats.Elapsed.Milliseconds(), 100)
	// Verify reconciliation loop is not unexpectedly slow
	require.Less(t, stats.Elapsed, 5*time.Second)
}

func newNoopEnqueuer() *notifications.NoopEnqueuer {
	return notifications.NewNoopEnqueuer()