Function TestAppHealth_500

agent/apphealth_test.go:115–162

(t *testing.T)

}

func TestAppHealth_500(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(w http.ResponseWriter, r *http.Request) {
			httpapi.Write(r.Context(), w, http.StatusInternalServerError, nil)
		}),
	}

	mClock := quartz.NewMock(t)
	healthcheckTrap := mClock.Trap().TickerFunc("healthcheck")
	defer healthcheckTrap.Close()
	reportTrap := mClock.Trap().TickerFunc("report")
	defer reportTrap.Close()

	fakeAPI, closeFn := setupAppReporter(ctx, t, slices.Clone(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.Advance(time.Millisecond).MustWait(ctx)
	reportTrap.MustWait(ctx).MustRelease(ctx)

	mClock.Advance(999 * time.Millisecond).MustWait(ctx) // check gets triggered
	mClock.Advance(time.Millisecond).MustWait(ctx)       // report gets triggered, but unsent since we are at the threshold

	mClock.Advance(999 * time.Millisecond).MustWait(ctx) // 2nd check, crosses threshold
	mClock.Advance(time.Millisecond).MustWait(ctx)       // 2nd report, sends update

	update := testutil.TryReceive(ctx, t, fakeAPI.AppHealthCh())
	require.Len(t, update.GetUpdates(), 1)
	applyUpdate(t, apps, update)
	require.Equal(t, codersdk.WorkspaceAppHealthUnhealthy, apps[0].Health)
}

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