github/prysm
1
1
Fork 0
mirror of https://github.com/OffchainLabs/prysm.git synced 2026-01-09 15:37:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

Replace context.Background with testing.TB.Context where possible (#15416)

Browse Source 
* Replace context.Background with testing.TB.Context where possible

* Fix failing tests
This commit is contained in:
Preston Van Loon
2025-06-16 17:09:18 -05:00
committed by GitHub
parent 6a13ba9125
commit 62fec4d1f3
409 changed files with 3175 additions and 3456 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
beacon-chain/execution/engine_client_test.go
View File

@@ -66,7 +66,7 @@ func TestClient_IPC(t *testing.T) {
defer rpcClient.Close()
srv := &Service{}
srv.rpcClient = rpcClient
ctx := context.Background()
ctx := t.Context()
fix := fixtures()
params.SetupTestConfigCleanup(t)
@@ -159,7 +159,7 @@ func TestClient_IPC(t *testing.T) {
}
func TestClient_HTTP(t *testing.T) {
ctx := context.Background()
ctx := t.Context()
fix := fixtures()
params.SetupTestConfigCleanup(t)
@@ -954,7 +954,7 @@ func TestClient_HTTP(t *testing.T) {
}
func TestReconstructFullBellatrixBlock(t *testing.T) {
ctx := context.Background()
ctx := t.Context()
t.Run("nil block", func(t *testing.T) {
service := &Service{}
@@ -1046,7 +1046,7 @@ func TestReconstructFullBellatrixBlock(t *testing.T) {
}
func TestReconstructFullBellatrixBlockBatch(t *testing.T) {
ctx := context.Background()
ctx := t.Context()
t.Run("nil block", func(t *testing.T) {
service := &Service{}
@@ -1374,7 +1374,7 @@ func TestServer_getPowBlockHashAtTerminalTotalDifficulty(t *testing.T) {
ExecutionBlock: tt.currentPowBlock,
BlockByHashMap: m,
}
b, e, err := client.GetTerminalBlockHash(context.Background(), 1)
b, e, err := client.GetTerminalBlockHash(t.Context(), 1)
if tt.errString != "" {
require.ErrorContains(t, tt.errString, err)
} else {
@@ -1908,7 +1908,7 @@ func TestHeaderByHash_NotFound(t *testing.T) {
srv := &Service{}
srv.rpcClient = RPCClientBad{}
_, err := srv.HeaderByHash(context.Background(), [32]byte{})
_, err := srv.HeaderByHash(t.Context(), [32]byte{})
assert.Equal(t, ethereum.NotFound, err)
}
@@ -1916,7 +1916,7 @@ func TestHeaderByNumber_NotFound(t *testing.T) {
srv := &Service{}
srv.rpcClient = RPCClientBad{}
_, err := srv.HeaderByNumber(context.Background(), big.NewInt(100))
_, err := srv.HeaderByNumber(t.Context(), big.NewInt(100))
assert.Equal(t, ethereum.NotFound, err)
}
@@ -2300,7 +2300,7 @@ func newPayloadV4Setup(t *testing.T, status *pb.PayloadStatus, payload *pb.Execu
func TestReconstructBlindedBlockBatch(t *testing.T) {
t.Run("empty response works", func(t *testing.T) {
ctx := context.Background()
ctx := t.Context()
cli, srv := newMockEngine(t)
srv.registerDefault(func(*jsonrpcMessage, http.ResponseWriter, *http.Request) {
@@ -2311,7 +2311,7 @@ func TestReconstructBlindedBlockBatch(t *testing.T) {
require.Equal(t, 0, len(results))
})
t.Run("expected error for nil response", func(t *testing.T) {
ctx := context.Background()
ctx := t.Context()
slot, err := slots.EpochStart(params.BeaconConfig().DenebForkEpoch)
require.NoError(t, err)
blk, _ := util.GenerateTestDenebBlockWithSidecar(t, [32]byte{}, slot, 0)
@@ -2345,7 +2345,7 @@ func Test_ExchangeCapabilities(t *testing.T) {
err := json.NewEncoder(w).Encode(resp)
require.NoError(t, err)
}))
ctx := context.Background()
ctx := t.Context()
logHook := logTest.NewGlobal()
rpcClient, err := rpc.DialHTTP(srv.URL)
@@ -2378,7 +2378,7 @@ func Test_ExchangeCapabilities(t *testing.T) {
err := json.NewEncoder(w).Encode(resp)
require.NoError(t, err)
}))
ctx := context.Background()
ctx := t.Context()
rpcClient, err := rpc.DialHTTP(srv.URL)
require.NoError(t, err)
@@ -2413,7 +2413,7 @@ func TestReconstructBlobSidecars(t *testing.T) {
sb, err := blocks.NewSignedBeaconBlock(b)
require.NoError(t, err)
ctx := context.Background()
ctx := t.Context()
t.Run("all seen", func(t *testing.T) {
hi := mockSummary(t, []bool{true, true, true, true, true, true})
verifiedBlobs, err := client.ReconstructBlobSidecars(ctx, sb, r, hi)