fromsqlobjectimport EnumCol, SQLObject, UnicodeCol
fromsqlobject.colimport validators
fromsqlobject.tests.dbtestimport raises, setupClass
########################################
# Enum test
########################################
[docs]
classEnum1(SQLObject):
cl = EnumCol(enumValues=['a', 'bcd', 'e'])
[docs]
deftestBad():
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]
classEnumWithNone(SQLObject):
cl = EnumCol(enumValues=['a', 'bcd', 'e', None])
[docs]
deftestNone():
setupClass(EnumWithNone)
for _l in [None, 'a', 'bcd', 'a', 'e', None]:
e = EnumWithNone(cl=_l)
assert e.cl == _l
[docs]
classEnumWithDefaultNone(SQLObject):
cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default=None)
[docs]
deftestDefaultNone():
setupClass(EnumWithDefaultNone)
e = EnumWithDefaultNone()
assert e.cl is None
[docs]
classEnumWithDefaultOther(SQLObject):
cl = EnumCol(enumValues=['a', 'bcd', 'e', None], default='a')
[docs]
deftestDefaultOther():
setupClass(EnumWithDefaultOther)
e = EnumWithDefaultOther()
assert e.cl == 'a'
[docs]
classEnumUnicode(SQLObject):
n = UnicodeCol()
cl = EnumCol(enumValues=['a', 'b'])
[docs]
deftestUnicode():
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')