Commit b12428de authored by Russell Keith-Magee's avatar Russell Keith-Magee
Browse files

[1.2.X] Migrated m2o_recursive and m2o_recursive2 tests, merging them into a... 

[1.2.X] Migrated m2o_recursive and m2o_recursive2 tests, merging them into a single package. Thanks to George Sakkis for the patches.

Backport of r14163 from trunk.

git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@14169 bcc190cf-cafb-0310-a4f2-bffc1f526a37
parent d327611b

tests/modeltests/m2o_recursive/models.py

+7 −19
Original line number Diff line number Diff line
@@ -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

tests/modeltests/m2o_recursive/tests.py

0 → 100644
+38 −0
Original line number Diff line number Diff line 
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(), [])

tests/modeltests/m2o_recursive2/__init__.py

deleted100644 → 0
+0 −0

Empty file deleted.

tests/modeltests/m2o_recursive2/models.py

deleted100644 → 0
+0 −43
Original line number Diff line number Diff line 
"""

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()

[]

"""}