Fixed typing of limit, offset and fetch to allow ``None``.

Fixes: #9183
Change-Id: I1ac3e3698034826122ea8a0cdc9f8f55a10ed6c1

2 files changed, 17 insertions, 1 deletions

diff --git a/test/ext/mypy/plain_files/session.py b/test/ext/mypy/plain_files/session.py
index 636e3854a..9106b9016 100644
--- a/test/ext/mypy/plain_files/session.py
+++ b/test/ext/mypy/plain_files/session.py
@@ -89,4 +89,8 @@ with Session(e) as sess:
         # EXPECTED_TYPE: User
         reveal_type(uobj1)
 
+    sess.query(User).limit(None).offset(None).limit(10).offset(10).limit(
+        User.id
+    ).offset(User.id)
+
 # more result tests in typed_results.py
diff --git a/test/ext/mypy/plain_files/typed_queries.py b/test/ext/mypy/plain_files/typed_queries.py
index fb988c985..3e67a7132 100644
--- a/test/ext/mypy/plain_files/typed_queries.py
+++ b/test/ext/mypy/plain_files/typed_queries.py
@@ -48,7 +48,19 @@ def t_select_1() -> None:
 
 
 def t_select_2() -> None:
-    stmt = select(User).filter(User.id == 5)
+    stmt = (
+        select(User)
+        .filter(User.id == 5)
+        .limit(1)
+        .offset(3)
+        .offset(None)
+        .limit(None)
+        .limit(User.id)
+        .offset(User.id)
+        .fetch(1)
+        .fetch(None)
+        .fetch(User.id)
+    )
 
     # EXPECTED_TYPE: Select[Tuple[User]]
     reveal_type(stmt)