Added cleanup for an introspection test.

This commit is contained in:
Tim Graham 2017-09-27 09:51:49 -04:00
parent e8c4596329
commit c7e9e22695
1 changed files with 6 additions and 3 deletions

View File

@ -44,9 +44,12 @@ class IntrospectionTests(TransactionTestCase):
self.fail("The test user has no CREATE VIEW privileges")
else:
raise
self.assertIn('introspection_article_view', connection.introspection.table_names(include_views=True))
self.assertNotIn('introspection_article_view', connection.introspection.table_names())
try:
self.assertIn('introspection_article_view', connection.introspection.table_names(include_views=True))
self.assertNotIn('introspection_article_view', connection.introspection.table_names())
finally:
with connection.cursor() as cursor:
cursor.execute('DROP VIEW introspection_article_view')
def test_unmanaged_through_model(self):
tables = connection.introspection.django_table_names()