Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Migrated many-to-one doctests. Thanks to George Sakkis for the patch.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14281 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit 634d0a92b8bbbf6f2894cc3590cc8e9c7c38e0a7 1 parent bdaaac1
@freakboy3742 freakboy3742 authored
View
287 tests/modeltests/many_to_one/models.py
@@ -24,290 +24,3 @@ def __unicode__(self):
class Meta:
ordering = ('headline',)
-
-__test__ = {'API_TESTS':"""
-# Create a few Reporters.
->>> r = Reporter(first_name='John', last_name='Smith', email='john@example.com')
->>> r.save()
-
->>> r2 = Reporter(first_name='Paul', last_name='Jones', email='paul@example.com')
->>> r2.save()
-
-# Create an Article.
->>> from datetime import datetime
->>> a = Article(id=None, headline="This is a test", pub_date=datetime(2005, 7, 27), reporter=r)
->>> a.save()
-
->>> a.reporter.id
-1
-
->>> a.reporter
-<Reporter: John Smith>
-
-# Article objects have access to their related Reporter objects.
->>> r = a.reporter
-
-# These are strings instead of unicode strings because that's what was used in
-# the creation of this reporter (and we haven't refreshed the data from the
-# database, which always returns unicode strings).
->>> r.first_name, r.last_name
-('John', 'Smith')
-
-# Create an Article via the Reporter object.
->>> new_article = r.article_set.create(headline="John's second story", pub_date=datetime(2005, 7, 29))
->>> new_article
-<Article: John's second story>
->>> new_article.reporter.id
-1
-
-# Create a new article, and add it to the article set.
->>> new_article2 = Article(headline="Paul's story", pub_date=datetime(2006, 1, 17))
->>> r.article_set.add(new_article2)
->>> new_article2.reporter.id
-1
->>> r.article_set.all()
-[<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>]
-
-# Add the same article to a different article set - check that it moves.
->>> r2.article_set.add(new_article2)
->>> new_article2.reporter.id
-2
-
-# Adding an object of the wrong type raises TypeError
->>> r.article_set.add(r2)
-Traceback (most recent call last):
-...
-TypeError: 'Article' instance expected
-
->>> r.article_set.all()
-[<Article: John's second story>, <Article: This is a test>]
->>> r2.article_set.all()
-[<Article: Paul's story>]
-
-# Assign the article to the reporter directly using the descriptor
->>> new_article2.reporter = r
->>> new_article2.save()
->>> new_article2.reporter
-<Reporter: John Smith>
->>> new_article2.reporter.id
-1
->>> r.article_set.all()
-[<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>]
->>> r2.article_set.all()
-[]
-
-# Set the article back again using set descriptor.
->>> r2.article_set = [new_article, new_article2]
->>> r.article_set.all()
-[<Article: This is a test>]
->>> r2.article_set.all()
-[<Article: John's second story>, <Article: Paul's story>]
-
-# Funny case - assignment notation can only go so far; because the
-# ForeignKey cannot be null, existing members of the set must remain
->>> r.article_set = [new_article]
->>> r.article_set.all()
-[<Article: John's second story>, <Article: This is a test>]
->>> r2.article_set.all()
-[<Article: Paul's story>]
-
-# Reporter cannot be null - there should not be a clear or remove method
->>> hasattr(r2.article_set, 'remove')
-False
->>> hasattr(r2.article_set, 'clear')
-False
-
-# Reporter objects have access to their related Article objects.
->>> r.article_set.all()
-[<Article: John's second story>, <Article: This is a test>]
-
->>> r.article_set.filter(headline__startswith='This')
-[<Article: This is a test>]
-
->>> r.article_set.count()
-2
-
->>> r2.article_set.count()
-1
-
-# Get articles by id
->>> Article.objects.filter(id__exact=1)
-[<Article: This is a test>]
->>> Article.objects.filter(pk=1)
-[<Article: This is a test>]
-
-# Query on an article property
->>> Article.objects.filter(headline__startswith='This')
-[<Article: This is a test>]
-
-# The API automatically follows relationships as far as you need.
-# Use double underscores to separate relationships.
-# This works as many levels deep as you want. There's no limit.
-# Find all Articles for any Reporter whose first name is "John".
->>> Article.objects.filter(reporter__first_name__exact='John')
-[<Article: John's second story>, <Article: This is a test>]
-
-# Check that implied __exact also works
->>> Article.objects.filter(reporter__first_name='John')
-[<Article: John's second story>, <Article: This is a test>]
-
-# Query twice over the related field.
->>> Article.objects.filter(reporter__first_name__exact='John', reporter__last_name__exact='Smith')
-[<Article: John's second story>, <Article: This is a test>]
-
-# The underlying query only makes one join when a related table is referenced twice.
->>> queryset = Article.objects.filter(reporter__first_name__exact='John', reporter__last_name__exact='Smith')
->>> sql = queryset.query.get_compiler(queryset.db).as_sql()[0]
->>> sql.count('INNER JOIN')
-1
-
-# The automatically joined table has a predictable name.
->>> Article.objects.filter(reporter__first_name__exact='John').extra(where=["many_to_one_reporter.last_name='Smith'"])
-[<Article: John's second story>, <Article: This is a test>]
-
-# And should work fine with the unicode that comes out of
-# forms.Form.cleaned_data
->>> Article.objects.filter(reporter__first_name__exact='John').extra(where=["many_to_one_reporter.last_name='%s'" % u'Smith'])
-[<Article: John's second story>, <Article: This is a test>]
-
-# Find all Articles for the Reporter whose ID is 1.
-# Use direct ID check, pk check, and object comparison
->>> Article.objects.filter(reporter__id__exact=1)
-[<Article: John's second story>, <Article: This is a test>]
->>> Article.objects.filter(reporter__pk=1)
-[<Article: John's second story>, <Article: This is a test>]
->>> Article.objects.filter(reporter=1)
-[<Article: John's second story>, <Article: This is a test>]
->>> Article.objects.filter(reporter=r)
-[<Article: John's second story>, <Article: This is a test>]
-
->>> Article.objects.filter(reporter__in=[1,2]).distinct()
-[<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>]
->>> Article.objects.filter(reporter__in=[r,r2]).distinct()
-[<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>]
-
-# You can also use a queryset instead of a literal list of instances.
-# The queryset must be reduced to a list of values using values(),
-# then converted into a query
->>> Article.objects.filter(reporter__in=Reporter.objects.filter(first_name='John').values('pk').query).distinct()
-[<Article: John's second story>, <Article: This is a test>]
-
-# You need two underscores between "reporter" and "id" -- not one.
->>> Article.objects.filter(reporter_id__exact=1)
-Traceback (most recent call last):
- ...
-FieldError: Cannot resolve keyword 'reporter_id' into field. Choices are: headline, id, pub_date, reporter
-
-# You need to specify a comparison clause
->>> Article.objects.filter(reporter_id=1)
-Traceback (most recent call last):
- ...
-FieldError: Cannot resolve keyword 'reporter_id' into field. Choices are: headline, id, pub_date, reporter
-
-# You can also instantiate an Article by passing
-# the Reporter's ID instead of a Reporter object.
->>> a3 = Article(id=None, headline="This is a test", pub_date=datetime(2005, 7, 27), reporter_id=r.id)
->>> a3.save()
->>> a3.reporter.id
-1
->>> a3.reporter
-<Reporter: John Smith>
-
-# Similarly, the reporter ID can be a string.
->>> a4 = Article(id=None, headline="This is a test", pub_date=datetime(2005, 7, 27), reporter_id="1")
->>> a4.save()
->>> a4.reporter
-<Reporter: John Smith>
-
-# Reporters can be queried
->>> Reporter.objects.filter(id__exact=1)
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(pk=1)
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(first_name__startswith='John')
-[<Reporter: John Smith>]
-
-# Reporters can query in opposite direction of ForeignKey definition
->>> Reporter.objects.filter(article__id__exact=1)
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article__pk=1)
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article=1)
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article=a)
-[<Reporter: John Smith>]
-
->>> Reporter.objects.filter(article__in=[1,4]).distinct()
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article__in=[1,a3]).distinct()
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article__in=[a,a3]).distinct()
-[<Reporter: John Smith>]
-
->>> Reporter.objects.filter(article__headline__startswith='This')
-[<Reporter: John Smith>, <Reporter: John Smith>, <Reporter: John Smith>]
->>> Reporter.objects.filter(article__headline__startswith='This').distinct()
-[<Reporter: John Smith>]
-
-# Counting in the opposite direction works in conjunction with distinct()
->>> Reporter.objects.filter(article__headline__startswith='This').count()
-3
->>> Reporter.objects.filter(article__headline__startswith='This').distinct().count()
-1
-
-# Queries can go round in circles.
->>> Reporter.objects.filter(article__reporter__first_name__startswith='John')
-[<Reporter: John Smith>, <Reporter: John Smith>, <Reporter: John Smith>, <Reporter: John Smith>]
->>> Reporter.objects.filter(article__reporter__first_name__startswith='John').distinct()
-[<Reporter: John Smith>]
->>> Reporter.objects.filter(article__reporter__exact=r).distinct()
-[<Reporter: John Smith>]
-
-# Regression for #12876 -- Model methods that include queries that
-# recursive don't cause recursion depth problems under deepcopy.
->>> r.cached_query = Article.objects.filter(reporter=r)
->>> from copy import deepcopy
->>> deepcopy(r)
-<Reporter: John Smith>
-
-# Check that implied __exact also works.
->>> Reporter.objects.filter(article__reporter=r).distinct()
-[<Reporter: John Smith>]
-
-# It's possible to use values() calls across many-to-one relations. (Note, too, that we clear the ordering here so as not to drag the 'headline' field into the columns being used to determine uniqueness.)
->>> d = {'reporter__first_name': u'John', 'reporter__last_name': u'Smith'}
->>> list(Article.objects.filter(reporter=r).distinct().order_by().values('reporter__first_name', 'reporter__last_name')) == [d]
-True
-
-# If you delete a reporter, his articles will be deleted.
->>> Article.objects.all()
-[<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>, <Article: This is a test>, <Article: This is a test>]
->>> Reporter.objects.order_by('first_name')
-[<Reporter: John Smith>, <Reporter: Paul Jones>]
->>> r2.delete()
->>> Article.objects.all()
-[<Article: John's second story>, <Article: This is a test>, <Article: This is a test>, <Article: This is a test>]
->>> Reporter.objects.order_by('first_name')
-[<Reporter: John Smith>]
-
-# You can delete using a JOIN in the query.
->>> Reporter.objects.filter(article__headline__startswith='This').delete()
->>> Reporter.objects.all()
-[]
->>> Article.objects.all()
-[]
-
-# Check that Article.objects.select_related().dates() works properly when
-# there are multiple Articles with the same date but different foreign-key
-# objects (Reporters).
->>> r1 = Reporter.objects.create(first_name='Mike', last_name='Royko', email='royko@suntimes.com')
->>> r2 = Reporter.objects.create(first_name='John', last_name='Kass', email='jkass@tribune.com')
->>> a1 = Article.objects.create(headline='First', pub_date=datetime(1980, 4, 23), reporter=r1)
->>> a2 = Article.objects.create(headline='Second', pub_date=datetime(1980, 4, 23), reporter=r2)
->>> Article.objects.select_related().dates('pub_date', 'day')
-[datetime.datetime(1980, 4, 23, 0, 0)]
->>> Article.objects.select_related().dates('pub_date', 'month')
-[datetime.datetime(1980, 4, 1, 0, 0)]
->>> Article.objects.select_related().dates('pub_date', 'year')
-[datetime.datetime(1980, 1, 1, 0, 0)]
-"""}
View
372 tests/modeltests/many_to_one/tests.py
@@ -0,0 +1,372 @@
+from datetime import datetime
+from django.test import TestCase
+from django.core.exceptions import FieldError
+from models import Article, Reporter
+
+class ManyToOneTests(TestCase):
+
+ def setUp(self):
+ # Create a few Reporters.
+ self.r = Reporter(first_name='John', last_name='Smith', email='john@example.com')
+ self.r.save()
+ self.r2 = Reporter(first_name='Paul', last_name='Jones', email='paul@example.com')
+ self.r2.save()
+ # Create an Article.
+ self.a = Article(id=None, headline="This is a test",
+ pub_date=datetime(2005, 7, 27), reporter=self.r)
+ self.a.save()
+
+ def test_get(self):
+ # Article objects have access to their related Reporter objects.
+ r = self.a.reporter
+ self.assertEqual(r.id, self.r.id)
+ # These are strings instead of unicode strings because that's what was used in
+ # the creation of this reporter (and we haven't refreshed the data from the
+ # database, which always returns unicode strings).
+ self.assertEqual((r.first_name, self.r.last_name), ('John', 'Smith'))
+
+ def test_create(self):
+ # You can also instantiate an Article by passing the Reporter's ID
+ # instead of a Reporter object.
+ a3 = Article(id=None, headline="Third article",
+ pub_date=datetime(2005, 7, 27), reporter_id=self.r.id)
+ a3.save()
+ self.assertEqual(a3.reporter.id, self.r.id)
+
+ # Similarly, the reporter ID can be a string.
+ a4 = Article(id=None, headline="Fourth article",
+ pub_date=datetime(2005, 7, 27), reporter_id=str(self.r.id))
+ a4.save()
+ self.assertEqual(repr(a4.reporter), "<Reporter: John Smith>")
+
+ def test_add(self):
+ # Create an Article via the Reporter object.
+ new_article = self.r.article_set.create(headline="John's second story",
+ pub_date=datetime(2005, 7, 29))
+ self.assertEqual(repr(new_article), "<Article: John's second story>")
+ self.assertEqual(new_article.reporter.id, self.r.id)
+
+ # Create a new article, and add it to the article set.
+ new_article2 = Article(headline="Paul's story", pub_date=datetime(2006, 1, 17))
+ self.r.article_set.add(new_article2)
+ self.assertEqual(new_article2.reporter.id, self.r.id)
+ self.assertQuerysetEqual(self.r.article_set.all(),
+ [
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ "<Article: This is a test>",
+ ])
+
+ # Add the same article to a different article set - check that it moves.
+ self.r2.article_set.add(new_article2)
+ self.assertEqual(new_article2.reporter.id, self.r2.id)
+ self.assertQuerysetEqual(self.r2.article_set.all(), ["<Article: Paul's story>"])
+
+ # Adding an object of the wrong type raises TypeError.
+ self.assertRaises(TypeError, self.r.article_set.add, self.r2)
+ self.assertQuerysetEqual(self.r.article_set.all(),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+
+ def test_assign(self):
+ new_article = self.r.article_set.create(headline="John's second story",
+ pub_date=datetime(2005, 7, 29))
+ new_article2 = self.r2.article_set.create(headline="Paul's story",
+ pub_date=datetime(2006, 1, 17))
+ # Assign the article to the reporter directly using the descriptor.
+ new_article2.reporter = self.r
+ new_article2.save()
+ self.assertEqual(repr(new_article2.reporter), "<Reporter: John Smith>")
+ self.assertEqual(new_article2.reporter.id, self.r.id)
+ self.assertQuerysetEqual(self.r.article_set.all(), [
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(self.r2.article_set.all(), [])
+ # Set the article back again using set descriptor.
+ self.r2.article_set = [new_article, new_article2]
+ self.assertQuerysetEqual(self.r.article_set.all(), ["<Article: This is a test>"])
+ self.assertQuerysetEqual(self.r2.article_set.all(),
+ [
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ ])
+
+ # Funny case - assignment notation can only go so far; because the
+ # ForeignKey cannot be null, existing members of the set must remain.
+ self.r.article_set = [new_article]
+ self.assertQuerysetEqual(self.r.article_set.all(),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(self.r2.article_set.all(), ["<Article: Paul's story>"])
+ # Reporter cannot be null - there should not be a clear or remove method
+ self.assertFalse(hasattr(self.r2.article_set, 'remove'))
+ self.assertFalse(hasattr(self.r2.article_set, 'clear'))
+
+ def test_selects(self):
+ new_article = self.r.article_set.create(headline="John's second story",
+ pub_date=datetime(2005, 7, 29))
+ new_article2 = self.r2.article_set.create(headline="Paul's story",
+ pub_date=datetime(2006, 1, 17))
+ # Reporter objects have access to their related Article objects.
+ self.assertQuerysetEqual(self.r.article_set.all(), [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(self.r.article_set.filter(headline__startswith='This'),
+ ["<Article: This is a test>"])
+ self.assertEqual(self.r.article_set.count(), 2)
+ self.assertEqual(self.r2.article_set.count(), 1)
+ # Get articles by id
+ self.assertQuerysetEqual(Article.objects.filter(id__exact=self.a.id),
+ ["<Article: This is a test>"])
+ self.assertQuerysetEqual(Article.objects.filter(pk=self.a.id),
+ ["<Article: This is a test>"])
+ # Query on an article property
+ self.assertQuerysetEqual(Article.objects.filter(headline__startswith='This'),
+ ["<Article: This is a test>"])
+ # The API automatically follows relationships as far as you need.
+ # Use double underscores to separate relationships.
+ # This works as many levels deep as you want. There's no limit.
+ # Find all Articles for any Reporter whose first name is "John".
+ self.assertQuerysetEqual(Article.objects.filter(reporter__first_name__exact='John'),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # Check that implied __exact also works
+ self.assertQuerysetEqual(Article.objects.filter(reporter__first_name='John'),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # Query twice over the related field.
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__first_name__exact='John',
+ reporter__last_name__exact='Smith'),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # The underlying query only makes one join when a related table is referenced twice.
+ queryset = Article.objects.filter(reporter__first_name__exact='John',
+ reporter__last_name__exact='Smith')
+ self.assertNumQueries(1, list, queryset)
+ self.assertEqual(queryset.query.get_compiler(queryset.db).as_sql()[0].count('INNER JOIN'), 1)
+
+ # The automatically joined table has a predictable name.
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__first_name__exact='John').extra(
+ where=["many_to_one_reporter.last_name='Smith'"]),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # ... and should work fine with the unicode that comes out of forms.Form.cleaned_data
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__first_name__exact='John'
+ ).extra(where=["many_to_one_reporter.last_name='%s'" % u'Smith']),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # Find all Articles for a Reporter.
+ # Use direct ID check, pk check, and object comparison
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__id__exact=self.r.id),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__pk=self.r.id),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter=self.r.id),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter=self.r),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__in=[self.r.id,self.r2.id]).distinct(),
+ [
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(reporter__in=[self.r,self.r2]).distinct(),
+ [
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ "<Article: This is a test>",
+ ])
+ # You can also use a queryset instead of a literal list of instances.
+ # The queryset must be reduced to a list of values using values(),
+ # then converted into a query
+ self.assertQuerysetEqual(
+ Article.objects.filter(
+ reporter__in=Reporter.objects.filter(first_name='John').values('pk').query
+ ).distinct(),
+ [
+ "<Article: John's second story>",
+ "<Article: This is a test>",
+ ])
+ # You need two underscores between "reporter" and "id" -- not one.
+ self.assertRaises(FieldError, Article.objects.filter, reporter_id__exact=self.r.id)
+ # You need to specify a comparison clause
+ self.assertRaises(FieldError, Article.objects.filter, reporter_id=self.r.id)
+
+ def test_reverse_selects(self):
+ a3 = Article.objects.create(id=None, headline="Third article",
+ pub_date=datetime(2005, 7, 27), reporter_id=self.r.id)
+ a4 = Article.objects.create(id=None, headline="Fourth article",
+ pub_date=datetime(2005, 7, 27), reporter_id=str(self.r.id))
+ # Reporters can be queried
+ self.assertQuerysetEqual(Reporter.objects.filter(id__exact=self.r.id),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(Reporter.objects.filter(pk=self.r.id),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(Reporter.objects.filter(first_name__startswith='John'),
+ ["<Reporter: John Smith>"])
+ # Reporters can query in opposite direction of ForeignKey definition
+ self.assertQuerysetEqual(Reporter.objects.filter(article__id__exact=self.a.id),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(Reporter.objects.filter(article__pk=self.a.id),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(Reporter.objects.filter(article=self.a.id),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(Reporter.objects.filter(article=self.a),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__in=[self.a.id,a3.id]).distinct(),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__in=[self.a.id,a3]).distinct(),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__in=[self.a,a3]).distinct(),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__headline__startswith='T'),
+ ["<Reporter: John Smith>", "<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__headline__startswith='T').distinct(),
+ ["<Reporter: John Smith>"])
+
+ # Counting in the opposite direction works in conjunction with distinct()
+ self.assertEqual(
+ Reporter.objects.filter(article__headline__startswith='T').count(), 2)
+ self.assertEqual(
+ Reporter.objects.filter(article__headline__startswith='T').distinct().count(), 1)
+
+ # Queries can go round in circles.
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__reporter__first_name__startswith='John'),
+ [
+ "<Reporter: John Smith>",
+ "<Reporter: John Smith>",
+ "<Reporter: John Smith>",
+ ])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__reporter__first_name__startswith='John').distinct(),
+ ["<Reporter: John Smith>"])
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__reporter__exact=self.r).distinct(),
+ ["<Reporter: John Smith>"])
+
+ # Check that implied __exact also works.
+ self.assertQuerysetEqual(
+ Reporter.objects.filter(article__reporter=self.r).distinct(),
+ ["<Reporter: John Smith>"])
+
+ # It's possible to use values() calls across many-to-one relations.
+ # (Note, too, that we clear the ordering here so as not to drag the
+ # 'headline' field into the columns being used to determine uniqueness)
+ d = {'reporter__first_name': u'John', 'reporter__last_name': u'Smith'}
+ self.assertEqual([d],
+ list(Article.objects.filter(reporter=self.r).distinct().order_by()
+ .values('reporter__first_name', 'reporter__last_name')))
+
+ def test_select_related(self):
+ # Check that Article.objects.select_related().dates() works properly when
+ # there are multiple Articles with the same date but different foreign-key
+ # objects (Reporters).
+ r1 = Reporter.objects.create(first_name='Mike', last_name='Royko', email='royko@suntimes.com')
+ r2 = Reporter.objects.create(first_name='John', last_name='Kass', email='jkass@tribune.com')
+ a1 = Article.objects.create(headline='First', pub_date=datetime(1980, 4, 23), reporter=r1)
+ a2 = Article.objects.create(headline='Second', pub_date=datetime(1980, 4, 23), reporter=r2)
+ self.assertEqual(list(Article.objects.select_related().dates('pub_date', 'day')),
+ [
+ datetime(1980, 4, 23, 0, 0),
+ datetime(2005, 7, 27, 0, 0),
+ ])
+ self.assertEqual(list(Article.objects.select_related().dates('pub_date', 'month')),
+ [
+ datetime(1980, 4, 1, 0, 0),
+ datetime(2005, 7, 1, 0, 0),
+ ])
+ self.assertEqual(list(Article.objects.select_related().dates('pub_date', 'year')),
+ [
+ datetime(1980, 1, 1, 0, 0),
+ datetime(2005, 1, 1, 0, 0),
+ ])
+
+ def test_delete(self):
+ new_article = self.r.article_set.create(headline="John's second story",
+ pub_date=datetime(2005, 7, 29))
+ new_article2 = self.r2.article_set.create(headline="Paul's story",
+ pub_date=datetime(2006, 1, 17))
+ a3 = Article.objects.create(id=None, headline="Third article",
+ pub_date=datetime(2005, 7, 27), reporter_id=self.r.id)
+ a4 = Article.objects.create(id=None, headline="Fourth article",
+ pub_date=datetime(2005, 7, 27), reporter_id=str(self.r.id))
+ # If you delete a reporter, his articles will be deleted.
+ self.assertQuerysetEqual(Article.objects.all(),
+ [
+ "<Article: Fourth article>",
+ "<Article: John's second story>",
+ "<Article: Paul's story>",
+ "<Article: Third article>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(Reporter.objects.order_by('first_name'),
+ [
+ "<Reporter: John Smith>",
+ "<Reporter: Paul Jones>",
+ ])
+ self.r2.delete()
+ self.assertQuerysetEqual(Article.objects.all(),
+ [
+ "<Article: Fourth article>",
+ "<Article: John's second story>",
+ "<Article: Third article>",
+ "<Article: This is a test>",
+ ])
+ self.assertQuerysetEqual(Reporter.objects.order_by('first_name'),
+ ["<Reporter: John Smith>"])
+ # You can delete using a JOIN in the query.
+ Reporter.objects.filter(article__headline__startswith='This').delete()
+ self.assertQuerysetEqual(Reporter.objects.all(), [])
+ self.assertQuerysetEqual(Article.objects.all(), [])
+
+ def test_regression_12876(self):
+ # Regression for #12876 -- Model methods that include queries that
+ # recursive don't cause recursion depth problems under deepcopy.
+ self.r.cached_query = Article.objects.filter(reporter=self.r)
+ from copy import deepcopy
+ self.assertEqual(repr(deepcopy(self.r)), "<Reporter: John Smith>")
Please sign in to comment.
Something went wrong with that request. Please try again.