From 29a3f8b4bb1c4174a2b63766018b9106593b8ab6 Mon Sep 17 00:00:00 2001 From: Simon Charette Date: Mon, 8 Aug 2016 12:28:37 -0400 Subject: [PATCH] Refs #22343 -- Corrected a test for missing select_for_update(nowait=True) support. --- tests/select_for_update/tests.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/select_for_update/tests.py b/tests/select_for_update/tests.py index 55ead7ef8f..dcdd742ad4 100644 --- a/tests/select_for_update/tests.py +++ b/tests/select_for_update/tests.py @@ -132,12 +132,12 @@ class SelectForUpdateTests(TransactionTestCase): @skipUnlessDBFeature('has_select_for_update') def test_unsupported_nowait_raises_error(self): """ - If a SELECT...FOR UPDATE NOWAIT is run on a database backend - that supports FOR UPDATE but not NOWAIT, then we should find - that a DatabaseError is raised. + DatabaseError is raised if a SELECT...FOR UPDATE NOWAIT is run on + a database backend that supports FOR UPDATE but not NOWAIT. """ - with self.assertRaises(DatabaseError): - list(Person.objects.all().select_for_update(nowait=True)) + with self.assertRaisesMessage(DatabaseError, 'NOWAIT is not supported on this database backend.'): + with transaction.atomic(): + Person.objects.select_for_update(nowait=True).get() @skipIfDBFeature('has_select_for_update_skip_locked') @skipUnlessDBFeature('has_select_for_update')