context.Background() -> t.Context() (#3739)

Author: mmetc

pkg/acquisition/modules/file/file_test.go

@@ -1,7 +1,6 @@
 package fileacquisition_test
 
 import (
-	"context"
 	"fmt"
 	"os"
 	"path/filepath"
@@ -545,6 +544,7 @@ mode: tail
 }
 
 func TestDiscoveryPolling(t *testing.T) {
+	ctx := t.Context()
 	dir := t.TempDir()
 
 	pattern := filepath.Join(dir, "*.log")
@@ -569,7 +569,7 @@ mode: tail
 	tomb := tomb.Tomb{}
 
 	// Start acquisition
-	err = f.StreamingAcquisition(context.Background(), eventChan, &tomb)
+	err = f.StreamingAcquisition(ctx, eventChan, &tomb)
 	require.NoError(t, err)
 
 	// Create a test file

pkg/apiserver/alerts_test.go

@@ -148,7 +148,7 @@ func TestCreateAlert(t *testing.T) {
 }
 
 func TestCreateAllowlistedAlert(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	allowlist, err := lapi.DBClient.CreateAllowList(ctx, "test", "test", "", false)

pkg/apiserver/allowlists_test.go

@@ -1,7 +1,6 @@
 package apiserver
 
 import (
-	"context"
 	"encoding/json"
 	"net/http"
 	"strings"
@@ -16,7 +15,7 @@ import (
 )
 
 func TestAllowlistList(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	_, err := lapi.DBClient.CreateAllowList(ctx, "test", "test", "", false)
@@ -37,7 +36,7 @@ func TestAllowlistList(t *testing.T) {
 }
 
 func TestGetAllowlist(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	l, err := lapi.DBClient.CreateAllowList(ctx, "test", "test", "", false)
@@ -72,7 +71,7 @@ func TestGetAllowlist(t *testing.T) {
 }
 
 func TestCheckInAllowlist(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	l, err := lapi.DBClient.CreateAllowList(ctx, "test", "test", "", false)
@@ -150,7 +149,7 @@ func TestCheckInAllowlist(t *testing.T) {
 }
 
 func TestBulkCheckAllowlist(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	// create an allowlist and add one live entry
@@ -191,7 +190,7 @@ func TestBulkCheckAllowlist(t *testing.T) {
 }
 
 func TestBulkCheckAllowlist_BadRequest(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	lapi := SetupLAPITest(t, ctx)
 
 	// missing or empty body should yield 400

pkg/database/allowlists_test.go

@@ -27,7 +27,7 @@ func getDBClient(t *testing.T, ctx context.Context) *Client {
 }
 
 func TestCheckAllowlist(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dbClient := getDBClient(t, ctx)
 
 	allowlist, err := dbClient.CreateAllowList(ctx, "test", "test", "", false)
@@ -106,7 +106,7 @@ func TestCheckAllowlist(t *testing.T) {
 }
 
 func TestIsAllowListedBy_SingleAndMultiple(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dbClient := getDBClient(t, ctx)
 
 	list1, err := dbClient.CreateAllowList(ctx, "list1", "first list", "", false)
@@ -149,7 +149,7 @@ func TestIsAllowListedBy_SingleAndMultiple(t *testing.T) {
 }
 
 func TestIsAllowListedBy_NoMatch(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dbClient := getDBClient(t, ctx)
 
 	list, err := dbClient.CreateAllowList(ctx, "solo", "single", "", false)