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

class SRThrough1(SQLObject):

three = ForeignKey('SRThrough3')

twos = SQLMultipleJoin('SRThrough2', joinColumn='oneID')

class SRThrough2(SQLObject):

one = ForeignKey('SRThrough1')

threes = SQLRelatedJoin('SRThrough3', addRemoveName='Three')

class SRThrough3(SQLObject):

name = StringCol()

ones = SQLMultipleJoin('SRThrough1', joinColumn='threeID')

twos = SQLRelatedJoin('SRThrough2')

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])

def testBadRef():

with pytest.raises(AttributeError):

threes[0].throughTo.four

def testThroughFK():

assert list(threes[0].ones.throughTo.three) == [threes[0]]

def testThroughMultipleJoin():

assert list(threes[0].ones.throughTo.twos) == [twos[0], twos[1]]

def testThroughRelatedJoin():

assert list(threes[0].twos.throughTo.threes) == [threes[0], threes[1]]

assert list(

SRThrough3.select(SRThrough3.q.id == threes[0].id).throughTo.twos) == \

list(threes[0].twos)

def testThroughFKAndJoin():

assert list(threes[0].ones.throughTo.three.throughTo.twos) == [twos[0]]