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
########################################
[docs]class Enum1(SQLObject):
    cl = EnumCol(enumValues=['a', 'bcd', 'e']) 
[docs]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') 
[docs]class EnumWithNone(SQLObject):
    cl = EnumCol(enumValues=['a', 'bcd', 'e', None]) 
[docs]def testNone():
    setupClass(EnumWithNone)
    for _l in [None, 'a', 'bcd', 'a', 'e', None]:
        e = EnumWithNone(cl=_l)
        assert e.cl == _l 
[docs]class EnumWithDefaultNone(SQLObject):
    cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default=None) 
[docs]def testDefaultNone():
    setupClass(EnumWithDefaultNone)
    e = EnumWithDefaultNone()
    assert e.cl is None 
[docs]class EnumWithDefaultOther(SQLObject):
    cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default='a') 
[docs]def testDefaultOther():
    setupClass(EnumWithDefaultOther)
    e = EnumWithDefaultOther()
    assert e.cl == 'a' 
[docs]class EnumUnicode(SQLObject):
    n = UnicodeCol()
    cl = EnumCol(enumValues=['a', 'b']) 
[docs]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')