add a polymorphic get() test

diff --git a/VERSION b/VERSION
index 267577d47e497a0630bc454b3f74c4fd9a10ced4..2b7c5ae01848a77d95e2792eb83ab605c9aed91a 100644 (file)
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-0.4.1
+0.4.2

diff --git a/test/orm/inheritance/basic.py b/test/orm/inheritance/basic.py
index 32420300f2ddc2910d9dbd38377011f3c587fce5..68d7956cc5461f773f51a20112608c74447f5324 100644 (file)
--- a/test/orm/inheritance/basic.py
+++ b/test/orm/inheritance/basic.py
@@ -121,7 +121,7 @@ class GetTest(ORMTest):
                     assert sess.query(Bar).get(b.id) == b
                     assert sess.query(Bar).get(bl.id) == bl
                     assert sess.query(Blub).get(bl.id) == bl
-
+                    
                 self.assert_sql_count(testbase.db, go, 0)
             else:
                 # this is testing the 'wrong' behavior of using get()
@@ -145,6 +145,7 @@ class GetTest(ORMTest):
                     assert sess.query(Blub).get(bl.id) == bl
 
                 self.assert_sql_count(testbase.db, go, 3)
+            
         test_get.__name__ = name
         return test_get
 

diff --git a/test/orm/inheritance/polymorph.py b/test/orm/inheritance/polymorph.py
index f067b330bd81e26b096dd604351cd69d5698c0a2..e45a11a5427d6bc4f66b7083b48b49de6a0c0f29 100644 (file)
--- a/test/orm/inheritance/polymorph.py
+++ b/test/orm/inheritance/polymorph.py
@@ -254,6 +254,8 @@ def generate_round_trip_test(include_base=False, lazy_relation=True, redefine_co
         c = Company(name='company1')
         c.employees.append(Manager(status='AAB', manager_name='manager1', **{person_attribute_name:'pointy haired boss'}))
         c.employees.append(Engineer(status='BBA', engineer_name='engineer1', primary_language='java', **{person_attribute_name:'dilbert'}))
+        dilbert = c.employees[-1]
+        
         if include_base:
             c.employees.append(Person(status='HHH', **{person_attribute_name:'joesmith'}))
         c.employees.append(Engineer(status='CGG', engineer_name='engineer2', primary_language='python', **{person_attribute_name:'wally'}))
@@ -262,6 +264,11 @@ def generate_round_trip_test(include_base=False, lazy_relation=True, redefine_co
         print session.new
         session.flush()
         session.clear()
+        
+        dilbert = session.query(Person).get(dilbert.person_id)
+        assert getattr(dilbert, person_attribute_name) == 'dilbert'
+        session.clear()
+        
         id = c.company_id
         def go():
             c = session.query(Company).get(id)