From aa7ca6e3fcb97f58191eb5ef1f717f1b16125cb3 Mon Sep 17 00:00:00 2001 From: Anler Hp Date: Mon, 2 Jun 2014 16:57:48 +0200 Subject: [PATCH] Reconnect disconnected signals in integration tests --- tests/integration/test_neighbors.py | 6 +++++- tests/utils.py | 19 ++++++++++++++++--- 2 files changed, 21 insertions(+), 4 deletions(-) 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()