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