summaryrefslogtreecommitdiff
path: root/tests/select_for_update/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/select_for_update/tests.py')
-rw-r--r--tests/select_for_update/tests.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/select_for_update/tests.py b/tests/select_for_update/tests.py
index 6de268cb2b..a750dc61db 100644
--- a/tests/select_for_update/tests.py
+++ b/tests/select_for_update/tests.py
@@ -120,6 +120,28 @@ class SelectForUpdateTests(TransactionTestCase):
expected = [value.upper() for value in expected]
self.assertTrue(self.has_for_update_sql(ctx.captured_queries, of=expected))
+ @skipUnlessDBFeature('has_select_for_update_of')
+ def test_for_update_of_followed_by_values(self):
+ with transaction.atomic():
+ values = list(Person.objects.select_for_update(of=('self',)).values('pk'))
+ self.assertEqual(values, [{'pk': self.person.pk}])
+
+ @skipUnlessDBFeature('has_select_for_update_of')
+ def test_for_update_of_followed_by_values_list(self):
+ with transaction.atomic():
+ values = list(Person.objects.select_for_update(of=('self',)).values_list('pk'))
+ self.assertEqual(values, [(self.person.pk,)])
+
+ @skipUnlessDBFeature('has_select_for_update_of')
+ def test_for_update_of_self_when_self_is_not_selected(self):
+ """
+ select_for_update(of=['self']) when the only columns selected are from
+ related tables.
+ """
+ with transaction.atomic():
+ values = list(Person.objects.select_related('born').select_for_update(of=('self',)).values('born__name'))
+ self.assertEqual(values, [{'born__name': self.city1.name}])
+
@skipUnlessDBFeature('has_select_for_update_nowait')
def test_nowait_raises_error_on_block(self):
"""