Function TestCalculateDesiredInstances

coderd/prebuilds/preset_snapshot_test.go:1206–1526

(t *testing.T)

}

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

	mkPreset := func(instances int32, timezone string) database.GetTemplatePresetsWithPrebuildsRow {
		return database.GetTemplatePresetsWithPrebuildsRow{
			DesiredInstances: sql.NullInt32{
				Int32: instances,
				Valid: true,
			},
			SchedulingTimezone: timezone,
		}
	}
	mkSchedule := func(cronExpr string, instances int32) database.TemplateVersionPresetPrebuildSchedule {
		return database.TemplateVersionPresetPrebuildSchedule{
			CronExpression:   cronExpr,
			DesiredInstances: instances,
		}
	}
	mkSnapshot := func(preset database.GetTemplatePresetsWithPrebuildsRow, schedules ...database.TemplateVersionPresetPrebuildSchedule) prebuilds.PresetSnapshot {
		return prebuilds.NewPresetSnapshot(
			preset,
			schedules,
			nil,
			nil,
			nil,
			0,
			nil,
			false,
			quartz.NewMock(t),
			testutil.Logger(t),
		)
	}

	testCases := []struct {
		name                        string
		snapshot                    prebuilds.PresetSnapshot
		at                          time.Time
		expectedCalculatedInstances int32
	}{
		// "* 9-18 * * 1-5" should be interpreted as a continuous time range from 09:00:00 to 18:59:59, Monday through Friday
		{
			name: "Right before the start of the time range",
			snapshot: mkSnapshot(
				mkPreset(1, "UTC"),
				mkSchedule("* 9-18 * * 1-5", 3),
			),
			at:                          mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 8:59:59 UTC"),
			expectedCalculatedInstances: 1,
		},
		{
			name: "Start of the time range",
			snapshot: mkSnapshot(
				mkPreset(1, "UTC"),
				mkSchedule("* 9-18 * * 1-5", 3),
			),
			at:                          mustParseTime(t, time.RFC1123, "Mon, 02 Jun 2025 9:00:00 UTC"),
			expectedCalculatedInstances: 3,
		},