Function TestScheduleOverride

cli/schedule_test.go:334–386

nolint:paralleltest // t.Setenv

(t *testing.T)

//nolint:paralleltest // t.Setenv
func TestScheduleOverride(t *testing.T) {
	tests := []struct {
		command string
	}{
		{command: "extend"},
		// test for backwards compatibility
		{command: "override-stop"},
	}

	for _, tt := range tests {
		t.Run(tt.command, func(t *testing.T) {
			// Given
			// Set timezone to Asia/Kolkata to surface any timezone-related bugs.
			t.Setenv("TZ", "Asia/Kolkata")
			loc, err := tz.TimezoneIANA()
			require.NoError(t, err)
			require.Equal(t, "Asia/Kolkata", loc.String())
			sched, err := cron.Weekly("CRON_TZ=Europe/Dublin 30 7 * * Mon-Fri")
			require.NoError(t, err, "invalid schedule")
			ownerClient, _, _, ws := setupTestSchedule(t, sched)
			now := time.Now()

			// When: we override the stop schedule
			inv, root := clitest.New(t,
				"schedule", tt.command, ws[0].OwnerName+"/"+ws[0].Name, "10h",
			)

			clitest.SetupConfig(t, ownerClient, root)
			pty := ptytest.New(t).Attach(inv)
			require.NoError(t, inv.Run())

			// Fetch the workspace to get the actual deadline set by the
			// server. Computing our own expected deadline from a separately
			// captured time.Now() is racy: the CLI command calls time.Now()
			// internally, and with the Asia/Kolkata +05:30 offset the hour
			// boundary falls at :30 UTC minutes. A small delay between our
			// time.Now() and the command's is enough to land in different
			// hours.
			updated, err := ownerClient.Workspace(context.Background(), ws[0].ID)
			require.NoError(t, err)
			require.False(t, updated.LatestBuild.Deadline.IsZero(), "deadline should be set after extend")
			require.WithinDuration(t, now.Add(10*time.Hour), updated.LatestBuild.Deadline.Time, 5*time.Minute)
			expectedDeadline := updated.LatestBuild.Deadline.Time.In(loc).Format(time.RFC3339)

			// Then: the updated schedule should be shown
			pty.ExpectMatch(ws[0].OwnerName + "/" + ws[0].Name)
			pty.ExpectMatch(sched.Humanize())
			pty.ExpectMatch(sched.Next(now).In(loc).Format(time.RFC3339))
			pty.ExpectMatch("8h")
			pty.ExpectMatch(expectedDeadline)
		})
	}
}

//nolint:paralleltest // t.Setenv
func TestScheduleStart_TemplateAutostartRequirement(t *testing.T) {