diff options
| author | Jon Dufresne <jon.dufresne@gmail.com> | 2017-11-28 05:12:28 -0800 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2017-11-28 11:28:09 -0500 |
| commit | 7a6fbf36b1fdb8978ea0842075ccce83bcd63789 (patch) | |
| tree | a48e72be7e31066d92de43a85458b0bfb7e2cfcd /tests/backends/postgresql/test_introspection.py | |
| parent | 3308085838f520db49f606b72345a301c1cf2a3e (diff) | |
Fixed #28853 -- Updated connection.cursor() uses to use a context manager.
Diffstat (limited to 'tests/backends/postgresql/test_introspection.py')
| -rw-r--r-- | tests/backends/postgresql/test_introspection.py | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/tests/backends/postgresql/test_introspection.py b/tests/backends/postgresql/test_introspection.py index cfa801a77f..4dcadbd733 100644 --- a/tests/backends/postgresql/test_introspection.py +++ b/tests/backends/postgresql/test_introspection.py @@ -9,15 +9,15 @@ from ..models import Person @unittest.skipUnless(connection.vendor == 'postgresql', "Test only for PostgreSQL") class DatabaseSequenceTests(TestCase): def test_get_sequences(self): - cursor = connection.cursor() - seqs = connection.introspection.get_sequences(cursor, Person._meta.db_table) - self.assertEqual( - seqs, - [{'table': Person._meta.db_table, 'column': 'id', 'name': 'backends_person_id_seq'}] - ) - cursor.execute('ALTER SEQUENCE backends_person_id_seq RENAME TO pers_seq') - seqs = connection.introspection.get_sequences(cursor, Person._meta.db_table) - self.assertEqual( - seqs, - [{'table': Person._meta.db_table, 'column': 'id', 'name': 'pers_seq'}] - ) + with connection.cursor() as cursor: + seqs = connection.introspection.get_sequences(cursor, Person._meta.db_table) + self.assertEqual( + seqs, + [{'table': Person._meta.db_table, 'column': 'id', 'name': 'backends_person_id_seq'}] + ) + cursor.execute('ALTER SEQUENCE backends_person_id_seq RENAME TO pers_seq') + seqs = connection.introspection.get_sequences(cursor, Person._meta.db_table) + self.assertEqual( + seqs, + [{'table': Person._meta.db_table, 'column': 'id', 'name': 'pers_seq'}] + ) |
