Fix with_expression() cache leak; don't adapt singletons

Fixed a cache leak involving the :func:`_orm.with_expression` loader
option, where the given SQL expression would not be correctly considered as
part of the cache key.

Additionally, fixed regression involving the corresponding
:func:`_orm.query_expression` feature. While the bug technically exists in
1.3 as well, it was not exposed until 1.4. The "default expr" value of
``null()`` would be rendered when not needed, and additionally was also not
adapted correctly when the ORM rewrites statements such as when using
joined eager loading. The fix ensures "singleton" expressions like ``NULL``
and ``true`` aren't "adapted" to refer to columns in ORM statements, and
additionally ensures that a :func:`_orm.query_expression` with no default
expression doesn't render in the statement if a
:func:`_orm.with_expression` isn't used.

Fixes: #6259
Change-Id: I5a70bc12dadad125bbc4324b64048c8d4a18916c

4 files changed, 178 insertions, 2 deletions

diff --git a/test/orm/test_cache_key.py b/test/orm/test_cache_key.py
index cd06ce56a..d120b05c0 100644
--- a/test/orm/test_cache_key.py
+++ b/test/orm/test_cache_key.py
@@ -1,9 +1,12 @@
 import random
 
+from sqlalchemy import func
 from sqlalchemy import inspect
+from sqlalchemy import null
 from sqlalchemy import select
 from sqlalchemy import testing
 from sqlalchemy import text
+from sqlalchemy import true
 from sqlalchemy.orm import aliased
 from sqlalchemy.orm import defaultload
 from sqlalchemy.orm import defer
@@ -16,6 +19,7 @@ from sqlalchemy.orm import relationship
 from sqlalchemy.orm import selectinload
 from sqlalchemy.orm import Session
 from sqlalchemy.orm import subqueryload
+from sqlalchemy.orm import with_expression
 from sqlalchemy.orm import with_loader_criteria
 from sqlalchemy.orm import with_polymorphic
 from sqlalchemy.sql.base import CacheableOptions
@@ -72,6 +76,23 @@ class CacheKeyTest(CacheKeyFixture, _fixtures.FixtureTest):
             compare_values=True,
         )
 
+    def test_query_expr(self):
+        (User,) = self.classes("User")
+
+        self._run_cache_key_fixture(
+            lambda: (
+                with_expression(User.name, true()),
+                with_expression(User.name, null()),
+                with_expression(User.name, func.foobar()),
+                with_expression(User.name, User.name == "test"),
+                Load(User).with_expression(User.name, true()),
+                Load(User).with_expression(User.name, null()),
+                Load(User).with_expression(User.name, func.foobar()),
+                Load(User).with_expression(User.name, User.name == "test"),
+            ),
+            compare_values=True,
+        )
+
     def test_loader_criteria(self):
         User, Address = self.classes("User", "Address")
 
diff --git a/test/orm/test_core_compilation.py b/test/orm/test_core_compilation.py
index a53b15bcb..cbc069a85 100644
--- a/test/orm/test_core_compilation.py
+++ b/test/orm/test_core_compilation.py
@@ -3,6 +3,7 @@ from sqlalchemy import exc
 from sqlalchemy import func
 from sqlalchemy import insert
 from sqlalchemy import literal_column
+from sqlalchemy import null
 from sqlalchemy import or_
 from sqlalchemy import select
 from sqlalchemy import testing
@@ -17,6 +18,7 @@ from sqlalchemy.orm import query_expression
 from sqlalchemy.orm import relationship
 from sqlalchemy.orm import with_expression
 from sqlalchemy.orm import with_polymorphic
+from sqlalchemy.sql import and_
 from sqlalchemy.sql import sqltypes
 from sqlalchemy.sql.selectable import Join as core_join
 from sqlalchemy.sql.selectable import LABEL_STYLE_DISAMBIGUATE_ONLY
@@ -405,12 +407,50 @@ class ExtraColsTest(QueryTest, AssertsCompiledSQL):
             self.tables.users,
             self.classes.User,
         )
+        addresses, Address = (self.tables.addresses, self.classes.Address)
 
         mapper(
             User,
             users,
-            properties=util.OrderedDict([("value", query_expression())]),
+            properties=util.OrderedDict(
+                [
+                    ("value", query_expression()),
+                ]
+            ),
         )
+        mapper(Address, addresses)
+
+        return User
+
+    @testing.fixture
+    def query_expression_w_joinedload_fixture(self):
+        users, User = (
+            self.tables.users,
+            self.classes.User,
+        )
+        addresses, Address = (self.tables.addresses, self.classes.Address)
+
+        mapper(
+            User,
+            users,
+            properties=util.OrderedDict(
+                [
+                    ("value", query_expression()),
+                    (
+                        "addresses",
+                        relationship(
+                            Address,
+                            primaryjoin=and_(
+                                addresses.c.user_id == users.c.id,
+                                addresses.c.email_address != None,
+                            ),
+                        ),
+                    ),
+                ]
+            ),
+        )
+        mapper(Address, addresses)
+
         return User
 
     @testing.fixture
@@ -528,6 +568,49 @@ class ExtraColsTest(QueryTest, AssertsCompiledSQL):
             "users.name || :name_1 AS foo FROM users) AS anon_1",
         )
 
+    def test_with_expr_three(self, query_expression_w_joinedload_fixture):
+        """test :ticket:`6259`"""
+        User = query_expression_w_joinedload_fixture
+
+        stmt = select(User).options(joinedload(User.addresses)).limit(1)
+
+        # test that the outer IS NULL is rendered
+        # test that the inner query does not include a NULL default
+        self.assert_compile(
+            stmt,
+            "SELECT anon_1.id, anon_1.name, addresses_1.id AS id_1, "
+            "addresses_1.user_id, addresses_1.email_address FROM "
+            "(SELECT users.id AS id, users.name AS name FROM users "
+            "LIMIT :param_1) AS anon_1 LEFT OUTER "
+            "JOIN addresses AS addresses_1 ON addresses_1.user_id = anon_1.id "
+            "AND addresses_1.email_address IS NOT NULL",
+        )
+
+    def test_with_expr_four(self, query_expression_w_joinedload_fixture):
+        """test :ticket:`6259`"""
+        User = query_expression_w_joinedload_fixture
+
+        stmt = (
+            select(User)
+            .options(
+                with_expression(User.value, null()), joinedload(User.addresses)
+            )
+            .limit(1)
+        )
+
+        # test that the outer IS NULL is rendered, not adapted
+        # test that the inner query includes the NULL we asked for
+        self.assert_compile(
+            stmt,
+            "SELECT anon_2.anon_1, anon_2.id, anon_2.name, "
+            "addresses_1.id AS id_1, addresses_1.user_id, "
+            "addresses_1.email_address FROM (SELECT NULL AS anon_1, "
+            "users.id AS id, users.name AS name FROM users LIMIT :param_1) "
+            "AS anon_2 LEFT OUTER JOIN addresses AS addresses_1 "
+            "ON addresses_1.user_id = anon_2.id "
+            "AND addresses_1.email_address IS NOT NULL",
+        )
+
     def test_joinedload_outermost(self, plain_fixture):
         User, Address = plain_fixture
 
diff --git a/test/orm/test_deferred.py b/test/orm/test_deferred.py
index d528cb935..decb456c6 100644
--- a/test/orm/test_deferred.py
+++ b/test/orm/test_deferred.py
@@ -2,6 +2,7 @@ import sqlalchemy as sa
 from sqlalchemy import ForeignKey
 from sqlalchemy import func
 from sqlalchemy import Integer
+from sqlalchemy import null
 from sqlalchemy import select
 from sqlalchemy import String
 from sqlalchemy import testing
@@ -1824,7 +1825,7 @@ class WithExpressionTest(fixtures.DeclarativeMappedTest):
         A = self.classes.A
 
         s = fixture_session()
-        a1 = s.query(A).first()
+        a1 = s.query(A).options(with_expression(A.my_expr, null())).first()
 
         def go():
             eq_(a1.my_expr, None)
diff --git a/test/sql/test_external_traversal.py b/test/sql/test_external_traversal.py
index e7c6cccca..e1490adfd 100644
--- a/test/sql/test_external_traversal.py
+++ b/test/sql/test_external_traversal.py
@@ -12,11 +12,13 @@ from sqlalchemy import Integer
 from sqlalchemy import literal
 from sqlalchemy import literal_column
 from sqlalchemy import MetaData
+from sqlalchemy import null
 from sqlalchemy import select
 from sqlalchemy import String
 from sqlalchemy import Table
 from sqlalchemy import testing
 from sqlalchemy import text
+from sqlalchemy import true
 from sqlalchemy import tuple_
 from sqlalchemy import union
 from sqlalchemy.sql import ClauseElement
@@ -402,6 +404,75 @@ class ClauseTest(fixtures.TestBase, AssertsCompiledSQL):
             select(f), "SELECT t1_1.col1 * :col1_1 AS anon_1 FROM t1 AS t1_1"
         )
 
+    @testing.combinations((null(),), (true(),))
+    def test_dont_adapt_singleton_elements(self, elem):
+        """test :ticket:`6259`"""
+        t1 = table("t1", column("c1"))
+
+        stmt = select(t1.c.c1, elem)
+
+        wherecond = t1.c.c1.is_(elem)
+
+        subq = stmt.subquery()
+
+        adapted_wherecond = sql_util.ClauseAdapter(subq).traverse(wherecond)
+        stmt = select(subq).where(adapted_wherecond)
+
+        self.assert_compile(
+            stmt,
+            "SELECT anon_1.c1, anon_1.anon_2 FROM (SELECT t1.c1 AS c1, "
+            "%s AS anon_2 FROM t1) AS anon_1 WHERE anon_1.c1 IS %s"
+            % (str(elem), str(elem)),
+            dialect="default_enhanced",
+        )
+
+    def test_adapt_funcs_etc_on_identity_one(self):
+        """Adapting to a function etc. will adapt if its on identity"""
+        t1 = table("t1", column("c1"))
+
+        elem = func.foobar()
+
+        stmt = select(t1.c.c1, elem)
+
+        wherecond = t1.c.c1 == elem
+
+        subq = stmt.subquery()
+
+        adapted_wherecond = sql_util.ClauseAdapter(subq).traverse(wherecond)
+        stmt = select(subq).where(adapted_wherecond)
+
+        self.assert_compile(
+            stmt,
+            "SELECT anon_1.c1, anon_1.foobar_1 FROM (SELECT t1.c1 AS c1, "
+            "foobar() AS foobar_1 FROM t1) AS anon_1 "
+            "WHERE anon_1.c1 = anon_1.foobar_1",
+            dialect="default_enhanced",
+        )
+
+    def test_adapt_funcs_etc_on_identity_two(self):
+        """Adapting to a function etc. will not adapt if they are different"""
+        t1 = table("t1", column("c1"))
+
+        elem = func.foobar()
+        elem2 = func.foobar()
+
+        stmt = select(t1.c.c1, elem)
+
+        wherecond = t1.c.c1 == elem2
+
+        subq = stmt.subquery()
+
+        adapted_wherecond = sql_util.ClauseAdapter(subq).traverse(wherecond)
+        stmt = select(subq).where(adapted_wherecond)
+
+        self.assert_compile(
+            stmt,
+            "SELECT anon_1.c1, anon_1.foobar_1 FROM (SELECT t1.c1 AS c1, "
+            "foobar() AS foobar_1 FROM t1) AS anon_1 "
+            "WHERE anon_1.c1 = foobar()",
+            dialect="default_enhanced",
+        )
+
     def test_join(self):
         clause = t1.join(t2, t1.c.col2 == t2.c.col2)
         c1 = str(clause)