test: add date filter

tortoise · Oct 8, 2023 · ab488d0 · ab488d0
1 parent 0484cf5
commit ab488d0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_filtering.py b/tests/test_filtering.py
@@ -7,6 +7,7 @@
     Reporter,
     Team,
     Tournament,
+    DateFields,
 )
 from tortoise.contrib import test
 from tortoise.contrib.test.condition import NotEQ
@@ -182,6 +183,11 @@ async def test_filter_exact(self):
         self.assertEqual(await DatetimeFields.filter(datetime__second=0).count(), 1)
         self.assertEqual(await DatetimeFields.filter(datetime__microsecond=0).count(), 1)
 
+        await DateFields.create(date=datetime.date(year=2021, month=6, day=21))
+        self.assertEqual(await DateFields.filter(date__year=2021).count(), 0)
+        self.assertEqual(await DateFields.filter(date__month=6).count(), 0)
+        self.assertEqual(await DateFields.filter(date__day=21).count(), 0)
+
     async def test_filter_by_aggregation_field(self):
         tournament = await Tournament.create(name="0")
         await Tournament.create(name="1")