Django工厂男孩在工厂中模拟模型方法

Question

我的模型的保存方法是在保存后调用 api。为了测试我的应用程序，我使用DjangoModelFactory生成用于测试的对象。但是，api仍然被调用。

class MyClass(models.Model):
    def save(self, *args, **kwargs):
        super().save(*args, **kwargs)
        self.call_api()

我尝试过模拟该方法，但它不起作用

#...
from .models import MyModel

@pytest.mark.django_db
@patch("MyModel.call_api")
class MyModelFactory(factory.django.DjangoModelFactory, factory.base.BaseFactory):
    class Meta:
        model = MyModel

我的问题是，当我将它与工厂一起使用时，如何使用模拟方法？

Answer 1

正确的扩展点是Factory._create：

class MyModelFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = MyModel

    @classmethod
    def _create(cls, model_class, *args, **kwargs):
        with patch("MyModel.call_api"):
            return super()._create(model_class, *args, **kwargs)

factory.base.Factory只是一个快速说明：除了 之外，您不需要继承factory.django.DjangoModelFactory，后者已经是前者的子类。