[change:api] Assert __task_classificationTypeName instead of classification_id in tests

Author: Aryamanz29

tests/integration/aio/test_task_client.py

@@ -108,7 +108,7 @@ async def test_task_search(
     assert task.created_by
     assert task.updated_time
     assert task.parameters
-    assert task.classification_id
+    assert task.parameters.get("__task_classificationTypeName")
     assert task.attempt_count is not None and task.attempt_count >= 0
     assert task.entity_guid == snowflake_column.guid
     assert task.type == AtlanTaskType.CLASSIFICATION_PROPAGATION_ADD

tests/integration/test_task_client.py

@@ -106,7 +106,7 @@ def test_task_search(
     assert task.created_by
     assert task.updated_time
     assert task.parameters
-    assert task.classification_id
+    assert task.parameters.get("__task_classificationTypeName")
     assert task.attempt_count is not None and task.attempt_count >= 0
     assert task.entity_guid == snowflake_column.guid
     assert task.type == AtlanTaskType.CLASSIFICATION_PROPAGATION_ADD

tests/unit/aio/test_task_client.py

@@ -131,7 +131,7 @@ async def test_task_search_get_when_given_request(
         assert task.attempt_count == 0
         assert task.entity_guid
         assert task.time_taken_in_seconds
-        assert task.classification_id
+        assert task.parameters.get("__task_classificationTypeName")
         assert task.status == AtlanTaskStatus.COMPLETE
         assert task.type == AtlanTaskType.CLASSIFICATION_PROPAGATION_ADD
     assert mock_async_api_caller._call_api.call_count == 2