Django REST框架 - 用于单元测试的伪对象

Question

我正在为我的Django应用程序编写单元测试。 目前我正在使用factory_boy制作假对象进行测试。 这适用于我的大多数测试，但我遇到以下问题： 我的factory_boy对象没有显示在{model}.objects 。 例如，我正在尝试在序列化程序中测试以下方法：

def get_can_edit(self, obj):
    request = self.context.get('request')
    user = request.user
    admin = SimpleLazyObject(obj.admin)
    user = User.objects.get(username=request.user)
    return user == obj.admin

使用调试器完成它，我已经确定request.user正确拥有我的假用户，但User.objects没有我的假用户。

我想知道是否有一个简单的替代factory_boy，它实际上会将我的假对象添加到{model}.objects或者我只是错误地使用factory_boy？ 或者也许有一种完全不同的方法......谁知道呢。

这是代码：

序列化器

考试

Answer 1

你需要从继承DjangoModelFactory而不是Factory的基类中的UserFactory才能将其保存到数据库中的对象。

来自factory_boy文档：

Django Model所有工厂都应该使用DjangoModelFactory基类。

class UserFactory(factory.DjangoModelFactory): # use DjangoModelFactory base class

    class Meta: 
        model = models.User