remove use of context.Background()

berachain · Jan 19, 2025 · 04166e7 · 04166e7
1 parent 5dfa536
commit 04166e7
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/da/blob/processor.go b/da/blob/processor.go
@@ -21,6 +21,7 @@
 package blob
 
 import (
+	"context"
 	"time"
 
 	"github.com/berachain/beacon-kit/chain"
@@ -66,6 +67,7 @@ func NewProcessor(
 
 // VerifySidecars verifies the blobs and ensures they match the local state.
 func (sp *Processor) VerifySidecars(
+	ctx context.Context,
 	sidecars datypes.BlobSidecars,
 	blkHeader *ctypes.BeaconBlockHeader,
 	kzgCommitments eip4844.KZGCommitments[common.ExecutionHash],
@@ -81,6 +83,7 @@ func (sp *Processor) VerifySidecars(
 
 	// Verify the blobs and ensure they match the local state.
 	return sp.verifier.verifySidecars(
+		ctx,
 		sidecars,
 		blkHeader,
 		kzgCommitments,

diff --git a/da/blob/verifier.go b/da/blob/verifier.go
@@ -62,6 +62,7 @@ func newVerifier(
 // verifySidecars verifies the blobs for both inclusion as well
 // as the KZG proofs.
 func (bv *verifier) verifySidecars(
+	ctx context.Context,
 	sidecars datypes.BlobSidecars,
 	blkHeader *ctypes.BeaconBlockHeader,
 	kzgCommitments eip4844.KZGCommitments[common.ExecutionHash],
@@ -71,7 +72,7 @@ func (bv *verifier) verifySidecars(
 		bv.proofVerifier.GetImplementation(),
 	)
 
-	g, _ := errgroup.WithContext(context.Background())
+	g, _ := errgroup.WithContext(ctx)
 
 	// Create lookup table for each blob sidecar commitment.
 	blobSidecarCommitments := make(map[eip4844.KZGCommitment]struct{})

diff --git a/state-transition/core/core_test.go b/state-transition/core/core_test.go
@@ -149,6 +149,7 @@ func setupState(t *testing.T, cs chain.Spec) (
 	)
 
 	ctx := &transition.Context{
+		Context:                 context.Background(),
 		SkipPayloadVerification: true,
 		SkipValidateResult:      true,
 		ProposerAddress:         dummyProposerAddr,

diff --git a/state-transition/core/state_processor_payload.go b/state-transition/core/state_processor_payload.go
@@ -38,7 +38,7 @@ func (sp *StateProcessor[ContextT]) processExecutionPayload(
 		body    = blk.GetBody()
 		payload = body.GetExecutionPayload()
 		header  *ctypes.ExecutionPayloadHeader
-		g, gCtx = errgroup.WithContext(context.Background())
+		g, gCtx = errgroup.WithContext(ctx)
 	)
 
 	payloadTimestamp := payload.GetTimestamp().Unwrap()