1. from django.test import TestCase

  2. 

  3. from .models import Category, Person

  4. 

  5. 

  6. class ManyToOneRecursiveTests(TestCase):

  7.     @classmethod

  8.     def setUpTestData(cls):

  9.         cls.r = Category.objects.create(id=None, name="Root category", parent=None)

  10.         cls.c = Category.objects.create(id=None, name="Child category", parent=cls.r)

  11. 

  12.     def test_m2o_recursive(self):

  13.         self.assertSequenceEqual(self.r.child_set.all(), [self.c])

  14.         self.assertEqual(self.r.child_set.get(name__startswith="Child").id, self.c.id)

  15.         self.assertIsNone(self.r.parent)

  16.         self.assertSequenceEqual(self.c.child_set.all(), [])

  17.         self.assertEqual(self.c.parent.id, self.r.id)

  18. 

  19. 

  20. class MultipleManyToOneRecursiveTests(TestCase):

  21.     @classmethod

  22.     def setUpTestData(cls):

  23.         cls.dad = Person.objects.create(

  24.             full_name="John Smith Senior", mother=None, father=None

  25.         )

  26.         cls.mom = Person.objects.create(

  27.             full_name="Jane Smith", mother=None, father=None

  28.         )

  29.         cls.kid = Person.objects.create(

  30.             full_name="John Smith Junior", mother=cls.mom, father=cls.dad

  31.         )

  32. 

  33.     def test_m2o_recursive2(self):

  34.         self.assertEqual(self.kid.mother.id, self.mom.id)

  35.         self.assertEqual(self.kid.father.id, self.dad.id)

  36.         self.assertSequenceEqual(self.dad.fathers_child_set.all(), [self.kid])

  37.         self.assertSequenceEqual(self.mom.mothers_child_set.all(), [self.kid])

  38.         self.assertSequenceEqual(self.kid.mothers_child_set.all(), [])

  39.         self.assertSequenceEqual(self.kid.fathers_child_set.all(), [])
About Sign Up Leave Feedback Contact