diff --git a/fetcher/helpers/block_visitor_test.go b/fetcher/helpers/block_visitor_test.go index 2fcf58d03..300e3d09c 100644 --- a/fetcher/helpers/block_visitor_test.go +++ b/fetcher/helpers/block_visitor_test.go @@ -21,6 +21,8 @@ import ( "github.com/stretchr/testify/require" ) +var bg = context.Background() + func TestFetchGraphToBlocks(t *testing.T) { block3, node3, link3 := testutil.EncodeBlock(fluent.MustBuildMap(basicnode.Prototype__Map{}, 1, func(na fluent.MapAssembler) { na.AssembleEntry("three").AssignBool(true) @@ -49,13 +51,13 @@ func TestFetchGraphToBlocks(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block1) + err := hasBlock.Exchange.HasBlock(bg, block1) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block2) + err = hasBlock.Exchange.HasBlock(bg, block2) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block3) + err = hasBlock.Exchange.HasBlock(bg, block3) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block4) + err = hasBlock.Exchange.HasBlock(bg, block4) require.NoError(t, err) wantsBlock := peers[1] @@ -102,11 +104,11 @@ func TestFetchGraphToUniqueBlocks(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block1) + err := hasBlock.Exchange.HasBlock(bg, block1) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block2) + err = hasBlock.Exchange.HasBlock(bg, block2) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block3) + err = hasBlock.Exchange.HasBlock(bg, block3) require.NoError(t, err) wantsBlock := peers[1] diff --git a/fetcher/impl/blockservice/fetcher_test.go b/fetcher/impl/blockservice/fetcher_test.go index 42d31d436..33cd2ee53 100644 --- a/fetcher/impl/blockservice/fetcher_test.go +++ b/fetcher/impl/blockservice/fetcher_test.go @@ -27,6 +27,8 @@ import ( "github.com/stretchr/testify/require" ) +var bg = context.Background() + func TestFetchIPLDPrimeNode(t *testing.T) { block, node, _ := testutil.EncodeBlock(fluent.MustBuildMap(basicnode.Prototype__Map{}, 3, func(na fluent.MapAssembler) { na.AssembleEntry("foo").AssignBool(true) @@ -44,7 +46,7 @@ func TestFetchIPLDPrimeNode(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block) + err := hasBlock.Exchange.HasBlock(bg, block) require.NoError(t, err) wantsBlock := peers[1] @@ -90,13 +92,13 @@ func TestFetchIPLDGraph(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block1) + err := hasBlock.Exchange.HasBlock(bg, block1) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block2) + err = hasBlock.Exchange.HasBlock(bg, block2) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block3) + err = hasBlock.Exchange.HasBlock(bg, block3) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block4) + err = hasBlock.Exchange.HasBlock(bg, block4) require.NoError(t, err) wantsBlock := peers[1] @@ -150,7 +152,7 @@ func TestFetchIPLDPath(t *testing.T) { defer hasBlock.Exchange.Close() for _, blk := range []blocks.Block{block1, block2, block3, block4, block5} { - err := hasBlock.Exchange.HasBlock(blk) + err := hasBlock.Exchange.HasBlock(bg, blk) require.NoError(t, err) } @@ -212,13 +214,13 @@ func TestHelpers(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block1) + err := hasBlock.Exchange.HasBlock(bg, block1) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block2) + err = hasBlock.Exchange.HasBlock(bg, block2) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block3) + err = hasBlock.Exchange.HasBlock(bg, block3) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block4) + err = hasBlock.Exchange.HasBlock(bg, block4) require.NoError(t, err) wantsBlock := peers[1] @@ -329,11 +331,11 @@ func TestNodeReification(t *testing.T) { hasBlock := peers[0] defer hasBlock.Exchange.Close() - err := hasBlock.Exchange.HasBlock(block2) + err := hasBlock.Exchange.HasBlock(bg, block2) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block3) + err = hasBlock.Exchange.HasBlock(bg, block3) require.NoError(t, err) - err = hasBlock.Exchange.HasBlock(block4) + err = hasBlock.Exchange.HasBlock(bg, block4) require.NoError(t, err) wantsBlock := peers[1]