Method test_circular_reference

tests/test_dereference.py:364–395

Ensure you can handle circular references

(self)

        assert Post.objects.all()[0].user_lists == [[u1, u2], [u3]]

    def test_circular_reference(self):
        """Ensure you can handle circular references"""

        class Relation(EmbeddedDocument):
            name = StringField()
            person = ReferenceField("Person")

        class Person(Document):
            name = StringField()
            relations = ListField(EmbeddedDocumentField("Relation"))

            def __repr__(self):
                return "<Person: %s>" % self.name

        Person.drop_collection()
        mother = Person(name="Mother")
        daughter = Person(name="Daughter")

        mother.save()
        daughter.save()

        daughter_rel = Relation(name="Daughter", person=daughter)
        mother.relations.append(daughter_rel)
        mother.save()

        mother_rel = Relation(name="Daughter", person=mother)
        self_rel = Relation(name="Self", person=daughter)
        daughter.relations.append(mother_rel)
        daughter.relations.append(self_rel)
        daughter.save()

        assert "[<Person: Mother>, <Person: Daughter>]" == "%s" % Person.objects()

    def test_circular_reference_on_self(self):
        """Ensure you can handle circular references"""