Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Migrated m2o_recursive and m2o_recursive2 tests, merging them into a …

…single package. Thanks to George Sakkis for the patches.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14163 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit 539af4deeceeda5c60265c9c2e107722a9ae0d28 1 parent fbc1fca
@freakboy3742 freakboy3742 authored
View
26 tests/modeltests/m2o_recursive/models.py
@@ -19,22 +19,10 @@ class Category(models.Model):
def __unicode__(self):
return self.name
-__test__ = {'API_TESTS':"""
-# Create a few Category objects.
->>> r = Category(id=None, name='Root category', parent=None)
->>> r.save()
->>> c = Category(id=None, name='Child category', parent=r)
->>> c.save()
-
->>> r.child_set.all()
-[<Category: Child category>]
->>> r.child_set.get(name__startswith='Child')
-<Category: Child category>
->>> print r.parent
-None
-
->>> c.child_set.all()
-[]
->>> c.parent
-<Category: Root category>
-"""}
+class Person(models.Model):
+ full_name = models.CharField(max_length=20)
+ mother = models.ForeignKey('self', null=True, related_name='mothers_child_set')
+ father = models.ForeignKey('self', null=True, related_name='fathers_child_set')
+
+ def __unicode__(self):
+ return self.full_name
View
38 tests/modeltests/m2o_recursive/tests.py
@@ -0,0 +1,38 @@
+from django.test import TestCase
+from models import Category, Person
+
+class ManyToOneRecursiveTests(TestCase):
+
+ def setUp(self):
+ self.r = Category(id=None, name='Root category', parent=None)
+ self.r.save()
+ self.c = Category(id=None, name='Child category', parent=self.r)
+ self.c.save()
+
+ def test_m2o_recursive(self):
+ self.assertQuerysetEqual(self.r.child_set.all(),
+ ['<Category: Child category>'])
+ self.assertEqual(self.r.child_set.get(name__startswith='Child').id, self.c.id)
+ self.assertEqual(self.r.parent, None)
+ self.assertQuerysetEqual(self.c.child_set.all(), [])
+ self.assertEqual(self.c.parent.id, self.r.id)
+
+class MultipleManyToOneRecursiveTests(TestCase):
+
+ def setUp(self):
+ self.dad = Person(full_name='John Smith Senior', mother=None, father=None)
+ self.dad.save()
+ self.mom = Person(full_name='Jane Smith', mother=None, father=None)
+ self.mom.save()
+ self.kid = Person(full_name='John Smith Junior', mother=self.mom, father=self.dad)
+ self.kid.save()
+
+ def test_m2o_recursive2(self):
+ self.assertEqual(self.kid.mother.id, self.mom.id)
+ self.assertEqual(self.kid.father.id, self.dad.id)
+ self.assertQuerysetEqual(self.dad.fathers_child_set.all(),
+ ['<Person: John Smith Junior>'])
+ self.assertQuerysetEqual(self.mom.mothers_child_set.all(),
+ ['<Person: John Smith Junior>'])
+ self.assertQuerysetEqual(self.kid.mothers_child_set.all(), [])
+ self.assertQuerysetEqual(self.kid.fathers_child_set.all(), [])
View
0  tests/modeltests/m2o_recursive2/__init__.py
No changes.
View
43 tests/modeltests/m2o_recursive2/models.py
@@ -1,43 +0,0 @@
-"""
-12. Relating a model to another model more than once
-
-In this example, a ``Person`` can have a ``mother`` and ``father`` -- both of
-which are other ``Person`` objects.
-
-Set ``related_name`` to designate what the reverse relationship is called.
-"""
-
-from django.db import models
-
-class Person(models.Model):
- full_name = models.CharField(max_length=20)
- mother = models.ForeignKey('self', null=True, related_name='mothers_child_set')
- father = models.ForeignKey('self', null=True, related_name='fathers_child_set')
-
- def __unicode__(self):
- return self.full_name
-
-__test__ = {'API_TESTS':"""
-# Create two Person objects -- the mom and dad in our family.
->>> dad = Person(full_name='John Smith Senior', mother=None, father=None)
->>> dad.save()
->>> mom = Person(full_name='Jane Smith', mother=None, father=None)
->>> mom.save()
-
-# Give mom and dad a kid.
->>> kid = Person(full_name='John Smith Junior', mother=mom, father=dad)
->>> kid.save()
-
->>> kid.mother
-<Person: Jane Smith>
->>> kid.father
-<Person: John Smith Senior>
->>> dad.fathers_child_set.all()
-[<Person: John Smith Junior>]
->>> mom.mothers_child_set.all()
-[<Person: John Smith Junior>]
->>> kid.mothers_child_set.all()
-[]
->>> kid.fathers_child_set.all()
-[]
-"""}
Please sign in to comment.
Something went wrong with that request. Please try again.