Commit 6f7977bb authored by Andrew Godwin's avatar Andrew Godwin
Browse files

Fixed #21029: Test for previously-commited SchemaEditor.__exit__ bug.

parent ffe21e1f

tests/schema/tests.py

+13 −0
Original line number Diff line number Diff line
@@ -643,3 +643,16 @@ class SchemaTests(TransactionTestCase): 
        self.assertTrue(

            connection.introspection.get_indexes(connection.cursor(), Tag._meta.db_table)['slug']['primary_key'],

        )



    def test_context_manager_exit(self):

        """

        Ensures transaction is correctly closed when an error occurs

        inside a SchemaEditor context.

        """

        class SomeError(Exception):

            pass

        try:

            with connection.schema_editor() as editor:

                raise SomeError

        except SomeError:

            self.assertFalse(connection.in_atomic_block)