importpytest
fromsqlobjectimport SQLObject, StringCol
fromsqlobject.dberrorsimport DatabaseError, DuplicateEntryError, \
OperationalError, ProgrammingError
fromsqlobject.tests.dbtestimport getConnection, raises, setupClass, supports
########################################
# Table aliases and self-joins
########################################
[docs]
classSOTestException(SQLObject):
name = StringCol(unique=True, length=100)
[docs]
classSOTestExceptionWithNonexistingTable(SQLObject):
pass
[docs]
deftest_exceptions():
if not supports("exceptions"):
pytest.skip("exceptions aren't supported")
setupClass(SOTestException)
SOTestException(name="test")
raises(DuplicateEntryError, SOTestException, name="test")
connection = getConnection()
SOTestExceptionWithNonexistingTable.setConnection(connection)
try:
list(SOTestExceptionWithNonexistingTable.select())
except ProgrammingError as e:
assert e.args[0].code in (1146, '42P01')
except OperationalError:
assert connection.dbName == 'sqlite'
except DatabaseError:
assert connection.dbName == 'postgres' \
and connection.driver == 'pg8000'
else:
assert False, "DID NOT RAISE"