From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Tue, 21 Dec 2010 21:01:38 +0000 (-0500)
Subject: - merged the test for [ticket:1997], r1b1f195483aa , into
X-Git-Tag: rel_0_7b1~128
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=b0f48ca2a910e0a8527b9640c56cc3f342f4f31d;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

- merged the test for [ticket:1997], r1b1f195483aa , into
0.7.  the fix itself no longer applies.
---

diff --git a/CHANGES b/CHANGES
index 300b6b4bb7..5e79a27aec 100644
--- a/CHANGES
+++ b/CHANGES
@@ -29,6 +29,17 @@ CHANGES
     that weren't previously saved in the "mutable changes"
     dictionary.
 
+  - Fixed bug which prevented composite mapped
+    attributes from being used on a mapped select statement. 
+    [ticket:1997]. Note the workings of composite are slated to 
+    change significantly in 0.7.
+
+  - active_history flag also added to composite(). 
+    The flag has no effect in 0.6, but is instead 
+    a placeholder flag for forwards compatibility,
+    as it applies in 0.7 for composites.
+    [ticket:1976]
+
   - Fixed uow bug whereby expired objects passed to 
     Session.delete() would not have unloaded references 
     or collections taken into account when deleting
@@ -59,12 +70,6 @@ CHANGES
     always load the "old" value, so that it's available to
     attributes.get_history(). [ticket:1961]
     
-  - active_history flag also added to composite(). 
-    The flag has no effect in 0.6, but is instead 
-    a placeholder flag for forwards compatibility,
-    as it will be needed in 0.7 for composites.
-    [ticket:1976]
-
   - Query.get() will raise if the number of params
     in a composite key is too large, as well as too 
     small. [ticket:1977]
diff --git a/test/orm/test_composites.py b/test/orm/test_composites.py
index 54a1d571a2..558a80b15f 100644
--- a/test/orm/test_composites.py
+++ b/test/orm/test_composites.py
@@ -1,7 +1,8 @@
 from test.lib.testing import assert_raises, assert_raises_message
 import sqlalchemy as sa
 from test.lib import testing
-from sqlalchemy import MetaData, Integer, String, ForeignKey, func, util
+from sqlalchemy import MetaData, Integer, String, ForeignKey, func, \
+    util, select
 from test.lib.schema import Table, Column
 from sqlalchemy.orm import mapper, relationship, backref, \
     class_mapper,  \
@@ -335,3 +336,79 @@ class DefaultsTest(_base.MappedTest):
         
         assert f1.foob == FBComposite(2, 5, 15, None)
     
+class MappedSelectTest(_base.MappedTest):
+    @classmethod
+    def define_tables(cls, metadata):
+        Table('descriptions', metadata,
+            Column('id', Integer, primary_key=True, 
+                            test_needs_autoincrement=True),
+            Column('d1', String(20)),
+            Column('d2', String(20)),
+        )
+
+        Table('values', metadata,
+            Column('id', Integer, primary_key=True),
+            Column('description_id', Integer, ForeignKey('descriptions.id'),
+                   nullable=False),
+            Column('v1', String(20)),
+            Column('v2', String(20)),
+        )
+    
+    @classmethod
+    @testing.resolve_artifact_names
+    def setup_mappers(cls):
+        class Descriptions(_base.BasicEntity):
+            pass
+
+        class Values(_base.BasicEntity):
+            pass
+
+        class CustomValues(_base.BasicEntity, list):
+            def __init__(self, *args):
+                self.extend(args)
+
+            def __composite_values__(self):
+                return self
+
+        desc_values = select(
+            [values, descriptions.c.d1, descriptions.c.d2],
+            descriptions.c.id == values.c.description_id
+        ).alias('descriptions_values') 
+
+        mapper(Descriptions, descriptions, properties={
+            'values': relationship(Values, lazy='dynamic'),
+            'custom_descriptions': composite(
+                                CustomValues,
+                                        descriptions.c.d1,
+                                        descriptions.c.d2),
+
+        })
+
+        mapper(Values, desc_values, properties={
+            'custom_values': composite(CustomValues, 
+                                            desc_values.c.v1,
+                                            desc_values.c.v2),
+
+        })
+        
+    @testing.resolve_artifact_names
+    def test_set_composite_attrs_via_selectable(self):
+        session = Session()
+        d = Descriptions(
+            custom_descriptions = CustomValues('Color', 'Number'),
+            values =[
+                Values(custom_values = CustomValues('Red', '5')),
+                Values(custom_values=CustomValues('Blue', '1'))
+            ]
+        )
+
+        session.add(d)
+        session.commit()
+        eq_(
+            testing.db.execute(descriptions.select()).fetchall(),
+            [(1, u'Color', u'Number')]
+        )
+        eq_(
+            testing.db.execute(values.select()).fetchall(),
+            [(1, 1, u'Red', u'5'), (2, 1, u'Blue', u'1')]
+        )