Changeset 534 for elixir/trunk/tests

Timestamp:

01/29/11 20:53:10 (16 months ago)

Author:

ged

Message:

- Fixed bad foreign key constraint generated for classes inheriting from a

class with multiple primary keys when using the "multi" inheritance.
Patch from & closes #114.

Files:

• elixir/trunk/tests/test_inherit.py (modified) (2 diffs)

elixir/trunk/tests/test_inherit.py

@@ -8 +8 @@
 def setup():
     metadata.bind = 'sqlite://'
-#    metadata.bind = 'postgres://@/test'
+#    metadata.bind = 'postgresql://@/test'
 #    metadata.bind.echo = True
     elixir.options_defaults['shortnames'] = True
@@ -144 +144 @@
         setup_all()
 
+    def test_multi_pk(self):
+        class A(Entity):
+            using_options(inheritance='multi')
+            firstname = Field(String(50), primary_key=True)
+            lastname = Field(String(50), primary_key=True)
+
+        class B(A):
+            using_options(inheritance='multi')
+
+        setup_all(True)
+
+        b1 = B(firstname='1', lastname='b')
+        b2 = B(firstname='2', lastname='b')
+
+        session.commit()
+
+        b1.delete()
+
+        session.commit()
+
+        assert len(B.query.all()) == 1
+
     def test_multitable_polymorphic_load(self):
         class A(Entity):