Method test_recursive_reference

tests/test_dereference.py:274–338

Ensure that ReferenceFields can reference their own documents.

(self)

        assert isinstance(raw_data["members"][0], ObjectId)

    def test_recursive_reference(self):
        """Ensure that ReferenceFields can reference their own documents."""

        class Employee(Document):
            name = StringField()
            boss = ReferenceField("self")
            friends = ListField(ReferenceField("self"))

        Employee.drop_collection()

        bill = Employee(name="Bill Lumbergh")
        bill.save()

        michael = Employee(name="Michael Bolton")
        michael.save()

        samir = Employee(name="Samir Nagheenanajar")
        samir.save()

        friends = [michael, samir]
        peter = Employee(name="Peter Gibbons", boss=bill, friends=friends)
        peter.save()

        Employee(name="Funky Gibbon", boss=bill, friends=friends).save()
        Employee(name="Funky Gibbon", boss=bill, friends=friends).save()
        Employee(name="Funky Gibbon", boss=bill, friends=friends).save()

        with query_counter() as q:
            assert q == 0

            peter = Employee.objects.with_id(peter.id)
            assert q == 1

            peter.boss
            assert q == 2

            peter.friends
            assert q == 3

        # Document select_related
        with query_counter() as q:
            assert q == 0

            peter = Employee.objects.with_id(peter.id).select_related()
            assert q == 2

            assert peter.boss == bill
            assert q == 2

            assert peter.friends == friends
            assert q == 2

        # Queryset select_related
        with query_counter() as q:
            assert q == 0

            employees = Employee.objects(boss=bill).select_related()
            assert q == 2