Refs #22343 -- Corrected a test for missing select_for_update(nowait=True) support.

1 files 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')