diff options
Diffstat (limited to 'tests/m2m_intermediary/tests.py')
-rw-r--r-- | tests/m2m_intermediary/tests.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tests/m2m_intermediary/tests.py b/tests/m2m_intermediary/tests.py new file mode 100644 index 0000000000..f261f23546 --- /dev/null +++ b/tests/m2m_intermediary/tests.py @@ -0,0 +1,41 @@ +from __future__ import absolute_import + +from datetime import datetime + +from django.test import TestCase +from django.utils import six + +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: (six.text_type(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: (six.text_type(w.reporter), w.position) + ) |