所以我有各种各样的信号和处理程序,它们通过应用程序发送。但是,当我执行测试/进入“测试模式”时,我希望禁用这些处理程序。
在测试模式下是否有一种特定于Django的禁用信号/处理程序的方法?我可以想到一个非常简单的方法(在if TESTING子句中包含处理程序)但我想知道Django中是否有更好的方法?...
不,那里没有。您可以轻松地建立条件连接:
import sys
if not 'test' in sys.argv:
signal.connect(listener, sender=FooModel)
我在寻找禁用一组测试用例的信号时发现了这个问题,而且Germano的回答引导我找到解决方案,但它采用相反的方法,所以我想我会添加它。
在您的测试类中:
class MyTest(TestCase):
def setUp(self):
# do some setup
signal.disconnect(listener, sender=FooModel)
而不是添加决策代码来添加信号,而是在测试时禁用它,这对我来说感觉就像是一个更好的解决方案(因为测试应该围绕代码编写,而不是围绕测试编写代码)。希望对同一条船上的人有用!
编辑:自写这篇文章以来,我已经介绍了另一种禁用信号进行测试的方法。这需要factory_boy包(v2.4.0 +),这对于简化Django中的测试非常有用。你真的很喜欢选择:
import factory
from django.db.models import signals
class MyTest(TestCase):
@factory.django.mute_signals(signals.pre_save, signals.post_save)
def test_something(self):
警告感谢ups:它会在工厂内部和创建对象时静音信号,但是当你想要显式保存()时,不会进一步测试内部 - 信号将在那里取消静音。如果这是一个问题,那么在setUp中使用简单断开可能是要走的路。
如果您不想使用FactoryBoy,这里有一个关于如何在测试中禁用特定信号的导入的完整示例。
from django.db.models import signals
from myapp.models import MyModel
class MyTest(TestCase):
def test_no_signal(self):
signals.post_save.disconnect(sender=MyModel, dispatch_uid="my_id")
... after this point, the signal is disabled ...
这应与您的接收器匹配,此示例将匹配此接收器:
@receiver(post_save, sender=MyModel, dispatch_uid="my_id")
我试图在没有指定dispatch_uid的情况下禁用信号,但它不起作用。
我有一个类似的问题,无法使用signals.post_save.disconnect()
成功断开我的信号。找到this替代方法,创建一个装饰器来覆盖指定测试和信号上的SUSPEND_SIGNALS
设置。可能对同一条船上的任何人都有用。
首先,创建装饰器:
import functools
from django.conf import settings
from django.dispatch import receiver
def suspendingreceiver(signal, **decorator_kwargs):
def our_wrapper(func):
@receiver(signal, **decorator_kwargs)
@functools.wraps(func)
def fake_receiver(sender, **kwargs):
if settings.SUSPEND_SIGNALS:
return
return func(sender, **kwargs)
return fake_receiver
return our_wrapper
用新的替换信号上常用的@receiver
装饰器:
@suspendingreceiver(post_save, sender=MyModel)
def mymodel_post_save(sender, **kwargs):
work()
在你的TestCase上使用Django的override_settings()
:
@override_settings(SUSPEND_SIGNALS=True)
class MyTestCase(TestCase):
def test_method(self):
Model.objects.create() # post_save_receiver won't execute
感谢撰写博客的Josh Smeaton。
如果将接收器连接到AppConfig.ready
中的信号(文档推荐),请参阅https://docs.djangoproject.com/en/2.2/topics/signals/#connecting-receiver-functions,您可以为其他信号接收器的测试创建替代AppConfig
。