c6c2c525a6
Also fix some tests that were broken by bumping golang and dependencies in the previous commits. Note that in addition to changes made to satisfy the linter which do not impact the behavior of the code, this commit also adds ReadHeaderTimeout to all usages of http.Server to satisfy the linter (and because it seemed like a good suggestion).
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
// Copyright 2020-2022 the Pinniped contributors. All Rights Reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package testutil
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.pinniped.dev/internal/crypto/ptls"
|
|
"go.pinniped.dev/internal/testutil/tlsserver"
|
|
)
|
|
|
|
// TLSTestServer starts a test server listening on a local port using a test CA. It returns the PEM CA bundle and the
|
|
// URL of the listening server. The lifetime of the server is bound to the provided *testing.T.
|
|
func TLSTestServer(t *testing.T, handler http.HandlerFunc) (caBundlePEM, url string) {
|
|
t.Helper()
|
|
|
|
server := tlsserver.TLSTestServer(t, handler, nil)
|
|
|
|
return string(tlsserver.TLSTestServerCA(server)), server.URL
|
|
}
|
|
|
|
func TLSTestServerWithCert(t *testing.T, handler http.HandlerFunc, certificate *tls.Certificate) (url string) {
|
|
t.Helper()
|
|
|
|
c := ptls.Default(nil) // mimic API server config
|
|
c.Certificates = []tls.Certificate{*certificate}
|
|
|
|
server := http.Server{
|
|
TLSConfig: c,
|
|
Handler: handler,
|
|
ReadHeaderTimeout: 10 * time.Second,
|
|
}
|
|
|
|
l, err := net.Listen("tcp", "127.0.0.1:0")
|
|
require.NoError(t, err)
|
|
|
|
serverShutdownChan := make(chan error)
|
|
go func() {
|
|
// Empty certFile and keyFile will use certs from Server.TLSConfig.
|
|
serverShutdownChan <- server.ServeTLS(l, "", "")
|
|
}()
|
|
|
|
t.Cleanup(func() {
|
|
_ = server.Close()
|
|
serveErr := <-serverShutdownChan
|
|
if !errors.Is(serveErr, http.ErrServerClosed) {
|
|
t.Log("Got an unexpected error while starting the fake http server!")
|
|
require.NoError(t, serveErr)
|
|
}
|
|
})
|
|
|
|
return l.Addr().String()
|
|
}
|