Replace context.Background() with t.Context() in tests

Fixes gocritic lint rule after the t.Context() migration on main.

Author: simonfaltum

cmd/auth/login_test.go

@@ -262,13 +262,13 @@ func TestHasNoProfiles_FreshMachine(t *testing.T) {
 	// On a fresh machine there is no config file. LoadProfiles returns
 	// ErrNoConfiguration. hasNoProfiles must treat this as "no profiles"
 	// (return true), not as an error (return false).
-	ctx := context.Background()
+	ctx := t.Context()
 	t.Setenv("DATABRICKS_CONFIG_FILE", filepath.Join(t.TempDir(), "nonexistent"))
 	assert.True(t, hasNoProfiles(ctx, profile.DefaultProfiler))
 }
 
 func TestHasNoProfiles_EmptyFile(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 	require.NoError(t, os.WriteFile(configFile, []byte(""), 0o600))
@@ -277,7 +277,7 @@ func TestHasNoProfiles_EmptyFile(t *testing.T) {
 }
 
 func TestHasNoProfiles_WithExistingProfile(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 	require.NoError(t, os.WriteFile(configFile, []byte("[p1]\nhost = https://abc\n"), 0o600))

cmd/auth/switch_test.go

@@ -1,7 +1,6 @@
 package auth
 
 import (
-	"context"
 	"fmt"
 	"os"
 	"path/filepath"
@@ -15,7 +14,7 @@ import (
 )
 
 func TestSwitchCommand_WithProfileFlag(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 
@@ -45,7 +44,7 @@ func TestSwitchCommand_WithProfileFlag(t *testing.T) {
 }
 
 func TestSwitchCommand_ProfileNotFound(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 
@@ -71,7 +70,7 @@ func TestSwitchCommand_ProfileNotFound(t *testing.T) {
 }
 
 func TestSwitchCommand_NonInteractiveNoProfile(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 
@@ -97,7 +96,7 @@ func TestSwitchCommand_NonInteractiveNoProfile(t *testing.T) {
 }
 
 func TestSwitchCommand_WritesSettingsSection(t *testing.T) {
-	ctx := context.Background()
+	ctx := t.Context()
 	dir := t.TempDir()
 	configFile := filepath.Join(dir, ".databrickscfg")
 

cmd/root/auth_test.go

@@ -385,7 +385,7 @@ token = flag-token
 				t.Setenv("DATABRICKS_CONFIG_PROFILE", tc.envProfile)
 			}
 
-			ctx := cmdio.MockDiscard(context.Background())
+			ctx := cmdio.MockDiscard(t.Context())
 			ctx = SkipLoadBundle(ctx)
 			cmd := New(ctx)
 
@@ -422,7 +422,7 @@ token = named-token
 
 	t.Setenv("DATABRICKS_CONFIG_FILE", configFile)
 
-	ctx := cmdio.MockDiscard(context.Background())
+	ctx := cmdio.MockDiscard(t.Context())
 	ctx = SkipLoadBundle(ctx)
 	cmd := New(ctx)