From 29a3f8b4bb1c4174a2b63766018b9106593b8ab6 Mon Sep 17 00:00:00 2001
From: Simon Charette <charette.s@gmail.com>
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')