143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
import datetime
|
|
from unittest import skipUnless
|
|
|
|
from django.core.exceptions import FieldError
|
|
from django.db import connection
|
|
from django.test import TestCase, override_settings
|
|
|
|
from .models import Article, Category, Comment
|
|
|
|
|
|
class DatesTests(TestCase):
|
|
def test_related_model_traverse(self):
|
|
a1 = Article.objects.create(
|
|
title="First one",
|
|
pub_date=datetime.date(2005, 7, 28),
|
|
)
|
|
a2 = Article.objects.create(
|
|
title="Another one",
|
|
pub_date=datetime.date(2010, 7, 28),
|
|
)
|
|
a3 = Article.objects.create(
|
|
title="Third one, in the first day",
|
|
pub_date=datetime.date(2005, 7, 28),
|
|
)
|
|
|
|
a1.comments.create(
|
|
text="Im the HULK!",
|
|
pub_date=datetime.date(2005, 7, 28),
|
|
)
|
|
a1.comments.create(
|
|
text="HULK SMASH!",
|
|
pub_date=datetime.date(2005, 7, 29),
|
|
)
|
|
a2.comments.create(
|
|
text="LMAO",
|
|
pub_date=datetime.date(2010, 7, 28),
|
|
)
|
|
a3.comments.create(
|
|
text="+1",
|
|
pub_date=datetime.date(2005, 8, 29),
|
|
)
|
|
|
|
c = Category.objects.create(name="serious-news")
|
|
c.articles.add(a1, a3)
|
|
|
|
self.assertSequenceEqual(
|
|
Comment.objects.dates("article__pub_date", "year"),
|
|
[
|
|
datetime.date(2005, 1, 1),
|
|
datetime.date(2010, 1, 1),
|
|
],
|
|
)
|
|
self.assertSequenceEqual(
|
|
Comment.objects.dates("article__pub_date", "month"),
|
|
[
|
|
datetime.date(2005, 7, 1),
|
|
datetime.date(2010, 7, 1),
|
|
],
|
|
)
|
|
self.assertSequenceEqual(
|
|
Comment.objects.dates("article__pub_date", "week"),
|
|
[
|
|
datetime.date(2005, 7, 25),
|
|
datetime.date(2010, 7, 26),
|
|
],
|
|
)
|
|
self.assertSequenceEqual(
|
|
Comment.objects.dates("article__pub_date", "day"),
|
|
[
|
|
datetime.date(2005, 7, 28),
|
|
datetime.date(2010, 7, 28),
|
|
],
|
|
)
|
|
self.assertSequenceEqual(
|
|
Article.objects.dates("comments__pub_date", "day"),
|
|
[
|
|
datetime.date(2005, 7, 28),
|
|
datetime.date(2005, 7, 29),
|
|
datetime.date(2005, 8, 29),
|
|
datetime.date(2010, 7, 28),
|
|
],
|
|
)
|
|
self.assertQuerysetEqual(
|
|
Article.objects.dates("comments__approval_date", "day"), []
|
|
)
|
|
self.assertSequenceEqual(
|
|
Category.objects.dates("articles__pub_date", "day"),
|
|
[
|
|
datetime.date(2005, 7, 28),
|
|
],
|
|
)
|
|
|
|
def test_dates_fails_when_no_arguments_are_provided(self):
|
|
with self.assertRaises(TypeError):
|
|
Article.objects.dates()
|
|
|
|
def test_dates_fails_when_given_invalid_field_argument(self):
|
|
with self.assertRaisesMessage(
|
|
FieldError,
|
|
"Cannot resolve keyword 'invalid_field' into field. Choices are: "
|
|
"categories, comments, id, pub_date, pub_datetime, title",
|
|
):
|
|
Article.objects.dates("invalid_field", "year")
|
|
|
|
def test_dates_fails_when_given_invalid_kind_argument(self):
|
|
msg = "'kind' must be one of 'year', 'month', 'week', or 'day'."
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
Article.objects.dates("pub_date", "bad_kind")
|
|
|
|
def test_dates_fails_when_given_invalid_order_argument(self):
|
|
msg = "'order' must be either 'ASC' or 'DESC'."
|
|
with self.assertRaisesMessage(ValueError, msg):
|
|
Article.objects.dates("pub_date", "year", order="bad order")
|
|
|
|
@override_settings(USE_TZ=False)
|
|
def test_dates_trunc_datetime_fields(self):
|
|
Article.objects.bulk_create(
|
|
Article(pub_date=pub_datetime.date(), pub_datetime=pub_datetime)
|
|
for pub_datetime in [
|
|
datetime.datetime(2015, 10, 21, 18, 1),
|
|
datetime.datetime(2015, 10, 21, 18, 2),
|
|
datetime.datetime(2015, 10, 22, 18, 1),
|
|
datetime.datetime(2015, 10, 22, 18, 2),
|
|
]
|
|
)
|
|
self.assertSequenceEqual(
|
|
Article.objects.dates("pub_datetime", "day", order="ASC"),
|
|
[
|
|
datetime.date(2015, 10, 21),
|
|
datetime.date(2015, 10, 22),
|
|
],
|
|
)
|
|
|
|
@skipUnless(connection.vendor == "mysql", "Test checks MySQL query syntax")
|
|
def test_dates_avoid_datetime_cast(self):
|
|
Article.objects.create(pub_date=datetime.date(2015, 10, 21))
|
|
for kind in ["day", "month", "year"]:
|
|
qs = Article.objects.dates("pub_date", kind)
|
|
if kind == "day":
|
|
self.assertIn("DATE(", str(qs.query))
|
|
else:
|
|
self.assertIn(" AS DATE)", str(qs.query))
|