From 70d2a0bd0857bbe8d4c5eba478afe8390af740ce Mon Sep 17 00:00:00 2001 From: stringsbuilder Date: Fri, 20 Mar 2026 21:24:45 +0800 Subject: [PATCH] refactor: replace context.WithCancel with t.Context Signed-off-by: stringsbuilder --- block/internal/da/async_block_retriever_test.go | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/block/internal/da/async_block_retriever_test.go b/block/internal/da/async_block_retriever_test.go index 496afe77c..d9e461398 100644 --- a/block/internal/da/async_block_retriever_test.go +++ b/block/internal/da/async_block_retriever_test.go @@ -73,8 +73,7 @@ func TestAsyncBlockRetriever_SubscriptionDrivenCaching(t *testing.T) { logger := zerolog.Nop() fetcher := NewAsyncBlockRetriever(client, logger, fiNs, 200*time.Millisecond, 100, 5) - ctx, cancel := context.WithCancel(context.Background()) - defer cancel() + ctx := t.Context() fetcher.Start(ctx) defer fetcher.Stop() @@ -128,8 +127,7 @@ func TestAsyncBlockRetriever_CatchupFillsGaps(t *testing.T) { logger := zerolog.Nop() fetcher := NewAsyncBlockRetriever(client, logger, fiNs, 100*time.Millisecond, 100, 10) - ctx, cancel := context.WithCancel(context.Background()) - defer cancel() + ctx := t.Context() fetcher.Start(ctx) defer fetcher.Stop() @@ -198,8 +196,7 @@ func TestAsyncBlockRetriever_StopGracefully(t *testing.T) { logger := zerolog.Nop() fetcher := NewAsyncBlockRetriever(client, logger, fiNs, 100*time.Millisecond, 100, 10) - ctx, cancel := context.WithCancel(context.Background()) - defer cancel() + ctx := t.Context() fetcher.Start(ctx) time.Sleep(100 * time.Millisecond) @@ -240,8 +237,7 @@ func TestAsyncBlockRetriever_ReconnectOnSubscriptionError(t *testing.T) { // Very short backoff so reconnect is fast in tests. fetcher := NewAsyncBlockRetriever(client, logger, fiNs, 50*time.Millisecond, 100, 5) - ctx, cancel := context.WithCancel(context.Background()) - defer cancel() + ctx := t.Context() fetcher.Start(ctx) defer fetcher.Stop()