From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 16 Jul 2005 07:31:36 +0000 (+0000)
Subject: (no commit message)
X-Git-Tag: rel_0_1_0~878
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=26d396111001c78aaa79f58a25fc4af4c39895d8;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

---

diff --git a/test/mapper.py b/test/mapper.py
index 735264f81c..1391aee240 100644
--- a/test/mapper.py
+++ b/test/mapper.py
@@ -12,12 +12,18 @@ import sqlalchemy.mapper as mapper
 
 class User:
     def __repr__(self):
-        return "User: " + repr(self.user_id) + " " + self.user_name + repr(getattr(self, 'addresses', None))
-
+        return ("User: " + repr(self.user_id) + " " + self.user_name + repr(getattr(self, 'addresses', None)) +
+            repr(getattr(self, 'orders', None))
+            )
+            
 class Address:
     def __repr__(self):
-        return "Address: " + repr(self.user_id) + " " + self.email_address
+        return "Address: " + repr(self.user_id) + " " + repr(self.email_address)
 
+class Order:
+    def __repr__(self):
+        return "Order: " + repr(self.description) + repr(self.isopen)
+        
 class MapperTest(PersistTest):
     
     def setUp(self):
@@ -32,15 +38,31 @@ class MapperTest(PersistTest):
             Column('email_address', VARCHAR(20)),
         )
         
+        self.orders = Table('orders', db,
+            Column('order_id', INT, primary_key = True),
+            Column('user_id', INT),
+            Column('description', VARCHAR(50)),
+            Column('isopen', INT)
+        )
+        
         self.users.build()
         self.users.insert().execute(user_id = 7, user_name = 'jack')
         self.users.insert().execute(user_id = 8, user_name = 'ed')
+        self.users.insert().execute(user_id = 9, user_name = 'fred')
         
         self.addresses.build()
         self.addresses.insert().execute(address_id = 1, user_id = 7, email_address = "jack@bean.com")
         self.addresses.insert().execute(address_id = 2, user_id = 8, email_address = "ed@wood.com")
         self.addresses.insert().execute(address_id = 3, user_id = 8, email_address = "ed@lala.com")
         
+        self.orders.build()
+        self.orders.insert().execute(order_id = 1, user_id = 7, description = 'order 1', isopen=0)
+        self.orders.insert().execute(order_id = 2, user_id = 7, description = 'order 2', isopen=0)
+        self.orders.insert().execute(order_id = 3, user_id = 7, description = 'order 3', isopen=1)
+        self.orders.insert().execute(order_id = 4, user_id = 7, description = 'order 4', isopen=1)
+        self.orders.insert().execute(order_id = 5, user_id = 7, description = 'order 5', isopen=0)
+        
+        
     def testmapper(self):
         m = mapper.Mapper(User, self.users)
         l = m.select()
@@ -53,10 +75,22 @@ class MapperTest(PersistTest):
         ))
         l = m.select()
         print repr(l)
+
+    def testmultieager(self):
+        openorders = alias(self.orders, 'open')
+        closedorders = alias(self.orders, 'closed')
+        m = mapper.Mapper(User, self.users, properties = dict(
+            orders_open = mapper.EagerLoader(mapper.Mapper(Order, openorders), and_(openorders.c.isopen == 1, self.users.c.user_id==openorders.c.user_id)),
+            orders_closed = mapper.EagerLoader(mapper.Mapper(Order, closedorders), and_(closedorders.c.isopen == 0, self.users.c.user_id==closedorders.c.user_id))
+        ))
+        l = m.select()
+        print repr(l)
+
         
     def tearDown(self):
         self.users.drop()
         self.addresses.drop()
+        self.orders.drop()
         
         
 if __name__ == "__main__":