Fix lint errors

Signed-off-by: AkramBitar <akram@il.ibm.com>

Author: AkramBitar

token/services/ttx/collectendorsements_test.go

@@ -7,7 +7,6 @@ SPDX-License-Identifier: Apache-2.0
 package ttx_test
 
 import (
-	"context"
 	"testing"
 
 	"github.com/hyperledger-labs/fabric-smart-client/pkg/utils/errors"
@@ -52,7 +51,7 @@ func TestCleanupExternalWallets_Success(t *testing.T) {
 	// Create a minimal view and context
 	view := &ttx.CollectEndorsementsView{}
 	ctx := &mock2.Context{}
-	ctx.ContextReturns(context.Background())
+	ctx.ContextReturns(t.Context())
 
 	// Call the cleanup method
 	view.CleanupExternalWallets(ctx, externalWallets)
@@ -74,7 +73,7 @@ func TestCleanupExternalWallets_MultipleWallets(t *testing.T) {
 
 	view := &ttx.CollectEndorsementsView{}
 	ctx := &mock2.Context{}
-	ctx.ContextReturns(context.Background())
+	ctx.ContextReturns(t.Context())
 
 	view.CleanupExternalWallets(ctx, externalWallets)
 
@@ -105,7 +104,7 @@ func TestCleanupExternalWallets_DoneError(t *testing.T) {
 
 	view := &ttx.CollectEndorsementsView{}
 	ctx := &mock2.Context{}
-	ctx.ContextReturns(context.Background())
+	ctx.ContextReturns(t.Context())
 
 	// Should not panic even if Done() returns errors
 	view.CleanupExternalWallets(ctx, externalWallets)
@@ -121,7 +120,7 @@ func TestCleanupExternalWallets_EmptyMap(t *testing.T) {
 
 	view := &ttx.CollectEndorsementsView{}
 	ctx := &mock2.Context{}
-	ctx.ContextReturns(context.Background())
+	ctx.ContextReturns(t.Context())
 
 	// Should not panic with empty map
 	view.CleanupExternalWallets(ctx, externalWallets)