Merge "generate stubs for func known functions" into main

1 files changed, 119 insertions, 0 deletions

diff --git a/test/ext/mypy/plain_files/functions.py b/test/ext/mypy/plain_files/functions.py
new file mode 100644
index 000000000..ecd404010
--- /dev/null
+++ b/test/ext/mypy/plain_files/functions.py
@@ -0,0 +1,119 @@
+"""this file is generated by tools/generate_sql_functions.py"""
+
+from sqlalchemy import column
+from sqlalchemy import func
+from sqlalchemy import select
+
+# START GENERATED FUNCTION TYPING TESTS
+
+# code within this block is **programmatically,
+# statically generated** by tools/generate_sql_functions.py
+
+stmt1 = select(func.char_length(column("x")))
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*int\]\]
+reveal_type(stmt1)
+
+
+stmt2 = select(func.concat())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*str\]\]
+reveal_type(stmt2)
+
+
+stmt3 = select(func.count(column("x")))
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*int\]\]
+reveal_type(stmt3)
+
+
+stmt4 = select(func.cume_dist())
+
+# EXPECTED_RE_TYPE: .*Select\[Any\]
+reveal_type(stmt4)
+
+
+stmt5 = select(func.current_date())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*date\]\]
+reveal_type(stmt5)
+
+
+stmt6 = select(func.current_time())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*time\]\]
+reveal_type(stmt6)
+
+
+stmt7 = select(func.current_timestamp())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*datetime\]\]
+reveal_type(stmt7)
+
+
+stmt8 = select(func.current_user())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*str\]\]
+reveal_type(stmt8)
+
+
+stmt9 = select(func.dense_rank())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*int\]\]
+reveal_type(stmt9)
+
+
+stmt10 = select(func.localtime())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*datetime\]\]
+reveal_type(stmt10)
+
+
+stmt11 = select(func.localtimestamp())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*datetime\]\]
+reveal_type(stmt11)
+
+
+stmt12 = select(func.next_value(column("x")))
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*int\]\]
+reveal_type(stmt12)
+
+
+stmt13 = select(func.now())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*datetime\]\]
+reveal_type(stmt13)
+
+
+stmt14 = select(func.percent_rank())
+
+# EXPECTED_RE_TYPE: .*Select\[Any\]
+reveal_type(stmt14)
+
+
+stmt15 = select(func.rank())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*int\]\]
+reveal_type(stmt15)
+
+
+stmt16 = select(func.session_user())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*str\]\]
+reveal_type(stmt16)
+
+
+stmt17 = select(func.sysdate())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*datetime\]\]
+reveal_type(stmt17)
+
+
+stmt18 = select(func.user())
+
+# EXPECTED_RE_TYPE: .*Select\[Tuple\[.*str\]\]
+reveal_type(stmt18)
+
+# END GENERATED FUNCTION TYPING TESTS