[fix/unit] Legacy audit search tests use camelCase keys from fixture

tests/unit and tests/unit/aio assert on audit_search_paging.json which was
updated to camelCase (entityQualifiedName, typeName, entityId); update
_assert_audit_search_results to read first['entityId'] etc.

Made-with: Cursor

Author: Aryamanz29

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/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