feat(sqla_factory): refresh object in async_session after commit to db

polyfactory/factories/sqlalchemy_factory.py

@@ -50,11 +50,14 @@ def __init__(self, session: AsyncSession) -> None:
     async def save(self, data: T) -> T:
         self.session.add(data)
         await self.session.commit()
+        await self.session.refresh(data)
         return data
 
     async def save_many(self, data: list[T]) -> list[T]:
         self.session.add_all(data)
         await self.session.commit()
+        for batch_item in data:
+            await self.session.refresh(batch_item)
         return data
 
 
@@ -97,7 +100,8 @@ def get_sqlalchemy_types(cls) -> dict[Any, Callable[[], Any]]:
             postgresql.TSTZRANGE: lambda: (cls.__faker__.past_datetime(), datetime.now()),  # noqa: DTZ005
             postgresql.HSTORE: lambda: cls.__faker__.pydict(),
             # `types.JSON` is compatible for sqlachemy extend dialects. Such as `pg.JSON` and `JSONB`
-            types.JSON: lambda: cls.__faker__.pydict(),
+            # accept `str,int,float` as basic types in case `sqlalchemy/sql/sqltypes.json.json_serializer` raise error
+            types.JSON: lambda: cls.__faker__.pydict(value_types=[int, str, float]),
         }
 
     @classmethod

tests/sqlalchemy_factory/test_sqlalchemy_factory_common.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass
+from datetime import datetime
 from enum import Enum
 from typing import Any, Callable, Type, Union
 from uuid import UUID
 
 import pytest
-from sqlalchemy import Column, ForeignKey, Integer, String, create_engine, inspect, orm, types
+from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, create_engine, func, inspect, orm, types
 from sqlalchemy.engine import Engine
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine
 from sqlalchemy.ext.hybrid import hybrid_property
@@ -318,6 +319,7 @@ class AsyncModel(Base):
         __tablename__ = "table"
 
         id: Any = Column(Integer(), primary_key=True)
+        test_datetime: Any = Column(DateTime(timezone=True), server_default=func.now())
 
     await create_tables(async_engine, Base)
 
@@ -328,12 +330,18 @@ class Factory(SQLAlchemyFactory[AsyncModel]):
             __model__ = AsyncModel
 
         result = await Factory.create_async()
+        assert result.test_datetime is not None
+        assert isinstance(result.test_datetime, datetime)
         assert inspect(result).persistent  # type: ignore[union-attr]
+        await session.delete(result)
+        await session.commit()
 
         batch_result = await Factory.create_batch_async(size=2)
         assert len(batch_result) == 2
         for batch_item in batch_result:
             assert inspect(batch_item).persistent  # type: ignore[union-attr]
+            assert batch_item.test_datetime is not None
+            assert isinstance(result.test_datetime, datetime)
 
 
 def test_alias() -> None: