summaryrefslogtreecommitdiff
path: root/tests/lookup/test_lookups.py
blob: 9b2d90fdd95b5d2aa0f6bcff4db8a85f79b1db15 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from datetime import datetime

from django.db.models import Value
from django.db.models.fields import DateTimeField
from django.db.models.lookups import YearLookup
from django.test import SimpleTestCase


class YearLookupTests(SimpleTestCase):
    def test_get_bound_params(self):
        look_up = YearLookup(
            lhs=Value(datetime(2010, 1, 1, 0, 0, 0), output_field=DateTimeField()),
            rhs=Value(datetime(2010, 1, 1, 23, 59, 59), output_field=DateTimeField()),
        )
        msg = 'subclasses of YearLookup must provide a get_bound_params() method'
        with self.assertRaisesMessage(NotImplementedError, msg):
            look_up.get_bound_params(datetime(2010, 1, 1, 0, 0, 0), datetime(2010, 1, 1, 23, 59, 59))