[test:fix] Fixed audit search assertions

Author: Aryamanz29

pyatlan/test_utils/base_vcr.py

@@ -267,5 +267,9 @@ def vcr_cassette_dir(self, request):
         """
         # Set self._CASSETTES_DIR or use the default directory path based on the test module name.
         # V9 tests (module name starting with tests_v9) use tests_v9/vcr_cassettes; legacy use tests/vcr_cassettes.
-        root = "tests_v9/vcr_cassettes" if request.module.__name__.startswith("tests_v9") else "tests/vcr_cassettes"
+        root = (
+            "tests_v9/vcr_cassettes"
+            if request.module.__name__.startswith("tests_v9")
+            else "tests/vcr_cassettes"
+        )
         return self._CASSETTES_DIR or os.path.join(root, request.module.__name__)

tests/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/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/unit/test_audit_search.py

@@ -41,23 +41,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