Function TestAppHealth_Timeout

agent/apphealth_test.go:164–230

(t *testing.T)

}

func TestAppHealth_Timeout(t *testing.T) {
	t.Parallel()
	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
	defer cancel()
	apps := []codersdk.WorkspaceApp{
		{
			ID:   uuid.UUID{2},
			Slug: "app2",
			Healthcheck: codersdk.Healthcheck{
				// URL: We don't set the URL for this test because the setup will
				// create a httptest server for us and set it for us.
				Interval:  1,
				Threshold: 1,
			},
			Health: codersdk.WorkspaceAppHealthInitializing,
		},
	}

	handlers := []http.Handler{
		http.HandlerFunc(func(_ http.ResponseWriter, r *http.Request) {
			// allow the request to time out
			<-r.Context().Done()
		}),
	}
	mClock := quartz.NewMock(t)
	start := mClock.Now()

	// for this test, it's easier to think in the number of milliseconds elapsed
	// since start.
	ms := func(n int) time.Time {
		return start.Add(time.Duration(n) * time.Millisecond)
	}
	healthcheckTrap := mClock.Trap().TickerFunc("healthcheck")
	defer healthcheckTrap.Close()
	reportTrap := mClock.Trap().TickerFunc("report")
	defer reportTrap.Close()
	timeoutTrap := mClock.Trap().AfterFunc("timeout")
	defer timeoutTrap.Close()

	fakeAPI, closeFn := setupAppReporter(ctx, t, apps, handlers, mClock)
	defer closeFn()
	healthcheckTrap.MustWait(ctx).MustRelease(ctx)
	// advance the clock 1ms before the report ticker starts, so that it's not
	// simultaneous with the checks.
	mClock.Set(ms(1)).MustWait(ctx)
	reportTrap.MustWait(ctx).MustRelease(ctx)

	w := mClock.Set(ms(1000)) // 1st check starts
	timeoutTrap.MustWait(ctx).MustRelease(ctx)
	mClock.Set(ms(1001)).MustWait(ctx) // report tick, no change
	mClock.Set(ms(1999))               // timeout pops
	w.MustWait(ctx)                    // 1st check finished
	w = mClock.Set(ms(2000))           // 2nd check starts
	timeoutTrap.MustWait(ctx).MustRelease(ctx)
	mClock.Set(ms(2001)).MustWait(ctx) // report tick, no change
	mClock.Set(ms(2999))               // timeout pops
	w.MustWait(ctx)                    // 2nd check finished
	// app is now unhealthy after 2 timeouts