Function New

internal/testutils/proxyserver/proxyserver.go:106–134

New initializes and starts a proxy server, registers a cleanup to stop it, and returns a ProxyServer.

(t *testing.T, reqCheck func(*http.Request), waitForServerHello bool)

// New initializes and starts a proxy server, registers a cleanup to
// stop it, and returns a ProxyServer.
func New(t *testing.T, reqCheck func(*http.Request), waitForServerHello bool) *ProxyServer {
	t.Helper()
	pLis, err := testutils.LocalTCPListener()
	if err != nil {
		t.Fatalf("failed to listen: %v", err)
	}

	p := &ProxyServer{
		lis:       pLis,
		onRequest: reqCheck,
		Addr:      pLis.Addr().String(),
	}

	// Start the proxy server.
	go func() {
		for {
			in, err := p.lis.Accept()
			if err != nil {
				return
			}
			// p.handleRequest is not invoked in a goroutine because the test
			// proxy currently supports handling only one connection at a time.
			p.handleRequest(t, in, waitForServerHello)
		}
	}()
	t.Logf("Started proxy at: %q", pLis.Addr().String())
	t.Cleanup(p.stop)
	return p
}