/tests/modeltests/m2m_intermediary/tests.py

 1from datetime import datetime
 2
 3from django.test import TestCase
 4
 5from models import Reporter, Article, Writer
 6
 7
 8class M2MIntermediaryTests(TestCase):
 9    def test_intermeiary(self):
10        r1 = Reporter.objects.create(first_name="John", last_name="Smith")
11        r2 = Reporter.objects.create(first_name="Jane", last_name="Doe")
12
13        a = Article.objects.create(
14            headline="This is a test", pub_date=datetime(2005, 7, 27)
15        )
16
17        w1 = Writer.objects.create(reporter=r1, article=a, position="Main writer")
18        w2 = Writer.objects.create(reporter=r2, article=a, position="Contributor")
19
20        self.assertQuerysetEqual(
21            a.writer_set.select_related().order_by("-position"), [
22                ("John Smith", "Main writer"),
23                ("Jane Doe", "Contributor"),
24            ],
25            lambda w: (unicode(w.reporter), w.position)
26        )
27        self.assertEqual(w1.reporter, r1)
28        self.assertEqual(w2.reporter, r2)
29
30        self.assertEqual(w1.article, a)
31        self.assertEqual(w2.article, a)
32
33        self.assertQuerysetEqual(
34            r1.writer_set.all(), [
35                ("John Smith", "Main writer")
36            ],
37            lambda w: (unicode(w.reporter), w.position)
38        )