Improved GIS migration test cleanup; refs #23504.

This commit is contained in:
Tim Graham 2014-09-17 19:45:42 -04:00
parent 33e817a6d8
commit 8facb02faf
1 changed files with 2 additions and 2 deletions

View File

@ -64,6 +64,7 @@ class OperationTests(TransactionTestCase):
operation.state_forwards("gis", new_state)
with connection.schema_editor() as editor:
operation.database_forwards("gis", editor, project_state, new_state)
self.current_state = new_state
self.assertColumnExists("gis_neighborhood", "path")
# Test GeometryColumns when available
@ -72,7 +73,6 @@ class OperationTests(TransactionTestCase):
GeometryColumns.objects.filter(**{GeometryColumns.table_name_col(): "gis_neighborhood"}).count(),
2
)
self.current_state = new_state
def test_remove_gis_field(self):
"""
@ -84,6 +84,7 @@ class OperationTests(TransactionTestCase):
operation.state_forwards("gis", new_state)
with connection.schema_editor() as editor:
operation.database_forwards("gis", editor, project_state, new_state)
self.current_state = new_state
self.assertColumnNotExists("gis_neighborhood", "geom")
# Test GeometryColumns when available
@ -92,4 +93,3 @@ class OperationTests(TransactionTestCase):
GeometryColumns.objects.filter(**{GeometryColumns.table_name_col(): "gis_neighborhood"}).count(),
0
)
self.current_state = new_state