Added a test for updating an annotated queryset.

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/update/tests.py b/tests/update/tests.py
index 4379809c08..3dc97c9173 100644
--- a/tests/update/tests.py
+++ b/tests/update/tests.py
@@ -1,5 +1,7 @@
 from __future__ import unicode_literals
 
+from django.core.exceptions import FieldError
+from django.db.models import F, Max
 from django.test import TestCase
 
 from .models import A, B, Bar, D, DataPoint, Foo, RelatedPoint
@@ -138,3 +140,21 @@ class AdvancedTests(TestCase):
         self.assertEqual(bar_qs[0].foo_id, a_foo.target)
         bar_qs.update(foo=b_foo)
         self.assertEqual(bar_qs[0].foo_id, b_foo.target)
+
+    def test_update_annotated_queryset(self):
+        """
+        Update of a queryset that's been annotated.
+        """
+        # Trivial annotated update
+        qs = DataPoint.objects.annotate(alias=F('value'))
+        self.assertEqual(qs.update(another_value='foo'), 3)
+        # Update where annotation is used for filtering
+        qs = DataPoint.objects.annotate(alias=F('value')).filter(alias='apple')
+        self.assertEqual(qs.update(another_value='foo'), 1)
+        # Update where annotation is used in update parameters
+        qs = DataPoint.objects.annotate(alias=F('value'))
+        self.assertEqual(qs.update(another_value=F('alias')), 3)
+        # Update where aggregation annotation is used in update parameters
+        qs = DataPoint.objects.annotate(max=Max('value'))
+        with self.assertRaisesMessage(FieldError, 'Aggregate functions are not allowed in this query'):
+            qs.update(another_value=F('max'))