python - Django:模拟模型上的字段

Question

如何将模拟对象分配给此模型上的用户字段？无论如何都要绕过“SomeModel.user”必须是“User”实例'检查吗？

 class SomeModel(models.Model):
     user = models.ForeignKey(User)

我不会保存这个模型，它纯粹是为了对该模型上的方法进行单元测试。我也不想创建一个真正的用户模型。

Answer 1

您可以使用unittest.patch来模拟属性:

class TestThing(django.test.TestCase):
    @unittest.patch('my_app.models.SomeModel.user', new_callable=unittest.PropertyMock)
    def test_thing(self, user_field):
        user_field.return_value = 5
        ...

这适用于 Django 字段。