summaryrefslogtreecommitdiff
path: root/tests/backends/postgresql/test_introspection.py
diff options
context:
space:
mode:
authorJon Dufresne <jon.dufresne@gmail.com>2017-11-28 05:12:28 -0800
committerTim Graham <timograham@gmail.com>2017-11-28 11:28:09 -0500
commit7a6fbf36b1fdb8978ea0842075ccce83bcd63789 (patch)
treea48e72be7e31066d92de43a85458b0bfb7e2cfcd /tests/backends/postgresql/test_introspection.py
parent3308085838f520db49f606b72345a301c1cf2a3e (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.py24
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'}]
+ )