Source code for sqlobject.tests.test_select_through
import pytest
from sqlobject import ForeignKey, SQLMultipleJoin, SQLObject, SQLRelatedJoin, \
    StringCol
from sqlobject.tests.dbtest import inserts, setupClass
# Tests retrieving objects through a join/fk on a selectResults
[docs]class SRThrough1(SQLObject):
    three = ForeignKey('SRThrough3')
    twos = SQLMultipleJoin('SRThrough2', joinColumn='oneID') 
[docs]class SRThrough2(SQLObject):
    one = ForeignKey('SRThrough1')
    threes = SQLRelatedJoin('SRThrough3', addRemoveName='Three') 
[docs]class SRThrough3(SQLObject):
    name = StringCol()
    ones = SQLMultipleJoin('SRThrough1', joinColumn='threeID')
    twos = SQLRelatedJoin('SRThrough2') 
[docs]def setup_module(mod):
    global ones, twos, threes
    setupClass([SRThrough3, SRThrough1, SRThrough2])
    threes = inserts(SRThrough3,
                     [('a',), ('b',), ('c',)],
                     'name')
    ones = inserts(SRThrough1,
                   [(threes[0].id,), (threes[0].id,), (threes[2].id,)],
                   'threeID')
    twos = inserts(SRThrough2,
                   [(ones[0].id,), (ones[1].id,), (ones[2].id,)],
                   'oneID')
    twos[0].addThree(threes[0])
    twos[0].addThree(threes[1]) 
[docs]def testBadRef():
    with pytest.raises(AttributeError):
        threes[0].throughTo.four 
[docs]def testThroughFK():
    assert list(threes[0].ones.throughTo.three) == [threes[0]] 
[docs]def testThroughMultipleJoin():
    assert list(threes[0].ones.throughTo.twos) == [twos[0], twos[1]] 
[docs]def testThroughFKAndJoin():
    assert list(threes[0].ones.throughTo.three.throughTo.twos) == [twos[0]]