Use t.Context() instead of context.Background() in clientidentity_test.go (#11738)

Author: iain-macdonald

enterprise/server/clientidentity/clientidentity_test.go

@@ -1,7 +1,6 @@
 package clientidentity_test
 
 import (
-	"context"
 	"testing"
 	"time"
 
@@ -38,7 +37,7 @@ func TestIdentity(t *testing.T) {
 	}, clientidentity.DefaultExpiration)
 	require.NoError(t, err)
 
-	ctx := metadata.NewIncomingContext(context.Background(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
+	ctx := metadata.NewIncomingContext(t.Context(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
 	ctx, err = sis.ValidateIncomingIdentity(ctx)
 	require.NoError(t, err)
 
@@ -63,7 +62,7 @@ func TestDuplicateHeaders(t *testing.T) {
 	headers := metadata.Pairs(
 		authutil.ClientIdentityHeaderName, headerValue,
 		authutil.ClientIdentityHeaderName, headerValue)
-	ctx := metadata.NewIncomingContext(context.Background(), headers)
+	ctx := metadata.NewIncomingContext(t.Context(), headers)
 	ctx, err = sis.ValidateIncomingIdentity(ctx)
 	require.NoError(t, err)
 
@@ -80,7 +79,7 @@ func TestMultipleHeaders(t *testing.T) {
 	headers := metadata.Pairs(
 		authutil.ClientIdentityHeaderName, "value1",
 		authutil.ClientIdentityHeaderName, "value2")
-	ctx := metadata.NewIncomingContext(context.Background(), headers)
+	ctx := metadata.NewIncomingContext(t.Context(), headers)
 	_, err := sis.ValidateIncomingIdentity(ctx)
 	require.Error(t, err)
 	require.True(t, status.IsPermissionDeniedError(err))
@@ -106,7 +105,7 @@ func TestStaleIdentity(t *testing.T) {
 
 	clock.Advance(clientidentity.DefaultExpiration + time.Second)
 
-	ctx := metadata.NewIncomingContext(context.Background(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
+	ctx := metadata.NewIncomingContext(t.Context(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
 	_, err = sis.ValidateIncomingIdentity(ctx)
 	require.Error(t, err)
 }
@@ -127,11 +126,11 @@ func TestRequired(t *testing.T) {
 
 	headers := metadata.Pairs(
 		authutil.ClientIdentityHeaderName, headerValue)
-	ctx := metadata.NewIncomingContext(context.Background(), headers)
+	ctx := metadata.NewIncomingContext(t.Context(), headers)
 	_, err = sis.ValidateIncomingIdentity(ctx)
 	require.NoError(t, err)
 
-	ctx = metadata.NewIncomingContext(context.Background(), nil)
+	ctx = metadata.NewIncomingContext(t.Context(), nil)
 	_, err = sis.ValidateIncomingIdentity(ctx)
 	require.Error(t, err)
 }
@@ -146,7 +145,7 @@ func TestClearIdentity(t *testing.T) {
 	}, clientidentity.DefaultExpiration)
 	require.NoError(t, err)
 
-	ctx := metadata.NewIncomingContext(context.Background(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
+	ctx := metadata.NewIncomingContext(t.Context(), metadata.Pairs(authutil.ClientIdentityHeaderName, headerValue))
 	ctx, err = sis.ValidateIncomingIdentity(ctx)
 	require.NoError(t, err)
 	_, err = sis.IdentityFromContext(ctx)
@@ -170,7 +169,7 @@ func TestOverridePropagated_DefaultTrue(t *testing.T) {
 	}, clientidentity.DefaultExpiration)
 	require.NoError(t, err)
 
-	ctx := metadata.NewOutgoingContext(context.Background(), metadata.Pairs(authutil.ClientIdentityHeaderName, existingHeader))
+	ctx := metadata.NewOutgoingContext(t.Context(), metadata.Pairs(authutil.ClientIdentityHeaderName, existingHeader))
 	ctx, err = sis.AddIdentityToContext(ctx)
 	require.NoError(t, err)
 
@@ -194,7 +193,7 @@ func TestOverridePropagated_False_PreservesExisting(t *testing.T) {
 	}, clientidentity.DefaultExpiration)
 	require.NoError(t, err)
 
-	ctx := metadata.NewOutgoingContext(context.Background(), metadata.Pairs(authutil.ClientIdentityHeaderName, existingHeader))
+	ctx := metadata.NewOutgoingContext(t.Context(), metadata.Pairs(authutil.ClientIdentityHeaderName, existingHeader))
 	ctx, err = sis.AddIdentityToContext(ctx)
 	require.NoError(t, err)
 
@@ -214,7 +213,7 @@ func TestOverridePropagated_False_AddsWhenMissing(t *testing.T) {
 	flags.Set(t, "app.client_identity.origin", "local-origin")
 
 	// No existing identity header in the outgoing context.
-	ctx := context.Background()
+	ctx := t.Context()
 	ctx, err := sis.AddIdentityToContext(ctx)
 	require.NoError(t, err)
 
@@ -228,7 +227,7 @@ func TestOverridePropagated_False_AddsWhenMissing(t *testing.T) {
 func BenchmarkAddIdentityToContext(b *testing.B) {
 	sis := newService(b, clockwork.NewRealClock())
 
-	ctx := context.Background()
+	ctx := b.Context()
 	b.ReportAllocs()
 	b.ResetTimer()
 	b.RunParallel(func(pb *testing.PB) {