diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2016-07-31 00:10:04 -0400 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2016-07-31 00:10:04 -0400 |
commit | 1269b08b1baecc50ed78f5e5facce0a736b125f8 (patch) | |
tree | 3db5d9ad2b5e8a0eb93530f9e9229661c46ac915 | |
parent | cda9defe58f80759f2b40e658fd7b25eb4d92610 (diff) | |
download | sqlalchemy-1269b08b1baecc50ed78f5e5facce0a736b125f8.tar.gz |
Ensure post-__clause_element__() expression are used in Index
The change in Index for 1.1 combined with the fix for
ref #3763 still fails to deliver the correct object
resolved by __clause_element__() to the list of expressions
for compilation. Make sure we use the expression that's
been unwrapped from __clause_element__().
Change-Id: Ie1df8db5090de665048331786f0024d52851923f
Fixes: #3763
-rw-r--r-- | lib/sqlalchemy/sql/schema.py | 4 | ||||
-rw-r--r-- | test/sql/test_metadata.py | 14 |
2 files changed, 17 insertions, 1 deletions
diff --git a/lib/sqlalchemy/sql/schema.py b/lib/sqlalchemy/sql/schema.py index 2c5daa17c..457259d61 100644 --- a/lib/sqlalchemy/sql/schema.py +++ b/lib/sqlalchemy/sql/schema.py @@ -3287,12 +3287,14 @@ class Index(DialectKWArgs, ColumnCollectionMixin, SchemaItem): self.table = None columns = [] + processed_expressions = [] for expr, column, strname, add_element in self.\ _extract_col_expression_collection(expressions): if add_element is not None: columns.append(add_element) + processed_expressions.append(expr) - self.expressions = expressions + self.expressions = processed_expressions self.name = quoted_name(name, kw.pop("quote", None)) self.unique = kw.pop('unique', False) if 'info' in kw: diff --git a/test/sql/test_metadata.py b/test/sql/test_metadata.py index cf7f7628a..af291b842 100644 --- a/test/sql/test_metadata.py +++ b/test/sql/test_metadata.py @@ -2168,6 +2168,20 @@ class ConstraintTest(fixtures.TestBase): eq_(set(t.indexes), set([idx])) + def test_clauseelement_extraction_three(self): + t = Table('t', MetaData(), Column('x', Integer), Column('y', Integer)) + + expr1 = t.c.x + 5 + + class MyThing(object): + def __clause_element__(self): + return expr1 + + idx = Index('bar', MyThing(), t.c.y) + + is_(idx.expressions[0], expr1) + is_(idx.expressions[1], t.c.y) + def test_table_references(self): t1, t2, t3 = self._single_fixture() assert list(t2.c.a.foreign_keys)[0].references(t1) |