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/db_utils | |
parent | 3308085838f520db49f606b72345a301c1cf2a3e (diff) | |
download | django-7a6fbf36b1fdb8978ea0842075ccce83bcd63789.tar.gz |
Fixed #28853 -- Updated connection.cursor() uses to use a context manager.
Diffstat (limited to 'tests/db_utils')
-rw-r--r-- | tests/db_utils/tests.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/db_utils/tests.py b/tests/db_utils/tests.py index 2a45342df5..4e35e6bb8b 100644 --- a/tests/db_utils/tests.py +++ b/tests/db_utils/tests.py @@ -26,10 +26,10 @@ class DatabaseErrorWrapperTests(TestCase): @unittest.skipUnless(connection.vendor == 'postgresql', 'PostgreSQL test') def test_reraising_backend_specific_database_exception(self): - cursor = connection.cursor() - msg = 'table "X" does not exist' - with self.assertRaisesMessage(ProgrammingError, msg) as cm: - cursor.execute('DROP TABLE "X"') + with connection.cursor() as cursor: + msg = 'table "X" does not exist' + with self.assertRaisesMessage(ProgrammingError, msg) as cm: + cursor.execute('DROP TABLE "X"') self.assertNotEqual(type(cm.exception), type(cm.exception.__cause__)) self.assertIsNotNone(cm.exception.__cause__) self.assertIsNotNone(cm.exception.__cause__.pgcode) |