mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-29 12:06:28 -04:00
Khairi Hafsham
772374735d
tested using pycodestyle version 2.2.0 Fixes: #3885 Change-Id: I5df43adc3aefe318f9eeab72a078247a548ec566 Pull-request: https://github.com/zzzeek/sqlalchemy/pull/343
98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
from sqlalchemy import *
|
|
from sqlalchemy import util
|
|
from sqlalchemy.orm import *
|
|
|
|
from sqlalchemy.testing.util import function_named
|
|
from sqlalchemy.testing import fixtures
|
|
from test.orm import _fixtures
|
|
from sqlalchemy.testing import eq_
|
|
from sqlalchemy.testing.schema import Table, Column
|
|
|
|
|
|
class ABCTest(fixtures.MappedTest):
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
global a, b, c
|
|
a = Table('a', metadata,
|
|
Column('id', Integer, primary_key=True,
|
|
test_needs_autoincrement=True),
|
|
Column('adata', String(30)),
|
|
Column('type', String(30)),
|
|
)
|
|
b = Table('b', metadata,
|
|
Column('id', Integer, ForeignKey('a.id'), primary_key=True),
|
|
Column('bdata', String(30)))
|
|
c = Table('c', metadata,
|
|
Column('id', Integer, ForeignKey('b.id'), primary_key=True),
|
|
Column('cdata', String(30)))
|
|
|
|
def _make_test(fetchtype):
|
|
def test_roundtrip(self):
|
|
class A(fixtures.ComparableEntity):
|
|
pass
|
|
|
|
class B(A):
|
|
pass
|
|
|
|
class C(B):
|
|
pass
|
|
|
|
if fetchtype == 'union':
|
|
abc = a.outerjoin(b).outerjoin(c)
|
|
bc = a.join(b).outerjoin(c)
|
|
else:
|
|
abc = bc = None
|
|
|
|
mapper(A, a, with_polymorphic=('*', abc),
|
|
polymorphic_on=a.c.type, polymorphic_identity='a')
|
|
mapper(B, b, with_polymorphic=('*', bc),
|
|
inherits=A, polymorphic_identity='b')
|
|
mapper(C, c, inherits=B, polymorphic_identity='c')
|
|
|
|
a1 = A(adata='a1')
|
|
b1 = B(bdata='b1', adata='b1')
|
|
b2 = B(bdata='b2', adata='b2')
|
|
b3 = B(bdata='b3', adata='b3')
|
|
c1 = C(cdata='c1', bdata='c1', adata='c1')
|
|
c2 = C(cdata='c2', bdata='c2', adata='c2')
|
|
c3 = C(cdata='c2', bdata='c2', adata='c2')
|
|
|
|
sess = create_session()
|
|
for x in (a1, b1, b2, b3, c1, c2, c3):
|
|
sess.add(x)
|
|
sess.flush()
|
|
sess.expunge_all()
|
|
|
|
# for obj in sess.query(A).all():
|
|
# print obj
|
|
eq_([A(adata='a1'),
|
|
B(bdata='b1', adata='b1'),
|
|
B(bdata='b2', adata='b2'),
|
|
B(bdata='b3', adata='b3'),
|
|
C(cdata='c1', bdata='c1', adata='c1'),
|
|
C(cdata='c2', bdata='c2', adata='c2'),
|
|
C(cdata='c2', bdata='c2', adata='c2')],
|
|
sess.query(A).order_by(A.id).all())
|
|
|
|
eq_([
|
|
B(bdata='b1', adata='b1'),
|
|
B(bdata='b2', adata='b2'),
|
|
B(bdata='b3', adata='b3'),
|
|
C(cdata='c1', bdata='c1', adata='c1'),
|
|
C(cdata='c2', bdata='c2', adata='c2'),
|
|
C(cdata='c2', bdata='c2', adata='c2'),
|
|
], sess.query(B).order_by(A.id).all())
|
|
|
|
eq_([
|
|
C(cdata='c1', bdata='c1', adata='c1'),
|
|
C(cdata='c2', bdata='c2', adata='c2'),
|
|
C(cdata='c2', bdata='c2', adata='c2'),
|
|
], sess.query(C).order_by(A.id).all())
|
|
|
|
test_roundtrip = function_named(
|
|
test_roundtrip, 'test_%s' % fetchtype)
|
|
return test_roundtrip
|
|
|
|
test_union = _make_test('union')
|
|
test_none = _make_test('none')
|