From 3a42d9730cbb07ffbb983791e631f5d0a6746f68 Mon Sep 17 00:00:00 2001 From: Claude Paroz Date: Mon, 1 Dec 2014 11:13:01 +0100 Subject: [1.7.x] Fixed #23909 -- Prevented crash when collecting SQL for RunSQL Thanks James Rivett-Carnac for the report and Markus Holtermann for the review. Backport of e11c6fd21 from master. --- tests/migrations/test_operations.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'tests') diff --git a/tests/migrations/test_operations.py b/tests/migrations/test_operations.py index 05e0f22696..5f4dc553a2 100644 --- a/tests/migrations/test_operations.py +++ b/tests/migrations/test_operations.py @@ -1219,6 +1219,12 @@ class OperationTests(OperationTestBase): self.assertEqual(len(new_state.models["test_runsql", "somethingelse"].fields), 1) # Make sure there's no table self.assertTableNotExists("i_love_ponies") + # Test SQL collection + with connection.schema_editor(collect_sql=True) as editor: + operation.database_forwards("test_runsql", editor, project_state, new_state) + self.assertIn("LIKE '%%ponies';", "\n".join(editor.collected_sql)) + operation.database_backwards("test_runsql", editor, project_state, new_state) + self.assertIn("LIKE '%%Ponies%%';", "\n".join(editor.collected_sql)) # Test the database alteration with connection.schema_editor() as editor: operation.database_forwards("test_runsql", editor, project_state, new_state) -- cgit v1.3