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
129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
"""
|
|
Comprehensive unit tests for API Endpoint entity.
|
|
"""
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
from uuid import UUID
|
|
|
|
from metadata.generated.schema.api.data.createAPIEndpoint import (
|
|
CreateAPIEndpointRequest,
|
|
)
|
|
from metadata.generated.schema.entity.data.apiEndpoint import (
|
|
APIEndpoint as APIEndpointEntity,
|
|
)
|
|
from metadata.sdk import APIEndpoints
|
|
|
|
|
|
class TestAPIEndpointEntity(unittest.TestCase):
|
|
"""Comprehensive tests for APIEndpoint entity operations"""
|
|
|
|
def setUp(self):
|
|
"""Set up test fixtures"""
|
|
self.mock_ometa = MagicMock()
|
|
APIEndpoints.set_default_client(self.mock_ometa)
|
|
|
|
self.entity_id = "550e8400-e29b-41d4-a716-446655440000"
|
|
self.entity_fqn = "service.api_endpoint.test_api_endpoint"
|
|
|
|
def test_create_api_endpoint(self):
|
|
"""Test creating a api endpoint"""
|
|
create_request = MagicMock(spec=CreateAPIEndpointRequest)
|
|
create_request.name = "test_api_endpoint"
|
|
create_request.displayName = "Test API Endpoint"
|
|
create_request.description = "Test api endpoint for unit tests"
|
|
|
|
expected_entity = MagicMock(spec=APIEndpointEntity)
|
|
expected_entity.id = UUID(self.entity_id)
|
|
expected_entity.name = "test_api_endpoint"
|
|
|
|
self.mock_ometa.create_or_update.return_value = expected_entity
|
|
|
|
result = APIEndpoints.create(create_request)
|
|
|
|
self.assertEqual(str(result.id), self.entity_id)
|
|
self.assertEqual(result.name, "test_api_endpoint")
|
|
self.mock_ometa.create_or_update.assert_called_once_with(create_request)
|
|
|
|
def test_retrieve_api_endpoint_by_id(self):
|
|
"""Test retrieving a api endpoint by ID"""
|
|
expected_entity = MagicMock(spec=APIEndpointEntity)
|
|
expected_entity.id = UUID(self.entity_id)
|
|
expected_entity.name = "test_api_endpoint"
|
|
|
|
self.mock_ometa.get_by_id.return_value = expected_entity
|
|
|
|
result = APIEndpoints.retrieve(self.entity_id)
|
|
|
|
self.assertEqual(str(result.id), self.entity_id)
|
|
self.mock_ometa.get_by_id.assert_called_once_with(
|
|
entity=APIEndpointEntity, entity_id=self.entity_id, fields=None
|
|
)
|
|
|
|
def test_retrieve_api_endpoint_by_name(self):
|
|
"""Test retrieving a api endpoint by name"""
|
|
expected_entity = MagicMock(spec=APIEndpointEntity)
|
|
expected_entity.fullyQualifiedName = self.entity_fqn
|
|
|
|
self.mock_ometa.get_by_name.return_value = expected_entity
|
|
|
|
result = APIEndpoints.retrieve_by_name(self.entity_fqn)
|
|
|
|
self.assertEqual(result.fullyQualifiedName, self.entity_fqn)
|
|
self.mock_ometa.get_by_name.assert_called_once_with(entity=APIEndpointEntity, fqn=self.entity_fqn, fields=None)
|
|
|
|
def test_update_api_endpoint(self):
|
|
"""Test updating a api endpoint"""
|
|
entity_to_update = MagicMock(spec=APIEndpointEntity)
|
|
entity_to_update.id = UUID(self.entity_id)
|
|
entity_to_update.description = "Updated description"
|
|
|
|
# Mock the get_by_id to return the current state
|
|
current_entity = MagicMock(spec=type(entity_to_update))
|
|
current_entity.id = entity_to_update.id if hasattr(entity_to_update, "id") else UUID(self.entity_id)
|
|
self.mock_ometa.get_by_id.return_value = current_entity
|
|
|
|
# Mock the patch to return the updated entity
|
|
self.mock_ometa.patch.return_value = entity_to_update
|
|
|
|
result = APIEndpoints.update(entity_to_update)
|
|
|
|
self.assertEqual(result.description, "Updated description")
|
|
# Verify get_by_id was called to fetch current state
|
|
self.mock_ometa.get_by_id.assert_called_once()
|
|
# Verify patch was called with source and destination
|
|
self.mock_ometa.patch.assert_called_once()
|
|
|
|
def test_delete_api_endpoint(self):
|
|
"""Test deleting a api endpoint"""
|
|
APIEndpoints.delete(self.entity_id, recursive=True, hard_delete=False)
|
|
|
|
self.mock_ometa.delete.assert_called_once_with(
|
|
entity=APIEndpointEntity,
|
|
entity_id=self.entity_id,
|
|
recursive=True,
|
|
hard_delete=False,
|
|
)
|
|
|
|
def test_list_api_endpoints(self):
|
|
"""Test listing api endpoints"""
|
|
mock_entity1 = MagicMock(spec=APIEndpointEntity)
|
|
mock_entity1.name = "entity1"
|
|
mock_entity2 = MagicMock(spec=APIEndpointEntity)
|
|
mock_entity2.name = "entity2"
|
|
|
|
mock_response = MagicMock()
|
|
mock_response.entities = [mock_entity1, mock_entity2]
|
|
|
|
self.mock_ometa.list_entities.return_value = mock_response
|
|
|
|
result = APIEndpoints.list(limit=10)
|
|
|
|
self.assertEqual(len(result.entities), 2)
|
|
self.assertEqual(result.entities[0].name, "entity1")
|
|
self.mock_ometa.list_entities.assert_called_once()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|