Method test_map_field_reference

tests/test_dereference.py:698–760

(self)

        Group.drop_collection()

    def test_map_field_reference(self):
        class User(Document):
            name = StringField()

        class Group(Document):
            members = MapField(ReferenceField(User))

        User.drop_collection()
        Group.drop_collection()

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

        group = Group(members={str(u.id): u for u in members})
        group.save()

        group = Group(members={str(u.id): u for u in 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 == 2

            for _, m in group_obj.members.items():
                assert isinstance(m, User)

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

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

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

            for k, m in group_obj.members.items():
                assert isinstance(m, User)

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

            group_objs = Group.objects.select_related()
            assert q == 2

            for group_obj in group_objs:
                _ = [m for m in group_obj.members]
                assert q == 2