import py.test
from sqlobject import DatabaseIndex, ForeignKey, IntCol, MultipleJoin, \
    SQLObject, StringCol
from sqlobject.dberrors import DatabaseError, IntegrityError, \
    OperationalError, ProgrammingError
from sqlobject.tests.dbtest import raises, setupClass, supports


########################################
# Indexes
########################################


class SOIndex1(SQLObject):
    name = StringCol(length=100)
    number = IntCol()

    nameIndex = DatabaseIndex('name', unique=True)
    nameIndex2 = DatabaseIndex(name, number)
    nameIndex3 = DatabaseIndex({'column': name,
                                'length': 3})


class SOIndex2(SQLObject):
    name = StringCol(length=100)
    nameIndex = DatabaseIndex({'expression': 'lower(name)'})


def test_indexes_1():
    setupClass(SOIndex1)
    n = 0
    for name in 'blah blech boring yep yort snort'.split():
        n += 1
        SOIndex1(name=name, number=n)
    mod = SOIndex1._connection.module
    raises(
        (mod.ProgrammingError, mod.IntegrityError,
         mod.OperationalError, mod.DatabaseError,
         ProgrammingError, IntegrityError, OperationalError, DatabaseError),
        SOIndex1, name='blah', number=0)


def test_indexes_2():
    if not supports('expressionIndex'):
        py.test.skip("expressionIndex isn't supported")
    setupClass(SOIndex2)
    SOIndex2(name='')


class PersonIndexGet(SQLObject):
    firstName = StringCol(length=100)
    lastName = StringCol(length=100)
    age = IntCol(alternateID=True)
    nameIndex = DatabaseIndex(firstName, lastName, unique=True)


def test_index_get_1():
    setupClass(PersonIndexGet, force=True)

    PersonIndexGet(firstName='Eric', lastName='Idle', age=62)
    PersonIndexGet(firstName='Terry', lastName='Gilliam', age=65)
    PersonIndexGet(firstName='John', lastName='Cleese', age=66)

    PersonIndexGet.get(1)
    PersonIndexGet.nameIndex.get('Terry', 'Gilliam')
    PersonIndexGet.nameIndex.get(firstName='John', lastName='Cleese')

    raises(Exception, PersonIndexGet.nameIndex.get,
           firstName='Graham', lastName='Chapman')

    raises(Exception, PersonIndexGet.nameIndex.get,
           'Terry', lastName='Gilliam')

    raises(Exception, PersonIndexGet.nameIndex.get, 'Terry', 'Gilliam', 65)

    raises(Exception, PersonIndexGet.nameIndex.get, 'Terry')


class PersonIndexGet2(SQLObject):
    name = StringCol(alternateID=True, length=100)
    age = IntCol()
    addresses = MultipleJoin('AddressIndexGet2')


class AddressIndexGet2(SQLObject):
    person = ForeignKey('PersonIndexGet2', notNone=True)
    type = StringCol(notNone=True, length=100)
    street = StringCol(notNone=True)
    pk = DatabaseIndex(person, type, unique=True)


def test_index_get_2():
    setupClass([PersonIndexGet2, AddressIndexGet2])

    p = PersonIndexGet2(name='Terry Guilliam', age=64)
    AddressIndexGet2(person=p, type='home', street='Terry Street 234')
    AddressIndexGet2(person=p, type='work', street='Guilliam Street 234')

    AddressIndexGet2.pk.get(p, 'work')
    AddressIndexGet2.pk.get(person=p, type='work')
