fix(v9): unit test fixes - Procedure updater, audit fixture camelCase, Entity UNSET defaults

- Procedure.updater(): make definition optional (default '') so updater(qualified_name, name) works
- Entity: set classification_names, meanings, labels, pending_tasks defaults to UNSET so workflow
  package serialization omits them (fixes test_packages AssertionError on connection payload)
- Audit search: audit_search_paging.json entity audit keys to camelCase (entityQualifiedName,
  typeName, entityId) for EntityAudit decode; assert helpers use first['entityId'] etc.

Made-with: Cursor

Author: Aryamanz29

pyatlan_v9/model/assets/entity.py

@@ -177,15 +177,15 @@ class Entity(msgspec.Struct, kw_only=True, omit_defaults=True, rename="camel"):
     remove_classifications: Union[list[Any], UnsetType] = UNSET
     """Classifications to remove from this entity (used during save)."""
 
-    classification_names: Union[list, None, UnsetType] = None
+    classification_names: Union[list, None, UnsetType] = UNSET
     """Simple list of classification type names assigned to this entity."""
 
     # Meanings - typed for better validation
-    meanings: Union[list[TermAssignment], None, UnsetType] = None
+    meanings: Union[list[TermAssignment], None, UnsetType] = UNSET
     """Glossary term assignments for this entity."""
 
     # Labels
-    labels: Union[list[str], None, UnsetType] = None
+    labels: Union[list[str], None, UnsetType] = UNSET
     """Simple string labels attached to this entity."""
 
     # Additional metadata
@@ -195,7 +195,7 @@ class Entity(msgspec.Struct, kw_only=True, omit_defaults=True, rename="camel"):
     custom_attributes: Union[dict[str, str], UnsetType] = UNSET
     """Custom key-value pairs for this entity."""
 
-    pending_tasks: Union[list[str], None, UnsetType] = None
+    pending_tasks: Union[list[str], None, UnsetType] = UNSET
     """Identifiers of pending tasks for this entity."""
 
     proxy: Union[bool, UnsetType] = UNSET

pyatlan_v9/model/assets/procedure.py

@@ -193,13 +193,20 @@ def creator(
         )
 
     @classmethod
-    def updater(cls, *, qualified_name: str, name: str, definition: str) -> "Procedure":
+    def updater(
+        cls,
+        *,
+        qualified_name: str,
+        name: str,
+        definition: str = "",
+    ) -> "Procedure":
         validate_required_fields(
-            ["qualified_name", "name", "definition"],
-            [qualified_name, name, definition],
+            ["qualified_name", "name"],
+            [qualified_name, name],
         )
         proc = cls(qualified_name=qualified_name, name=name)
-        proc.definition = definition
+        if definition:
+            proc.definition = definition
         return proc
 
     def trim_to_required(self) -> "Procedure":

tests/unit/data/search_responses/audit_search_paging.json

@@ -2,9 +2,9 @@
   "totalCount": 1,
   "entityAudits": [
     {
-      "entity_qualified_name": "sample_entity_1",
-      "type_name": "AtlasGlossaryTerm",
-      "entity_id": "guid_1",
+      "entityQualifiedName": "sample_entity_1",
+      "typeName": "AtlasGlossaryTerm",
+      "entityId": "guid_1",
       "timestamp": 1733491479782,
       "created": 1733491480576,
       "user": "user1",

tests_v9/unit/aio/test_audit_search.py

@@ -47,23 +47,21 @@ def load_json(filename):
 async def _assert_audit_search_results(
     results: AsyncAuditSearchResults, response_json, sorts, bulk=False
 ):
+    first = response_json["entityAudits"][0]
     async for audit in results:
-        assert audit.entity_id == response_json["entityAudits"][0]["entity_id"]
-        assert (
-            audit.entity_qualified_name
-            == response_json["entityAudits"][0]["entity_qualified_name"]
-        )
-        assert audit.type_name == response_json["entityAudits"][0]["type_name"]
+        assert audit.entity_id == first["entityId"]
+        assert audit.entity_qualified_name == first["entityQualifiedName"]
+        assert audit.type_name == first["typeName"]
         expected_timestamp = datetime.fromtimestamp(
-            response_json["entityAudits"][0]["timestamp"] / 1000, tz=timezone.utc
+            first["timestamp"] / 1000, tz=timezone.utc
         )
         assert audit.timestamp == expected_timestamp
         expected_created = datetime.fromtimestamp(
-            response_json["entityAudits"][0]["created"] / 1000, tz=timezone.utc
+            first["created"] / 1000, tz=timezone.utc
         )
         assert audit.created == expected_created
-        assert audit.user == response_json["entityAudits"][0]["user"]
-        assert audit.action == response_json["entityAudits"][0]["action"]
+        assert audit.user == first["user"]
+        assert audit.action == first["action"]
 
     assert results.total_count == response_json["totalCount"]
     assert results._bulk == bulk

tests_v9/unit/test_audit_search.py

@@ -55,23 +55,21 @@ def load_json(filename):
 def _assert_audit_search_results(
     results: AuditSearchResults, response_json, sorts, bulk=False
 ):
+    first = response_json["entityAudits"][0]
     for audit in results:
-        assert audit.entity_id == response_json["entityAudits"][0]["entity_id"]
-        assert (
-            audit.entity_qualified_name
-            == response_json["entityAudits"][0]["entity_qualified_name"]
-        )
-        assert audit.type_name == response_json["entityAudits"][0]["type_name"]
+        assert audit.entity_id == first["entityId"]
+        assert audit.entity_qualified_name == first["entityQualifiedName"]
+        assert audit.type_name == first["typeName"]
         expected_timestamp = datetime.fromtimestamp(
-            response_json["entityAudits"][0]["timestamp"] / 1000, tz=timezone.utc
+            first["timestamp"] / 1000, tz=timezone.utc
         )
         assert audit.timestamp == expected_timestamp
         expected_created = datetime.fromtimestamp(
-            response_json["entityAudits"][0]["created"] / 1000, tz=timezone.utc
+            first["created"] / 1000, tz=timezone.utc
         )
         assert audit.created == expected_created
-        assert audit.user == response_json["entityAudits"][0]["user"]
-        assert audit.action == response_json["entityAudits"][0]["action"]
+        assert audit.user == first["user"]
+        assert audit.action == first["action"]
 
     assert results.total_count == response_json["totalCount"]
     assert results._bulk == bulk