test0908
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Only swallow table-does-not-exist errors in tests

This commit is contained in:
Andrew Godwin 2012-09-24 13:30:17 +01:00
parent 0bcfc068b0
commit 15c1920964
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/modeltests/schema/tests.py
View File

@ -20,6 +20,7 @@ class SchemaTests(TestCase):
"""
models = [Author, AuthorWithM2M, Book, BookWithSlug, BookWithM2M, Tag, TagUniqueRename, UniqueTest]
no_table_strings = ["no such table", "unknown table", "does not exist"]
# Utility functions
@ -57,8 +58,11 @@ class SchemaTests(TestCase):
cursor.execute(connection.schema_editor().sql_delete_table % {
"table": connection.ops.quote_name(field.rel.through._meta.db_table),
})
except DatabaseError:
connection.rollback()
except DatabaseError as e:
if any([s in str(e).lower() for s in self.no_table_strings]):
connection.rollback()
else:
raise
else:
connection.commit()
# Then remove the main tables
@ -66,8 +70,11 @@ class SchemaTests(TestCase):
cursor.execute(connection.schema_editor().sql_delete_table % {
"table": connection.ops.quote_name(model._meta.db_table),
})
except DatabaseError:
connection.rollback()
except DatabaseError as e:
if any([s in str(e).lower() for s in self.no_table_strings]):
connection.rollback()
else:
raise
else:
connection.commit()
connection.enable_constraint_checking()