Added a test for updating an annotated queryset.

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

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