diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/signals/tests.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/tests/signals/tests.py b/tests/signals/tests.py index 0ed0f85f6b..b52cf1d11c 100644 --- a/tests/signals/tests.py +++ b/tests/signals/tests.py @@ -4,7 +4,7 @@ from django.apps.registry import Apps from django.db import models from django.db.models import signals from django.dispatch import receiver -from django.test import TestCase +from django.test import TestCase, mock from django.test.utils import isolate_apps from django.utils import six @@ -258,6 +258,19 @@ class SignalTests(BaseSignalTest): self.assertTrue(b._run) self.assertEqual(signals.post_save.receivers, []) + @mock.patch('weakref.ref') + def test_lazy_model_signal(self, ref): + def callback(sender, args, **kwargs): + pass + signals.pre_init.connect(callback) + signals.pre_init.disconnect(callback) + self.assertTrue(ref.called) + ref.reset_mock() + + signals.pre_init.connect(callback, weak=False) + signals.pre_init.disconnect(callback) + ref.assert_not_called() + class LazyModelRefTest(BaseSignalTest): def setUp(self): |
