diff --git a/beacon-chain/core/electra/registry_updates_test.go b/beacon-chain/core/electra/registry_updates_test.go index 2ab626c4b9a2..932ade806e91 100644 --- a/beacon-chain/core/electra/registry_updates_test.go +++ b/beacon-chain/core/electra/registry_updates_test.go @@ -1,7 +1,6 @@ package electra_test import ( - "context" "testing" "github.com/OffchainLabs/prysm/v6/beacon-chain/core/electra" @@ -131,7 +130,7 @@ func TestProcessRegistryUpdates(t *testing.T) { for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { - err := electra.ProcessRegistryUpdates(context.TODO(), tt.state) + err := electra.ProcessRegistryUpdates(t.Context(), tt.state) require.NoError(t, err) if tt.check != nil { tt.check(t, tt.state) @@ -164,7 +163,7 @@ func Benchmark_ProcessRegistryUpdates_MassEjection(b *testing.B) { } b.StartTimer() - if err := electra.ProcessRegistryUpdates(context.TODO(), st); err != nil { + if err := electra.ProcessRegistryUpdates(b.Context(), st); err != nil { panic(err) } } diff --git a/changelog/VolodymyrBg_feature-use-test-context-in-electra-tests.md b/changelog/VolodymyrBg_feature-use-test-context-in-electra-tests.md new file mode 100644 index 000000000000..4dbabea20152 --- /dev/null +++ b/changelog/VolodymyrBg_feature-use-test-context-in-electra-tests.md @@ -0,0 +1,3 @@ +### Fixed + +- Use testing contexts in Electra registry updates tests