wehub-resource-sync
bf2343b7e4
Integration Tests - MySQL + Elasticsearch / Detect Changes (push) Has been cancelled
Integration Tests - MySQL + Elasticsearch / integration-tests-mysql-elasticsearch (push) Has been cancelled
Integration Tests - PostgreSQL + Elasticsearch + Redis / Detect Changes (push) Has been cancelled
Integration Tests - PostgreSQL + Elasticsearch + Redis / integration-tests-postgres-elasticsearch-redis (push) Has been cancelled
Integration Tests - PostgreSQL + OpenSearch / Detect Changes (push) Has been cancelled
Integration Tests - PostgreSQL + OpenSearch / integration-tests-postgres-opensearch (push) Has been cancelled
Java Checkstyle / java-checkstyle (push) Has been cancelled
Maven Collate Tests / maven-collate-ci (push) Has been cancelled
OpenMetadata Service Unit Tests / openmetadata-service-unit-tests-status (push) Has been cancelled
Publish Package to Maven Central Repository / publish-maven-packages (push) Has been cancelled
OpenMetadata Service Unit Tests / Detect Changes (push) Has been cancelled
OpenMetadata Service Unit Tests / openmetadata-service-unit-tests (push) Has been cancelled
OpenMetadata Service Unit Tests / k8s_operator-unit-tests (push) Has been cancelled
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from metadata.clients.azure_client import AzureClient
|
|
from metadata.generated.schema.security.credentials.azureCredentials import (
|
|
AzureCredentials,
|
|
)
|
|
|
|
|
|
class TestAzureClient(unittest.TestCase):
|
|
@patch("azure.identity.ClientSecretCredential")
|
|
@patch("azure.identity.DefaultAzureCredential")
|
|
def test_create_client(
|
|
self,
|
|
mock_default_credential,
|
|
mock_client_secret_credential,
|
|
):
|
|
# Test with ClientSecretCredential
|
|
credentials = AzureCredentials(clientId="clientId", clientSecret="clientSecret", tenantId="tenantId")
|
|
instance = AzureClient(credentials)
|
|
instance.create_client()
|
|
|
|
mock_client_secret_credential.assert_called_once()
|
|
mock_client_secret_credential.reset_mock()
|
|
|
|
credentials = AzureCredentials(
|
|
clientId="clientId",
|
|
)
|
|
instance = AzureClient(credentials)
|
|
|
|
instance.create_client()
|
|
|
|
mock_default_credential.assert_called_once()
|
|
|
|
@patch("azure.storage.blob.BlobServiceClient")
|
|
def test_create_blob_client(self, mock_blob_service_client):
|
|
credentials = AzureCredentials(clientId="clientId", clientSecret="clientSecret", tenantId="tenantId")
|
|
with self.assertRaises(ValueError):
|
|
AzureClient(credentials=credentials).create_blob_client()
|
|
|
|
credentials.accountName = "accountName"
|
|
AzureClient(credentials=credentials).create_blob_client()
|
|
mock_blob_service_client.assert_called_once()
|
|
|
|
@patch("azure.keyvault.secrets.SecretClient")
|
|
def test_create_secret_client(self, mock_secret_client):
|
|
credentials = AzureCredentials(clientId="clientId", clientSecret="clientSecret", tenantId="tenantId")
|
|
with self.assertRaises(ValueError):
|
|
AzureClient(credentials=credentials).create_secret_client()
|
|
|
|
credentials.vaultName = "vaultName"
|
|
AzureClient(credentials=credentials).create_secret_client()
|
|
mock_secret_client.assert_called_once()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|