Reconnect disconnected signals in integration tests

remotes/origin/enhancement/email-actions
Anler Hp 2014-06-02 16:57:48 +02:00
parent 9923e50603
commit aa7ca6e3fc
2 changed files with 21 additions and 4 deletions

View File

@ -9,13 +9,17 @@ from taiga.base.utils.db import filter_by_tags
from taiga.base import neighbors as n from taiga.base import neighbors as n
from .. import factories as f from .. import factories as f
from ..utils import disconnect_signals from ..utils import disconnect_signals, reconnect_signals
def setup_module(): def setup_module():
disconnect_signals() disconnect_signals()
def teardown_module():
reconnect_signals()
class TestGetAttribute: class TestGetAttribute:
def test_no_attribute(self, object): def test_no_attribute(self, object):
object.first_name = "name" object.first_name = "name"

View File

@ -1,6 +1,19 @@
from django.db.models import signals from django.db.models import signals
def disconnect_signals(): def signals_switch():
pre_save = signals.pre_save.receivers
post_save = signals.post_save.receivers
def disconnect():
signals.pre_save.receivers = [] signals.pre_save.receivers = []
signals.post_save.receivers = [] signals.post_save.receivers = []
def reconnect():
signals.pre_save.receivers = pre_save
signals.post_save.receivers = post_save
return disconnect, reconnect
disconnect_signals, reconnect_signals = signals_switch()