Fix TestUcAccJobRunAsMutations (#3474)

* Fix TestUcAccJobRunAsMutations

* fixed

Author: mgyucht

internal/acceptance/init_test.go

@@ -354,19 +354,29 @@ func isInDebug() bool {
 	return strings.HasPrefix(path.Base(ex), "__debug_bin")
 }
 
+func setDebugLogger() {
+	logger.DefaultLogger = &logger.SimpleLogger{
+		Level: logger.LevelDebug,
+	}
+}
+
 func loadWorkspaceEnv(t *testing.T) {
+	setDebugLogger()
 	loadDebugEnvIfRunsFromIDE(t, "workspace")
 }
 
 func loadAccountEnv(t *testing.T) {
+	setDebugLogger()
 	loadDebugEnvIfRunsFromIDE(t, "account")
 }
 
 func loadUcwsEnv(t *testing.T) {
+	setDebugLogger()
 	loadDebugEnvIfRunsFromIDE(t, "ucws")
 }
 
 func loadUcacctEnv(t *testing.T) {
+	setDebugLogger()
 	loadDebugEnvIfRunsFromIDE(t, "ucacct")
 }
 
@@ -393,6 +403,20 @@ func isCloudEnvInList(t *testing.T, cloudEnvs []string) bool {
 	return slices.Contains(cloudEnvs, cloudEnv)
 }
 
+func isAuthedAsWorkspaceServicePrincipal(ctx context.Context) (bool, error) {
+	w := databricks.Must(databricks.NewWorkspaceClient())
+	user, err := w.CurrentUser.Me(ctx)
+	if err != nil {
+		return false, err
+	}
+	for _, emailValue := range user.Emails {
+		if emailValue.Primary && strings.Contains(emailValue.Value, "@") {
+			return false, nil
+		}
+	}
+	return true, nil
+}
+
 // loads debug environment from ~/.databricks/debug-env.json
 func loadDebugEnvIfRunsFromIDE(t *testing.T, key string) {
 	if !isInDebug() {

internal/acceptance/job_test.go

@@ -12,6 +12,7 @@ import (
 	"github.com/databricks/terraform-provider-databricks/common"
 	"github.com/databricks/terraform-provider-databricks/qa"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestAccJobTasks(t *testing.T) {
@@ -355,9 +356,21 @@ func TestUcAccJobRunAsServicePrincipal(t *testing.T) {
 	})
 }
 
+func getRunAsAttribute(t *testing.T, ctx context.Context) string {
+	isSp, err := isAuthedAsWorkspaceServicePrincipal(ctx)
+	require.NoError(t, err)
+	if isSp {
+		return "service_principal_name"
+	}
+	return "user_name"
+}
+
 func TestUcAccJobRunAsMutations(t *testing.T) {
 	loadUcwsEnv(t)
 	spId := GetEnvOrSkipTest(t, "ACCOUNT_LEVEL_SERVICE_PRINCIPAL_ID")
+	// Note: the attribute must match the type of principal that the test is run as.
+	ctx := context.Background()
+	attribute := getRunAsAttribute(t, ctx)
 	unityWorkspaceLevel(
 		t,
 		// Provision job with service principal `run_as`
@@ -366,7 +379,7 @@ func TestUcAccJobRunAsMutations(t *testing.T) {
 		},
 		// Update job to a user `run_as`
 		step{
-			Template: runAsTemplate(`user_name = data.databricks_current_user.me.user_name`),
+			Template: runAsTemplate(attribute + ` = data.databricks_current_user.me.user_name`),
 		},
 		// Update job back to a service principal `run_as`
 		step{