Source code for sqlobject.tests.test_enum

from sqlobject import EnumCol, SQLObject, UnicodeCol
from sqlobject.col import validators
from sqlobject.tests.dbtest import raises, setupClass


########################################
# Enum test
########################################


class Enum1(SQLObject):

    cl = EnumCol(enumValues=['a', 'bcd', 'e'])


def testBad():
    setupClass(Enum1)
    for _l in ['a', 'bcd', 'a', 'e']:
        Enum1(cl=_l)
    raises(
        (Enum1._connection.module.IntegrityError,
         Enum1._connection.module.ProgrammingError,
         validators.Invalid),
        Enum1, cl='b')


class EnumWithNone(SQLObject):

    cl = EnumCol(enumValues=['a', 'bcd', 'e', None])


def testNone():
    setupClass(EnumWithNone)
    for _l in [None, 'a', 'bcd', 'a', 'e', None]:
        e = EnumWithNone(cl=_l)
        assert e.cl == _l


class EnumWithDefaultNone(SQLObject):

    cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default=None)


def testDefaultNone():
    setupClass(EnumWithDefaultNone)

    e = EnumWithDefaultNone()
    assert e.cl is None


class EnumWithDefaultOther(SQLObject):

    cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default='a')


def testDefaultOther():
    setupClass(EnumWithDefaultOther)

    e = EnumWithDefaultOther()
    assert e.cl == 'a'


class EnumUnicode(SQLObject):

    n = UnicodeCol()
    cl = EnumCol(enumValues=['a', 'b'])


def testUnicode():
    setupClass(EnumUnicode)

    EnumUnicode(n=u'a', cl='a')
    EnumUnicode(n=u'b', cl=u'b')
    EnumUnicode(n=u'\u201c', cl='a')
    EnumUnicode(n=u'\u201c', cl=u'b')