Function TestTemplateUpdateBuildDeadlines

enterprise/coderd/schedule/template_test.go:36–321

(t *testing.T)

)

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

	db, _ := dbtestutil.NewDB(t)

	var (
		quietUser = dbgen.User(t, db, database.User{
			Username: "quiet",
		})
		noQuietUser = dbgen.User(t, db, database.User{
			Username: "no-quiet",
		})
		file = dbgen.File(t, db, database.File{
			CreatedBy: quietUser.ID,
		})
		templateJob = dbgen.ProvisionerJob(t, db, nil, database.ProvisionerJob{
			FileID:      file.ID,
			InitiatorID: quietUser.ID,
			Tags: database.StringMap{
				"foo": "bar",
			},
		})
		templateVersion = dbgen.TemplateVersion(t, db, database.TemplateVersion{
			OrganizationID: templateJob.OrganizationID,
			CreatedBy:      quietUser.ID,
			JobID:          templateJob.ID,
		})
		organizationID = templateJob.OrganizationID
	)

	const userQuietHoursSchedule = "CRON_TZ=UTC 0 0 * * *" // midnight UTC
	ctx := testutil.Context(t, testutil.WaitLong)
	quietUser, err := db.UpdateUserQuietHoursSchedule(ctx, database.UpdateUserQuietHoursScheduleParams{
		ID:                 quietUser.ID,
		QuietHoursSchedule: userQuietHoursSchedule,
	})
	require.NoError(t, err)

	realNow := time.Now().UTC()
	nowY, nowM, nowD := realNow.Date()
	buildTime := time.Date(nowY, nowM, nowD, 12, 0, 0, 0, time.UTC)       // noon today UTC
	nextQuietHours := time.Date(nowY, nowM, nowD+1, 0, 0, 0, 0, time.UTC) // midnight tomorrow UTC

	defaultTTL := 8 * time.Hour

	cases := []struct {
		name string
		now  time.Time
		// Before:
		deadline    time.Time
		maxDeadline time.Time
		// After:
		newDeadline    time.Time
		newMaxDeadline time.Time
		// Config:
		noQuietHours bool
		// Note that ttl will not influence the new build at all unless it's 0
		// AND the build does not have a max deadline post recalculation.