Updates for pydantic tests

Author: strawgate

key-value/key-value-aio/tests/adapters/test_pydantic.py

@@ -55,6 +55,13 @@ def model_type_from_log_record(record: LogRecord) -> str:
     return record.model_type  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
 
 
+def error_from_log_record(record: LogRecord) -> str:
+    if not hasattr(record, "error"):
+        msg = "Log record does not have an error attribute"
+        raise ValueError(msg)
+    return record.error  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
 def errors_from_log_record(record: LogRecord) -> list[str]:
     if not hasattr(record, "errors"):
         msg = "Log record does not have an errors attribute"
@@ -185,7 +192,7 @@ async def test_validation_error_logging(
 
         errors = errors_from_log_record(record)
         assert len(errors) == 1
-        assert "is_admin" in errors[0]
+        assert "is_admin" in str(errors[0])
 
     async def test_list_validation_error_logging(
         self, product_list_adapter: PydanticAdapter[list[Product]], store: MemoryStore, caplog: pytest.LogCaptureFixture
@@ -208,6 +215,5 @@ async def test_list_validation_error_logging(
         assert record.levelname == "ERROR"
         assert "Missing 'items' wrapper" in record.message
         assert model_type_from_log_record(record) == "Pydantic model"
-        errors = errors_from_log_record(record)
-        assert len(errors) == 1
-        assert "missing 'items' wrapper" in errors[0]
+        errors = error_from_log_record(record)
+        assert "missing 'items' wrapper" in str(errors)

key-value/key-value-sync/tests/code_gen/adapters/test_pydantic.py

@@ -58,6 +58,13 @@ def model_type_from_log_record(record: LogRecord) -> str:
     return record.model_type  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
 
 
+def error_from_log_record(record: LogRecord) -> str:
+    if not hasattr(record, "error"):
+        msg = "Log record does not have an error attribute"
+        raise ValueError(msg)
+    return record.error  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
 def errors_from_log_record(record: LogRecord) -> list[str]:
     if not hasattr(record, "errors"):
         msg = "Log record does not have an errors attribute"
@@ -188,7 +195,7 @@ def test_validation_error_logging(
 
         errors = errors_from_log_record(record)
         assert len(errors) == 1
-        assert "is_admin" in errors[0]
+        assert "is_admin" in str(errors[0])
 
     def test_list_validation_error_logging(
         self, product_list_adapter: PydanticAdapter[list[Product]], store: MemoryStore, caplog: pytest.LogCaptureFixture
@@ -211,6 +218,5 @@ def test_list_validation_error_logging(
         assert record.levelname == "ERROR"
         assert "Missing 'items' wrapper" in record.message
         assert model_type_from_log_record(record) == "Pydantic model"
-        errors = errors_from_log_record(record)
-        assert len(errors) == 1
-        assert "missing 'items' wrapper" in errors[0]
+        errors = error_from_log_record(record)
+        assert "missing 'items' wrapper" in str(errors)