Source code for sqlobject.tests.test_SQLRelatedJoin

import pytest
from sqlobject import RelatedJoin, SQLObject, SQLRelatedJoin, StringCol, \
    ForeignKey
from sqlobject.sqlbuilder import Alias
from sqlobject.tests.dbtest import setupClass, supports


[docs] class Fighter(SQLObject): class sqlmeta: idName = 'fighter_id' # test on a non-standard way name = StringCol() tourtments = RelatedJoin('Tourtment')
[docs] class Tourtment(SQLObject): class sqlmeta: table = 'competition' # test on a non-standard way name = StringCol() fightersAsList = RelatedJoin('Fighter') fightersAsSResult = SQLRelatedJoin('Fighter')
[docs] def createAllTables(): setupClass(Fighter) setupClass(Tourtment)
[docs] def createData(): createAllTables() # create some tourtments t1 = Tourtment(name='Tourtment #1') t2 = Tourtment(name='Tourtment #2') t3 = Tourtment(name='Tourtment #3') # create some fighters gokou = Fighter(name='gokou') vegeta = Fighter(name='vegeta') gohan = Fighter(name='gohan') trunks = Fighter(name='trunks') # relating them t1.addFighter(gokou) t1.addFighter(vegeta) t1.addFighter(gohan) t2.addFighter(gokou) t2.addFighter(vegeta) t2.addFighter(trunks) t3.addFighter(gohan) t3.addFighter(trunks) return t1, t2, t3, gokou, vegeta, gohan, trunks
[docs] def test_1(): t1, t2, t3, gokou, vegeta, gohan, trunks = createData() for i, j in zip(t1.fightersAsList, t1.fightersAsSResult): assert i is j assert len(t2.fightersAsList) == t2.fightersAsSResult.count()
[docs] class RecursiveGroup(SQLObject): name = StringCol(length=255, unique=True) subgroups = SQLRelatedJoin( 'RecursiveGroup', otherColumn='group_id', intermediateTable='rec_group_map', createRelatedTable=False, )
[docs] class RecGroupMap(SQLObject): recursive_group = ForeignKey('RecursiveGroup') group = ForeignKey('RecursiveGroup')
[docs] def test_rec_group(): setupClass([RecursiveGroup, RecGroupMap]) a = RecursiveGroup(name='a') a1 = RecursiveGroup(name='a1') a.addRecursiveGroup(a1) a2 = RecursiveGroup(name='a2') a.addRecursiveGroup(a2) assert sorted(a.subgroups, key=lambda x: x.name) == [a1, a2] # Please note this is a wrong result! # Do not filter by the table, use alias. See the example below. assert list( a.subgroups.filter(RecursiveGroup.q.name == 'a1').orderBy('name') ) == [a1, a2] rgroupAlias = Alias(RecursiveGroup, '_SO_SQLRelatedJoin_OtherTable') assert list(a.subgroups.filter(rgroupAlias.q.name == 'a1')) == [a1]