Function TestServer_Single

enterprise/cli/server_test.go:28–65

TestServer runs the enterprise server command and waits for /healthz to return "OK".

(t *testing.T)

// TestServer runs the enterprise server command
// and waits for /healthz to return "OK".
func TestServer_Single(t *testing.T) {
	t.Parallel()

	ctx, cancelFunc := context.WithTimeout(context.Background(), testutil.WaitLong)
	defer cancelFunc()

	var root cli.RootCmd
	cmd, err := root.Command(root.EnterpriseSubcommands())
	require.NoError(t, err)

	inv, cfg := clitest.NewWithCommand(t, cmd,
		"server",
		dbArg(t),
		"--http-address", ":0",
		"--access-url", "http://example.com",
	)
	clitest.Start(t, inv.WithContext(ctx))
	accessURL := waitAccessURL(t, cfg)
	client := &http.Client{}
	require.Eventually(t, func() bool {
		reqCtx := testutil.Context(t, testutil.IntervalMedium)
		req, err := http.NewRequestWithContext(reqCtx, http.MethodGet, accessURL.String()+"/healthz", nil)
		if err != nil {
			panic(err)
		}
		resp, err := client.Do(req)
		if err != nil {
			t.Log("/healthz not ready yet")
			return false
		}
		defer resp.Body.Close()
		bs, err := io.ReadAll(resp.Body)
		if err != nil {
			panic(err)
		}
		return assert.Equal(t, "OK", string(bs))
	}, testutil.WaitShort, testutil.IntervalMedium)
}

func waitAccessURL(t *testing.T, cfg config.Root) *url.URL {
	t.Helper()