Skip to content

Commit

Permalink
[1.2.X] Migrated m2m_intermediary doctests. Thanks to Alex Gaynor.
Browse files Browse the repository at this point in the history
Backport of r13787 from trunk.

git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@13804 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information
freakboy3742 committed Sep 12, 2010
1 parent 42f7bbd commit a4a19b8
Show file tree
Hide file tree
Showing 3 changed files with 43 additions and 37 deletions.
10 changes: 5 additions & 5 deletions tests/modeltests/m2m_and_m2o/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,21 +8,21 @@ class RelatedObjectTests(TestCase):
def test_m2m_and_m2o(self):
r = User.objects.create(username="russell")
g = User.objects.create(username="gustav")

i1 = Issue(num=1)
i1.client = r
i1.save()

i2 = Issue(num=2)
i2.client = r
i2.save()
i2.cc.add(r)

i3 = Issue(num=3)
i3.client = g
i3.save()
i3.cc.add(r)

self.assertQuerysetEqual(
Issue.objects.filter(client=r.id), [
1,
Expand All @@ -46,7 +46,7 @@ def test_m2m_and_m2o(self):
],
lambda i: i.num
)

# These queries combine results from the m2m and the m2o relationships.
# They're three ways of saying the same thing.
self.assertQuerysetEqual(
Expand Down
32 changes: 0 additions & 32 deletions tests/modeltests/m2m_intermediary/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,35 +34,3 @@ class Writer(models.Model):
def __unicode__(self):
return u'%s (%s)' % (self.reporter, self.position)

__test__ = {'API_TESTS':"""
# Create a few Reporters.
>>> r1 = Reporter(first_name='John', last_name='Smith')
>>> r1.save()
>>> r2 = Reporter(first_name='Jane', last_name='Doe')
>>> r2.save()
# Create an Article.
>>> from datetime import datetime
>>> a = Article(headline='This is a test', pub_date=datetime(2005, 7, 27))
>>> a.save()
# Create a few Writers.
>>> w1 = Writer(reporter=r1, article=a, position='Main writer')
>>> w1.save()
>>> w2 = Writer(reporter=r2, article=a, position='Contributor')
>>> w2.save()
# Play around with the API.
>>> a.writer_set.select_related().order_by('-position')
[<Writer: John Smith (Main writer)>, <Writer: Jane Doe (Contributor)>]
>>> w1.reporter
<Reporter: John Smith>
>>> w2.reporter
<Reporter: Jane Doe>
>>> w1.article
<Article: This is a test>
>>> w2.article
<Article: This is a test>
>>> r1.writer_set.all()
[<Writer: John Smith (Main writer)>]
"""}
38 changes: 38 additions & 0 deletions tests/modeltests/m2m_intermediary/tests.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
from datetime import datetime

from django.test import TestCase

from models import Reporter, Article, Writer


class M2MIntermediaryTests(TestCase):
def test_intermeiary(self):
r1 = Reporter.objects.create(first_name="John", last_name="Smith")
r2 = Reporter.objects.create(first_name="Jane", last_name="Doe")

a = Article.objects.create(
headline="This is a test", pub_date=datetime(2005, 7, 27)
)

w1 = Writer.objects.create(reporter=r1, article=a, position="Main writer")
w2 = Writer.objects.create(reporter=r2, article=a, position="Contributor")

self.assertQuerysetEqual(
a.writer_set.select_related().order_by("-position"), [
("John Smith", "Main writer"),
("Jane Doe", "Contributor"),
],
lambda w: (unicode(w.reporter), w.position)
)
self.assertEqual(w1.reporter, r1)
self.assertEqual(w2.reporter, r2)

self.assertEqual(w1.article, a)
self.assertEqual(w2.article, a)

self.assertQuerysetEqual(
r1.writer_set.all(), [
("John Smith", "Main writer")
],
lambda w: (unicode(w.reporter), w.position)
)

0 comments on commit a4a19b8

Please sign in to comment.