summaryrefslogtreecommitdiff
path: root/test/sql
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2023-03-22 11:56:04 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2023-03-22 11:56:04 -0400
commitd856f640b5803d52fa702a750e504990e80d8724 (patch)
tree0a53facf650c521086333dbf380d6f59a4aae777 /test/sql
parentce4e1770f01a6c19d3a55621529e3f98607d7824 (diff)
downloadsqlalchemy-d856f640b5803d52fa702a750e504990e80d8724.tar.gz
use clone, not constructor, in BindParameter.render_literal_execute()
Fixed issue where the :meth:`_sql.BindParameter.render_literal_execute` method would fail when called on a parameter that also had ORM annotations associated with it. In practice, this would be observed as a failure of SQL compilation when using some combinations of a dialect that uses "FETCH FIRST" such as Oracle along with a :class:`_sql.Select` construct that uses :meth:`_sql.Select.limit`, within some ORM contexts, including if the statement were embedded within a relationship primaryjoin expression. Fixes: #9526 Change-Id: I2f512b6760a90293d274e60b06a891f10b276ecc
Diffstat (limited to 'test/sql')
-rw-r--r--test/sql/test_external_traversal.py66
1 files changed, 66 insertions, 0 deletions
diff --git a/test/sql/test_external_traversal.py b/test/sql/test_external_traversal.py
index b8f6e5685..e474e75d7 100644
--- a/test/sql/test_external_traversal.py
+++ b/test/sql/test_external_traversal.py
@@ -33,6 +33,7 @@ from sqlalchemy.sql import util as sql_util
from sqlalchemy.sql import visitors
from sqlalchemy.sql.elements import _clone
from sqlalchemy.sql.expression import _from_objects
+from sqlalchemy.sql.util import _deep_annotate
from sqlalchemy.sql.visitors import ClauseVisitor
from sqlalchemy.sql.visitors import cloned_traverse
from sqlalchemy.sql.visitors import CloningVisitor
@@ -508,6 +509,71 @@ class ClauseTest(fixtures.TestBase, AssertsCompiledSQL):
self.assert_compile(adapted, expected)
+ @testing.variation("annotate", [True, False])
+ def test_bindparam_render_literal_execute(self, annotate):
+ """test #9526"""
+
+ bp = bindparam("some_value")
+
+ if annotate:
+ bp = bp._annotate({"foo": "bar"})
+
+ bp = bp.render_literal_execute()
+ self.assert_compile(
+ column("q") == bp, "q = __[POSTCOMPILE_some_value]"
+ )
+
+ @testing.variation("limit_type", ["limit", "fetch"])
+ @testing.variation("dialect", ["default", "oracle"])
+ def test_annotated_fetch(self, limit_type: testing.Variation, dialect):
+ """test #9526"""
+
+ if limit_type.limit:
+ stmt = select(column("q")).limit(1)
+ elif limit_type.fetch:
+ stmt = select(column("q")).fetch(1)
+ else:
+ limit_type.fail()
+
+ stmt = _deep_annotate(stmt, {"foo": "bar"})
+
+ if limit_type.limit:
+ if dialect.default:
+ self.assert_compile(
+ stmt,
+ "SELECT q LIMIT :param_1",
+ use_literal_execute_for_simple_int=True,
+ dialect=dialect.name,
+ )
+ elif dialect.oracle:
+ self.assert_compile(
+ stmt,
+ "SELECT q FROM DUAL FETCH FIRST "
+ "__[POSTCOMPILE_param_1] ROWS ONLY",
+ dialect=dialect.name,
+ )
+ else:
+ dialect.fail()
+ elif limit_type.fetch:
+ if dialect.default:
+ self.assert_compile(
+ stmt,
+ "SELECT q FETCH FIRST __[POSTCOMPILE_param_1] ROWS ONLY",
+ use_literal_execute_for_simple_int=True,
+ dialect=dialect.name,
+ )
+ elif dialect.oracle:
+ self.assert_compile(
+ stmt,
+ "SELECT q FROM DUAL FETCH FIRST "
+ "__[POSTCOMPILE_param_1] ROWS ONLY",
+ dialect=dialect.name,
+ )
+ else:
+ dialect.fail()
+ else:
+ limit_type.fail()
+
@testing.combinations((null(),), (true(),))
def test_dont_adapt_singleton_elements(self, elem):
"""test :ticket:`6259`"""