diff --git a/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/EncryptionKeysIT.java b/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/EncryptionKeysIT.java index f36b13a8c..8b7324c8e 100644 --- a/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/EncryptionKeysIT.java +++ b/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/EncryptionKeysIT.java @@ -1,7 +1,11 @@ package com.databricks.sdk.integration; +import com.databricks.sdk.AccountClient; +import com.databricks.sdk.integration.framework.CollectionUtils; import com.databricks.sdk.integration.framework.EnvContext; import com.databricks.sdk.integration.framework.EnvTest; +import com.databricks.sdk.service.provisioning.CustomerManagedKey; +import org.junit.jupiter.api.Test; import org.junit.jupiter.api.condition.DisabledIfEnvironmentVariable; import org.junit.jupiter.api.extension.ExtendWith; @@ -9,15 +13,12 @@ @DisabledIfEnvironmentVariable(named = "ARM_CLIENT_ID", matches = ".*") @ExtendWith(EnvTest.class) public class EncryptionKeysIT { - // TODO: Enable this test when the test account is updated to support this. - // Either by upgrading the test account tier to Enterprise or by adding this - // feature to the test account. - // @Test - // void lists(AccountClient a) { - // Iterable list = a.encryptionKeys().list(); + @Test + void lists(AccountClient a) { + Iterable list = a.encryptionKeys().list(); - // java.util.List all = CollectionUtils.asList(list); + java.util.List all = CollectionUtils.asList(list); - // CollectionUtils.assertUnique(all); - // } + CollectionUtils.assertUnique(all); + } }