Function TestPrebuildScheduling

coderd/prebuilds/preset_snapshot_test.go:1044–1169

(t *testing.T)

}

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

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

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			templateID := uuid.New()
			templateVersionID := uuid.New()
			presetOpts1 := options{
				templateID:          templateID,
				templateVersionID:   templateVersionID,
				presetID:            uuid.New(),
				presetName:          "my-preset-1",
				prebuiltWorkspaceID: uuid.New(),
				workspaceName:       "prebuilds1",
			}
			presetOpts2 := options{
				templateID:          templateID,
				templateVersionID:   templateVersionID,
				presetID:            uuid.New(),