tests: Add the run_sync method on_close_pre tests

Author: amisadmin

tests/test_AbcAsyncDatabase.py

@@ -94,3 +94,10 @@ def delete_user(session: Session, instance: User):
     await db.async_run_sync(delete_user, user, is_session=True)
     user = await db.async_get(User, 1)
     assert user is None
+
+    # test on_close_pre
+    def get_user(session: Session, user_id: int):
+        return session.get(User, user_id)
+
+    user_id = await db.async_run_sync(get_user, 2, is_session=True, on_close_pre=lambda r: r.id)
+    assert user_id == 2

tests/test_AsyncDatabase.py

@@ -102,3 +102,10 @@ def delete_user(session: Session, instance: User):
     await db.run_sync(delete_user, user, is_session=True)
     user = await db.get(User, 1)
     assert user is None
+
+    # test on_close_pre
+    def get_user(session: Session, user_id: int):
+        return session.get(User, user_id)
+
+    user_id = await db.run_sync(get_user, 2, is_session=True, on_close_pre=lambda r: r.id)
+    assert user_id == 2

tests/test_Database.py

@@ -101,3 +101,10 @@ def delete_user(session: Session, instance: User):
     db.run_sync(delete_user, user, is_session=True)
     user = db.get(User, 1)
     assert user is None
+
+    # test on_close_pre
+    def get_user(session: Session, user_id: int):
+        return session.get(User, user_id)
+
+    user_id = db.run_sync(get_user, 2, is_session=True, on_close_pre=lambda r: r.id)
+    assert user_id == 2