def test_client_using_azure_ad_service_principal():
assert isinstance(
Client(auth_method=AuthMethods.AZURE_AD_SERVICE_PRINCIPAL,
databricks_instance="ddd",
access_token="ddd",
management_token="ddd",
resource_id="ddd"), AzureADServicePrincipalClient)
def test_wrong_auth_method():
with pytest.raises(Exception):
Client(auth_method="jwt",
databricks_instance="ddd",
access_token="ddd")
def test_bad_client_test_connection():
assert Client(databricks_instance="ddd",
personal_access_token="ddd").test_connection() == False
def test_client_using_personal_access_token():
assert isinstance(
Client(databricks_instance="ddd", personal_access_token="ddd"),
PersonalAccessTokenClient)
def test_client_using_azure_ad_user_non_admin_as_admin():
assert isinstance(
Client(auth_method=AuthMethods.AZURE_AD_USER,
databricks_instance="ddd",
access_token="ddd",
resource_id="hello"), AzureADUserClient)
def test_client_using_azure_ad_user():
assert isinstance(
Client(auth_method=AuthMethods.AZURE_AD_USER,
databricks_instance="ddd",
access_token="ddd"), AzureADUserClient)
def create_bad_client():
instance = env.str("DATABRICKS_INSTANCE")
return Client(databricks_instance=instance, personal_access_token="wrong")
def create_premium_client():
token = env.str("PERSONAL_ACCESS_TOKEN_PREMIUM")
instance = env.str("DATABRICKS_INSTANCE_PREMIUM")
return Client(databricks_instance=instance, personal_access_token=token)