diff options
Diffstat (limited to 'test/orm/abc_inheritance.py')
| -rw-r--r-- | test/orm/abc_inheritance.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/test/orm/abc_inheritance.py b/test/orm/abc_inheritance.py index 766bbc5df..9b2928fbe 100644 --- a/test/orm/abc_inheritance.py +++ b/test/orm/abc_inheritance.py @@ -84,9 +84,14 @@ def produce_test(parent, child, direction): class B(A):pass class C(B):pass - mapper(A, ta, polymorphic_on=abcjoin.c.type, select_table=abcjoin, polymorphic_identity="a") - mapper(B, tb, polymorphic_on=bcjoin.c.type, select_table=bcjoin, polymorphic_identity="b", inherits=A, inherit_condition=atob) - mapper(C, tc, polymorphic_identity="c", inherits=B, inherit_condition=btoc) + mapper(A, ta, polymorphic_on=abcjoin.c.type, select_table=abcjoin, polymorphic_identity="a", ) + mapper(B, tb, polymorphic_on=bcjoin.c.type, select_table=bcjoin, polymorphic_identity="b", inherits=A, inherit_condition=atob,) + mapper(C, tc, polymorphic_identity="c", inherits=B, inherit_condition=btoc, ) + + #print "KEYS:" + #print [c.key for c in class_mapper(A).primary_key] + #print [c.key for c in class_mapper(B).primary_key] + #print [c.key for c in class_mapper(C).primary_key] parent_mapper = class_mapper({ta:A, tb:B, tc:C}[parent_table]) child_mapper = class_mapper({ta:A, tb:B, tc:C}[child_table]) @@ -121,13 +126,14 @@ def produce_test(parent, child, direction): sess.clear() # assert result via direct get() of parent object - result = sess.query(parent_class).get(parent_obj.id) + # TODO: dual PK here is a temporary workaround until #185 is fixed again + result = sess.query(parent_class).get([parent_obj.id, parent_obj.id]) assert result.id == parent_obj.id assert result.collection[0].id == child_obj.id if direction == ONETOMANY: assert result.collection[1].id == child2.id elif direction == MANYTOONE: - result2 = sess.query(parent_class).get(parent2.id) + result2 = sess.query(parent_class).get([parent2.id, parent2.id]) assert result2.id == parent2.id assert result2.collection[0].id == child_obj.id |