diff --git a/tests/integration/test_neighbors.py b/tests/integration/test_neighbors.py index 00abab52..a55dd267 100644 --- a/tests/integration/test_neighbors.py +++ b/tests/integration/test_neighbors.py @@ -9,13 +9,17 @@ from taiga.base.utils.db import filter_by_tags from taiga.base import neighbors as n from .. import factories as f -from ..utils import disconnect_signals +from ..utils import disconnect_signals, reconnect_signals def setup_module(): disconnect_signals() +def teardown_module(): + reconnect_signals() + + class TestGetAttribute: def test_no_attribute(self, object): object.first_name = "name" diff --git a/tests/utils.py b/tests/utils.py index e70b3249..a96eb52e 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,6 +1,19 @@ from django.db.models import signals -def disconnect_signals(): - signals.pre_save.receivers = [] - signals.post_save.receivers = [] +def signals_switch(): + pre_save = signals.pre_save.receivers + post_save = signals.post_save.receivers + + def disconnect(): + signals.pre_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()