mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-06 08:56:51 -04:00
Mike Bayer
130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
from testbase import PersistTest, AssertMixin
|
|
import testbase
|
|
import unittest, sys, os
|
|
from sqlalchemy import *
|
|
import datetime
|
|
|
|
class LazyTest(AssertMixin):
|
|
def setUpAll(self):
|
|
global info_table, data_table, rel_table
|
|
engine = testbase.db
|
|
info_table = Table('infos', engine,
|
|
Column('pk', Integer, primary_key=True),
|
|
Column('info', String))
|
|
|
|
data_table = Table('data', engine,
|
|
Column('data_pk', Integer, primary_key=True),
|
|
Column('info_pk', Integer, ForeignKey(info_table.c.pk)),
|
|
Column('timeval', Integer),
|
|
Column('data_val', String))
|
|
|
|
rel_table = Table('rels', engine,
|
|
Column('rel_pk', Integer, primary_key=True),
|
|
Column('info_pk', Integer, ForeignKey(info_table.c.pk)),
|
|
Column('start', Integer),
|
|
Column('finish', Integer))
|
|
|
|
|
|
info_table.create()
|
|
rel_table.create()
|
|
data_table.create()
|
|
info_table.insert().execute(
|
|
{'pk':1, 'info':'pk_1_info'},
|
|
{'pk':2, 'info':'pk_2_info'},
|
|
{'pk':3, 'info':'pk_3_info'},
|
|
{'pk':4, 'info':'pk_4_info'},
|
|
{'pk':5, 'info':'pk_5_info'})
|
|
|
|
rel_table.insert().execute(
|
|
{'rel_pk':1, 'info_pk':1, 'start':10, 'finish':19},
|
|
{'rel_pk':2, 'info_pk':1, 'start':100, 'finish':199},
|
|
{'rel_pk':3, 'info_pk':2, 'start':20, 'finish':29},
|
|
{'rel_pk':4, 'info_pk':3, 'start':13, 'finish':23},
|
|
{'rel_pk':5, 'info_pk':5, 'start':15, 'finish':25})
|
|
|
|
data_table.insert().execute(
|
|
{'data_pk':1, 'info_pk':1, 'timeval':11, 'data_val':'11_data'},
|
|
{'data_pk':2, 'info_pk':1, 'timeval':9, 'data_val':'9_data'},
|
|
{'data_pk':3, 'info_pk':1, 'timeval':13, 'data_val':'13_data'},
|
|
{'data_pk':4, 'info_pk':2, 'timeval':23, 'data_val':'23_data'},
|
|
{'data_pk':5, 'info_pk':2, 'timeval':13, 'data_val':'13_data'},
|
|
{'data_pk':6, 'info_pk':1, 'timeval':15, 'data_val':'15_data'})
|
|
|
|
|
|
def tearDownAll(self):
|
|
data_table.drop()
|
|
rel_table.drop()
|
|
info_table.drop()
|
|
|
|
def setUp(self):
|
|
clear_mappers()
|
|
|
|
def testone(self):
|
|
"""tests a lazy load which has multiple join conditions, including two that are against
|
|
the same column in the child table"""
|
|
class Information(object):
|
|
pass
|
|
|
|
class Relation(object):
|
|
pass
|
|
|
|
class Data(object):
|
|
pass
|
|
|
|
# Create the basic mappers, with no frills or modifications
|
|
Information.mapper = mapper(Information, info_table)
|
|
Data.mapper = mapper(Data, data_table)
|
|
Relation.mapper = mapper(Relation, rel_table)
|
|
|
|
Relation.mapper.add_property('datas', relation(Data.mapper,
|
|
primaryjoin=and_(Relation.c.info_pk==Data.c.info_pk,
|
|
Data.c.timeval >= Relation.c.start,
|
|
Data.c.timeval <= Relation.c.finish
|
|
),
|
|
foreignkey=Data.c.info_pk))
|
|
|
|
Information.mapper.add_property('rels', relation(Relation.mapper))
|
|
|
|
info = Information.mapper.get(1)
|
|
assert info
|
|
assert len(info.rels) == 2
|
|
assert len(info.rels[0].datas) == 3
|
|
|
|
def testtwo(self):
|
|
"""same thing, but reversing the order of the cols in the join"""
|
|
class Information(object):
|
|
pass
|
|
|
|
class Relation(object):
|
|
pass
|
|
|
|
class Data(object):
|
|
pass
|
|
|
|
# Create the basic mappers, with no frills or modifications
|
|
Information.mapper = mapper(Information, info_table)
|
|
Data.mapper = mapper(Data, data_table)
|
|
Relation.mapper = mapper(Relation, rel_table)
|
|
|
|
Relation.mapper.add_property('datas', relation(Data.mapper,
|
|
primaryjoin=and_(Relation.c.info_pk==Data.c.info_pk,
|
|
Relation.c.start <= Data.c.timeval,
|
|
Relation.c.finish >= Data.c.timeval,
|
|
# Data.c.timeval >= Relation.c.start,
|
|
# Data.c.timeval <= Relation.c.finish
|
|
),
|
|
foreignkey=Data.c.info_pk))
|
|
|
|
Information.mapper.add_property('rels', relation(Relation.mapper))
|
|
|
|
info = Information.mapper.get(1)
|
|
assert info
|
|
assert len(info.rels) == 2
|
|
assert len(info.rels[0].datas) == 3
|
|
|
|
|
|
if __name__ == "__main__":
|
|
testbase.main()
|
|
|
|
|