queryset.py 文件源码

def setUp(self):
        """Setup the class."""
        self.dict_model = [
            {u"test": u"Hello %s" % count} for count in range(5)
        ]
        QuerySet.as_pymongo = MagicMock(return_value=self.dict_model)

        class TestModel(Document):
            test = db.StringField()

        self.as_pymongo = TestModel.objects.as_pymongo
        self.model_cls = TestModel
        self.model = [
            self.model_cls(test="Hello %s" % count) for count in range(5)
        ]