Fix issues between mute_signals() and factory inheritance (Closes #183).

Previously, if a factory was decorated with ``@mute_signals`` and one of
its descendant called another one of its descendant, signals weren't
unmuted properly.

1 files changed, 7 insertions, 2 deletions

diff --git a/factory/django.py b/factory/django.py
index 7050366..e823ee9 100644
--- a/factory/django.py
+++ b/factory/django.py
@@ -269,6 +269,9 @@ class mute_signals(object):
             signal.receivers = receivers
         self.paused = {}
 
+    def copy(self):
+        return mute_signals(*self.signals)
+
     def __call__(self, callable_obj):
         if isinstance(callable_obj, base.FactoryMetaClass):
             # Retrieve __func__, the *actual* callable object.
@@ -277,7 +280,8 @@ class mute_signals(object):
             @classmethod
             @functools.wraps(generate_method)
             def wrapped_generate(*args, **kwargs):
-                with self:
+                # A mute_signals() object is not reentrant; use a copy everytime.
+                with self.copy():
                     return generate_method(*args, **kwargs)
 
             callable_obj._generate = wrapped_generate
@@ -286,7 +290,8 @@ class mute_signals(object):
         else:
             @functools.wraps(callable_obj)
             def wrapper(*args, **kwargs):
-                with self:
+                # A mute_signals() object is not reentrant; use a copy everytime.
+                with self.copy():
                     return callable_obj(*args, **kwargs)
             return wrapper