Fixes #26321: correct Python SDK create-to-entity import naming for special cases (#26985)

* Fix SDK create-to-entity import mapping for TestCaseResult and TableProfile

* Address review: assert exact create-to-entity mapping classes in tests

* chore: ran python linting

---------

Co-authored-by: TeddyCr <teddy.crepineau@gmail.com>

Author: abhayguptas

ingestion/src/metadata/ingestion/ometa/ometa_api.py

@@ -429,6 +429,13 @@ def update_file_name(create: Type[C], file_name: str) -> str:
         """
         Update the filename for services and schemas
         """
+        explicit_file_name_map = {
+            "CreateTableProfileRequest": "table",
+            "CreateTestCaseResult": "basic",
+        }
+        if create.__name__ in explicit_file_name_map:
+            return explicit_file_name_map[create.__name__]
+
         if "service" in create.__name__.lower():
             return file_name.replace("service", "Service")
 

ingestion/tests/unit/test_ometa_endpoints.py

@@ -14,6 +14,9 @@
 """
 from unittest import TestCase
 
+from metadata.generated.schema.api.data.createTableProfile import (
+    CreateTableProfileRequest,
+)
 from metadata.generated.schema.api.data.createTopic import CreateTopicRequest
 from metadata.generated.schema.api.services.createDatabaseService import (
     CreateDatabaseServiceRequest,
@@ -22,6 +25,9 @@
     CreateIngestionPipelineRequest,
 )
 from metadata.generated.schema.api.teams.createUser import CreateUserRequest
+from metadata.generated.schema.api.tests.createTestCaseResult import (
+    CreateTestCaseResult,
+)
 from metadata.generated.schema.entity.data.chart import Chart
 from metadata.generated.schema.entity.data.dashboard import Dashboard
 from metadata.generated.schema.entity.data.database import Database
@@ -30,7 +36,7 @@
 from metadata.generated.schema.entity.data.mlmodel import MlModel
 from metadata.generated.schema.entity.data.pipeline import Pipeline
 from metadata.generated.schema.entity.data.report import Report
-from metadata.generated.schema.entity.data.table import Table
+from metadata.generated.schema.entity.data.table import Table, TableProfile
 from metadata.generated.schema.entity.data.topic import Topic
 from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
     OpenMetadataConnection,
@@ -46,6 +52,7 @@
 from metadata.generated.schema.security.client.openMetadataJWTClientConfig import (
     OpenMetadataJWTClientConfig,
 )
+from metadata.generated.schema.tests.basic import TestCaseResult
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
 
 
@@ -117,13 +124,19 @@ def test_get_entity_from_create(self):
         """
 
         entity = self.metadata.get_entity_from_create(CreateTopicRequest)
-        assert issubclass(entity, Topic)
+        assert entity is Topic
 
         entity = self.metadata.get_entity_from_create(CreateDatabaseServiceRequest)
-        assert issubclass(entity, DatabaseService)
+        assert entity is DatabaseService
 
         entity = self.metadata.get_entity_from_create(CreateUserRequest)
-        assert issubclass(entity, User)
+        assert entity is User
 
         entity = self.metadata.get_entity_from_create(CreateIngestionPipelineRequest)
-        assert issubclass(entity, IngestionPipeline)
+        assert entity is IngestionPipeline
+
+        entity = self.metadata.get_entity_from_create(CreateTestCaseResult)
+        assert entity is TestCaseResult
+
+        entity = self.metadata.get_entity_from_create(CreateTableProfileRequest)
+        assert entity is TableProfile