Skip to content

Commit

Permalink
Fix the e2e request ID test (again)
Browse files Browse the repository at this point in the history
  • Loading branch information
hslatman committed Feb 28, 2024
1 parent b83b8aa commit 535e2a9
Showing 1 changed file with 16 additions and 15 deletions.
31 changes: 16 additions & 15 deletions test/e2e/requestid_test.go
Expand Up @@ -19,23 +19,22 @@ import (
"go.step.sm/crypto/pemutil"
)

// reserveAddress "reserves" a TCP address by opening a listener on a random
// port and immediately closing it. The address can then be assumed to be
// reservePort "reserves" a TCP port by opening a listener on a random
// port and immediately closing it. The port can then be assumed to be
// available for running a server on.
func reserveAddress(t *testing.T) string {
func reservePort(t *testing.T) (host, port string) {
t.Helper()
l, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
if l, err = net.Listen("tcp6", "[::1]:0"); err != nil {
require.NoError(t, err, "failed to listen on a port")
}
}
l, err := net.Listen("tcp", ":0")
require.NoError(t, err)

address := l.Addr().String()
err = l.Close()
require.NoError(t, err)

return address
host, port, err = net.SplitHostPort(address)
require.NoError(t, err)

return
}

func Test_reflectRequestID(t *testing.T) {
Expand All @@ -57,13 +56,13 @@ func Test_reflectRequestID(t *testing.T) {

// get a random address to listen on and connect to; currently no nicer way to get one before starting the server
// TODO(hs): find/implement a nicer way to expose the CA URL, similar to how e.g. httptest.Server exposes it?
address := reserveAddress(t)
host, port := reservePort(t)

cfg := &config.Config{
Root: []string{rootFilepath},
IntermediateCert: intermediateCertFilepath,
IntermediateKey: intermediateKeyFilepath,
Address: address, // reuse the address that was just "reserved"
Address: net.JoinHostPort(host, port), // reuse the address that was just "reserved"
DNSNames: []string{"127.0.0.1", "[::1]", "localhost"},
AuthorityConfig: &config.AuthConfig{
AuthorityID: "stepca-test",
Expand All @@ -76,7 +75,7 @@ func Test_reflectRequestID(t *testing.T) {

// instantiate a client for the CA running at the random address
caClient, err := ca.NewClient(
fmt.Sprintf("https://%s", address),
fmt.Sprintf("https://localhost:%s", port),
ca.WithRootFile(rootFilepath),
)
require.NoError(t, err)
Expand All @@ -93,8 +92,10 @@ func Test_reflectRequestID(t *testing.T) {
// require OK health response as the baseline
ctx := context.Background()
healthResponse, err := caClient.HealthWithContext(ctx)
assert.NoError(t, err)
assert.Equal(t, "ok", healthResponse.Status)
require.NoError(t, err)
if assert.NotNil(t, healthResponse) {
require.Equal(t, "ok", healthResponse.Status)
}

// expect an error when retrieving an invalid root
rootResponse, err := caClient.RootWithContext(ctx, "invalid")
Expand Down

0 comments on commit 535e2a9

Please sign in to comment.