use t.context

Author: rohenaz

transaction/chaintracker/whatsonchain_test.go

@@ -3,7 +3,6 @@
 package chaintracker
 
 import (
-	"context"
 	"encoding/json"
 	"net/http"
 	"net/http/httptest"
@@ -54,7 +53,7 @@ func TestWhatsOnChainGetBlockHeaderSuccess(t *testing.T) {
 		client:  ts.Client(),
 	}
 
-	ctx := context.Background()
+	ctx := t.Context()
 	header, err := woc.GetBlockHeader(ctx, 100)
 	if err != nil {
 		t.Fatalf("expected no error, got %v", err)
@@ -83,7 +82,7 @@ func TestWhatsOnChainGetBlockHeaderNotFound(t *testing.T) {
 		client:  ts.Client(),
 	}
 
-	ctx := context.Background()
+	ctx := t.Context()
 	header, err := woc.GetBlockHeader(ctx, 100)
 	if err != nil {
 		t.Fatalf("expected no error, got %v", err)
@@ -108,7 +107,7 @@ func TestWhatsOnChainGetBlockHeaderErrorResponse(t *testing.T) {
 		client:  ts.Client(),
 	}
 
-	ctx := context.Background()
+	ctx := t.Context()
 	header, err := woc.GetBlockHeader(ctx, 100)
 	if err == nil {
 		t.Fatalf("expected error, got nil")
@@ -140,7 +139,7 @@ func TestWhatsOnChainIsValidRootForHeightSuccess(t *testing.T) {
 		client:  ts.Client(),
 	}
 
-	ctx := context.Background()
+	ctx := t.Context()
 	isValid, err := woc.IsValidRootForHeight(ctx, &merkleRootHash, 100)
 	if err != nil {
 		t.Fatalf("expected no error, got %v", err)
@@ -173,7 +172,7 @@ func TestWhatsOnChainIsValidRootForHeightInvalidRoot(t *testing.T) {
 		client:  ts.Client(),
 	}
 
-	ctx := context.Background()
+	ctx := t.Context()
 	isValid, err := woc.IsValidRootForHeight(ctx, &differentMerkleRootHash, 100)
 	if err != nil {
 		t.Fatalf("expected no error, got %v", err)