Fix UnifiedHostGroupsIT to follow unified config test pattern

Co-authored-by: Isaac

Author: hectorcast-db

databricks-sdk-java/src/test/java/com/databricks/sdk/integration/UnifiedHostGroupsIT.java

@@ -2,6 +2,7 @@
 
 import static org.junit.jupiter.api.Assertions.*;
 
+import com.databricks.sdk.AccountClient;
 import com.databricks.sdk.WorkspaceClient;
 import com.databricks.sdk.core.DatabricksConfig;
 import com.databricks.sdk.integration.framework.EnvContext;
@@ -10,23 +11,28 @@
 import com.databricks.sdk.service.iam.Group;
 import com.databricks.sdk.service.iam.ListGroupsRequest;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.condition.DisabledIfEnvironmentVariable;
+import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
 import org.junit.jupiter.api.extension.ExtendWith;
 
 @EnvContext("account")
 @ExtendWith(EnvTest.class)
+@EnabledIfEnvironmentVariable(named = "UNIFIED_HOST", matches = ".+")
 public class UnifiedHostGroupsIT {
   @Test
+  @DisabledIfEnvironmentVariable(named = "CLOUD_PROVIDER", matches = "GCP")
   void listWorkspaceGroupsViaUnifiedHost(
-      @EnvOrSkip("UNIFIED_HOST") String host,
+      AccountClient a,
+      @EnvOrSkip("UNIFIED_HOST") String unifiedHost,
       @EnvOrSkip("TEST_WORKSPACE_ID") String workspaceId,
-      @EnvOrSkip("DATABRICKS_CLIENT_ID") String clientId,
-      @EnvOrSkip("DATABRICKS_CLIENT_SECRET") String clientSecret) {
+      @EnvOrSkip("TEST_ACCOUNT_ID") String accountId) {
     DatabricksConfig config =
         new DatabricksConfig()
-            .setHost(host)
+            .setHost(unifiedHost)
+            .setClientId(a.config().getClientId())
+            .setClientSecret(a.config().getClientSecret())
             .setWorkspaceId(workspaceId)
-            .setClientId(clientId)
-            .setClientSecret(clientSecret);
+            .setAccountId(accountId);
     WorkspaceClient ws = new WorkspaceClient(config);
 
     Iterable<Group> groups = ws.groups().list(new ListGroupsRequest().setAttributes("displayName"));