From: Sam Bull <aa6bs0@sambull.org>
Date: Sat, 19 Nov 2022 21:29:22 +0000 (+0000)
Subject: Update sessionmakers.py
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=da689f53cdfb8a33d01f5154994f4c61903b1f6f;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Update sessionmakers.py
---

diff --git a/test/ext/mypy/plain_files/sessionmakers.py b/test/ext/mypy/plain_files/sessionmakers.py
index ce9b766385..46e3e0fed8 100644
--- a/test/ext/mypy/plain_files/sessionmakers.py
+++ b/test/ext/mypy/plain_files/sessionmakers.py
@@ -86,3 +86,26 @@ def main() -> None:
     sess = scoped_fac()
     # EXPECTED_TYPE: MySession
     reveal_type(sess)
+
+
+def test_8837_sync() -> None:
+    sm = sessionmaker()
+
+    # EXPECTED_TYPE: sessionmaker[Session]
+    reveal_type(sm)
+
+    session = sm()
+
+    # EXPECTED_TYPE: Session
+    reveal_type(session)
+
+def test_8837_async() -> None:
+    as_= async_sessionmaker()
+
+    # EXPECTED_TYPE: async_sessionmaker[AsyncSession]
+    reveal_type(as_)
+
+    async_session = as_()
+
+    # EXPECTED_TYPE: AsyncSession
+    reveal_type(async_session)