Method test_generic_reference

tests/test_dereference.py:465–548

(self)

        )

    def test_generic_reference(self):
        class UserA(Document):
            name = StringField()

        class UserB(Document):
            name = StringField()

        class UserC(Document):
            name = StringField()

        class Group(Document):
            members = ListField(GenericReferenceField())

        UserA.drop_collection()
        UserB.drop_collection()
        UserC.drop_collection()
        Group.drop_collection()

        members = []
        for i in range(1, 51):
            a = UserA(name="User A %s" % i)
            a.save()

            b = UserB(name="User B %s" % i)
            b.save()

            c = UserC(name="User C %s" % i)
            c.save()

            members += [a, b, c]

        group = Group(members=members)
        group.save()

        group = Group(members=members)
        group.save()

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

            group_obj = Group.objects.first()
            assert q == 1

            _ = [m for m in group_obj.members]
            assert q == 4

            _ = [m for m in group_obj.members]
            assert q == 4

            for m in group_obj.members:
                assert "User" in m.__class__.__name__

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

            group_obj = Group.objects.first().select_related()
            assert q == 4