summaryrefslogtreecommitdiff
path: root/tests/expressions/test_deprecation.py
blob: cdb1e43af6f2fb2a725e362c4e0765275938e895 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from django.db.models import Count, Func
from django.test import SimpleTestCase
from django.utils.deprecation import RemovedInDjango40Warning

from .models import Employee


class MissingAliasFunc(Func):
    template = '1'

    def get_group_by_cols(self):
        return []


class GetGroupByColsTest(SimpleTestCase):
    def test_missing_alias(self):
        msg = (
            '`alias=None` must be added to the signature of '
            'expressions.test_deprecation.MissingAliasFunc.get_group_by_cols().'
        )
        with self.assertRaisesMessage(RemovedInDjango40Warning, msg):
            Employee.objects.values(
                one=MissingAliasFunc(),
            ).annotate(cnt=Count('company_ceo_set'))