Add regression tests for unified-host account profiles

Table-driven tests for isAccountLevelConfig covering classic account
host, unified host with/without account ID, workspace host, and missing
host. Also adds a test for checkIdentity skipping the /me endpoint for
unified-host account profiles.

Co-authored-by: Isaac

Author: simonfaltum

cmd/doctor/doctor_test.go

@@ -357,6 +357,77 @@ func TestCheckIdentitySkippedForAccountLevel(t *testing.T) {
 	assert.Contains(t, result.Message, "account-level")
 }
 
+func TestIsAccountLevelConfig(t *testing.T) {
+	tests := []struct {
+		name string
+		cfg  *config.Config
+		want bool
+	}{
+		{
+			name: "classic account host",
+			cfg: &config.Config{
+				Host:      "https://accounts.cloud.databricks.com",
+				AccountID: "test-account-123",
+			},
+			want: true,
+		},
+		{
+			name: "unified host with account ID",
+			cfg: &config.Config{
+				Host:                       "https://myhost.databricks.com",
+				AccountID:                  "test-account-123",
+				Experimental_IsUnifiedHost: true,
+			},
+			want: true,
+		},
+		{
+			name: "unified host without account ID is workspace",
+			cfg: &config.Config{
+				Host:                       "https://myhost.databricks.com",
+				Experimental_IsUnifiedHost: true,
+			},
+			want: false,
+		},
+		{
+			name: "workspace host",
+			cfg: &config.Config{
+				Host: "https://myhost.databricks.com",
+			},
+			want: false,
+		},
+		{
+			name: "no host",
+			cfg: &config.Config{
+				AccountID: "test-account-123",
+			},
+			want: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assert.Equal(t, tt.want, isAccountLevelConfig(tt.cfg))
+		})
+	}
+}
+
+func TestCheckIdentitySkippedForUnifiedHostAccount(t *testing.T) {
+	cfg := &config.Config{
+		Host:                       "https://myhost.databricks.com",
+		AccountID:                  "test-account-123",
+		Token:                      "test-token",
+		Experimental_IsUnifiedHost: true,
+	}
+
+	ctx := cmdio.MockDiscard(t.Context())
+	cmd := newTestCmd(ctx)
+
+	result := checkIdentity(cmd, cfg)
+	assert.Equal(t, "Identity", result.Name)
+	assert.Equal(t, statusSkip, result.Status)
+	assert.Contains(t, result.Message, "account-level")
+}
+
 func TestCheckNetworkReachable(t *testing.T) {
 	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(http.StatusOK)