Ticket #13935: 13935_tests.diff

modeltests/dates/models.py

@@ +1 @@
+# coding: utf-8
+"""
+1. Tests for ticket #13935 (QuerySet .dates() method should span relationships)
+
+There are two models, one-to-many, and .dates() on second one with reference to first one field must succeed
+"""
+from django.db import models, DEFAULT_DB_ALIAS
+
+class Article(models.Model):
+    title = models.CharField(max_length=100)
+    pub_date = models.DateField()
+    pub_date_time = models.DateTimeField()
+
+    def __unicode__(self):
+        return self.title
+
+class Comment(models.Model):
+    article = models.ForeignKey(Article)
+    text = models.TextField()
+    pub_date = models.DateField()
+    pub_date_time = models.DateTimeField()
+
+    def __unicode__(self):
+        return 'Comment to %s (%s)' % (self.article.title, self.pub_date_time)
+
+
+__test__ = {'API_TESTS': """
+# first, some data
+>>> from datetime import datetime
+>>> a1 = Article(title="First one", pub_date=datetime(2005, 7, 28), pub_date_time=datetime(2005, 7, 28))
+>>> a1.save()
+>>> a2 = Article(title="Another one", pub_date=datetime(2010, 7, 28), pub_date_time=datetime(2010, 7, 28))
+>>> a2.save()
+>>> a3 = Article(title="Third one, in the first day", pub_date=datetime(2005, 7, 28), pub_date_time=datetime(2005, 7, 28))
+>>> a3.save()
+
+# are we sane?
+>>> Article.objects.all()
+[<Article: First one>, <Article: Another one>, <Article: Third one, in the first day>]
+
+# awesome! some comments
+>>> c1 = Comment(text="Im the HULK!", article=a1, pub_date=datetime(2005, 7, 28), pub_date_time=datetime(2005, 7, 28))
+>>> c1.save()
+>>> c2 = Comment(text="HULK SMASH!", article=a1, pub_date=datetime(2005, 7, 29), pub_date_time=datetime(2005, 7, 29))
+>>> c2.save()
+>>> c3 = Comment(text="LMAO", article=a2, pub_date=datetime(2010, 7, 28), pub_date_time=datetime(2010, 7, 28))
+>>> c3.save()
+>>> c4 = Comment(text="+1", article=a3, pub_date=datetime(2005, 8, 29), pub_date_time=datetime(2005, 8, 29))
+>>> c4.save()
+
+# good?
+>>> Comment.objects.all()
+[<Comment: Comment to First one (2005-07-28 00:00:00)>, <Comment: Comment to First one (2005-07-29 00:00:00)>, <Comment: Comment to Another one (2010-07-28 00:00:00)>, <Comment: Comment to Third one, in the first day (2005-08-29 00:00:00)>]
+
+# okay, we are ready now. let's go straight to the point
+>>> Comment.objects.dates('pub_date', 'year')
+[datetime.datetime(2005, 1, 1, 0, 0), datetime.datetime(2010, 1, 1, 0, 0)]
+>>> Comment.objects.dates('pub_date', 'month')
+[datetime.datetime(2005, 7, 1, 0, 0), datetime.datetime(2005, 8, 1, 0, 0), datetime.datetime(2010, 7, 1, 0, 0)]
+>>> Comment.objects.dates('pub_date', 'day')
+[datetime.datetime(2005, 7, 28, 0, 0), datetime.datetime(2005, 7, 29, 0, 0), datetime.datetime(2005, 8, 29, 0, 0), datetime.datetime(2010, 7, 28, 0, 0)]
+
+# that was expected. now, something not so easy?
+>>> Comment.objects.dates('article__pub_date', 'year')
+[datetime.datetime(2005, 1, 1, 0, 0), datetime.datetime(2010, 1, 1, 0, 0)]
+>>> Comment.objects.dates('article__pub_date', 'month')
+[datetime.datetime(2005, 7, 1, 0, 0), datetime.datetime(2010, 7, 1, 0, 0)]
+>>> Comment.objects.dates('article__pub_date', 'day')
+[datetime.datetime(2005, 7, 28, 0, 0), datetime.datetime(2010, 7, 28, 0, 0)]
+"""}
+