Function TestPrebuildScheduling

enterprise/coderd/prebuilds/reconcile_test.go:634–783

(t *testing.T)

}

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

	templateDeleted := false

	// The test includes 2 presets, each with 2 schedules.
	// It checks that the number of created prebuilds match expectations for various provided times,
	// based on the corresponding schedules.
	testCases := []struct {
		name string
		// now specifies the current time.
		now time.Time
		// expected prebuild counts for preset1 and preset2, respectively.
		expectedPrebuildCounts []int
	}{
		{
			name:                   "Before the 1st schedule",
			now:                    mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 01:00:00 UTC"),
			expectedPrebuildCounts: []int{1, 1},
		},
		{
			name:                   "1st schedule",
			now:                    mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 03:00:00 UTC"),
			expectedPrebuildCounts: []int{2, 1},
		},
		{
			name:                   "2nd schedule",
			now:                    mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 07:00:00 UTC"),
			expectedPrebuildCounts: []int{3, 1},
		},
		{
			name:                   "3rd schedule",
			now:                    mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 11:00:00 UTC"),
			expectedPrebuildCounts: []int{1, 4},
		},
		{
			name:                   "4th schedule",
			now:                    mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 15:00:00 UTC"),
			expectedPrebuildCounts: []int{1, 5},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			clock := quartz.NewMock(t)
			clock.Set(tc.now)
			ctx := testutil.Context(t, testutil.WaitShort)
			cfg := codersdk.PrebuildsConfig{}
			logger := slogtest.Make(
				t, &slogtest.Options{IgnoreErrors: true},
			).Leveled(slog.LevelDebug)
			db, pubSub := dbtestutil.NewDB(t)
			cache := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})
			controller := prebuilds.NewStoreReconciler(
				db, pubSub, cache, cfg, logger,
				clock,
				prometheus.NewRegistry(),