Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

[1.2.X] Converted signals tests from doctests to unittests. We have a…

…lways been at war with doctests. Backport of [14131].

Conflicts:

	tests/modeltests/signals/models.py

git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@14132 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit b014c81fa73c3ac613835a650da441ab06e8c6ee 1 parent e226a11
@alex alex authored
Showing with 124 additions and 108 deletions.
  1. +1 −105 tests/modeltests/signals/models.py
  2. +123 −3 tests/modeltests/signals/tests.py
View
106 tests/modeltests/signals/models.py
@@ -4,114 +4,10 @@
from django.db import models
+
class Person(models.Model):
first_name = models.CharField(max_length=20)
last_name = models.CharField(max_length=20)
def __unicode__(self):
return u"%s %s" % (self.first_name, self.last_name)
-
-def pre_save_test(signal, sender, instance, **kwargs):
- print 'pre_save signal,', instance
- if kwargs.get('raw'):
- print 'Is raw'
-
-def post_save_test(signal, sender, instance, **kwargs):
- print 'post_save signal,', instance
- if 'created' in kwargs:
- if kwargs['created']:
- print 'Is created'
- else:
- print 'Is updated'
- if kwargs.get('raw'):
- print 'Is raw'
-
-def pre_delete_test(signal, sender, instance, **kwargs):
- print 'pre_delete signal,', instance
- print 'instance.id is not None: %s' % (instance.id != None)
-
-# #8285: signals can be any callable
-class PostDeleteHandler(object):
- def __call__(self, signal, sender, instance, **kwargs):
- print 'post_delete signal,', instance
- print 'instance.id is None: %s' % (instance.id == None)
-
-post_delete_test = PostDeleteHandler()
-
-__test__ = {'API_TESTS':"""
-
-# Save up the number of connected signals so that we can check at the end
-# that all the signals we register get properly unregistered (#9989)
->>> pre_signals = (len(models.signals.pre_save.receivers),
-... len(models.signals.post_save.receivers),
-... len(models.signals.pre_delete.receivers),
-... len(models.signals.post_delete.receivers))
-
->>> models.signals.pre_save.connect(pre_save_test)
->>> models.signals.post_save.connect(post_save_test)
->>> models.signals.pre_delete.connect(pre_delete_test)
->>> models.signals.post_delete.connect(post_delete_test)
-
->>> p1 = Person(first_name='John', last_name='Smith')
->>> p1.save()
-pre_save signal, John Smith
-post_save signal, John Smith
-Is created
-
->>> p1.first_name = 'Tom'
->>> p1.save()
-pre_save signal, Tom Smith
-post_save signal, Tom Smith
-Is updated
-
-# Calling an internal method purely so that we can trigger a "raw" save.
->>> p1.save_base(raw=True)
-pre_save signal, Tom Smith
-Is raw
-post_save signal, Tom Smith
-Is updated
-Is raw
-
->>> p1.delete()
-pre_delete signal, Tom Smith
-instance.id is not None: True
-post_delete signal, Tom Smith
-instance.id is None: False
-
->>> p2 = Person(first_name='James', last_name='Jones')
->>> p2.id = 99999
->>> p2.save()
-pre_save signal, James Jones
-post_save signal, James Jones
-Is created
-
->>> p2.id = 99998
->>> p2.save()
-pre_save signal, James Jones
-post_save signal, James Jones
-Is created
-
->>> p2.delete()
-pre_delete signal, James Jones
-instance.id is not None: True
-post_delete signal, James Jones
-instance.id is None: False
-
->>> Person.objects.all()
-[<Person: James Jones>]
-
->>> models.signals.post_delete.disconnect(post_delete_test)
->>> models.signals.pre_delete.disconnect(pre_delete_test)
->>> models.signals.post_save.disconnect(post_save_test)
->>> models.signals.pre_save.disconnect(pre_save_test)
-
-# Check that all our signals got disconnected properly.
->>> post_signals = (len(models.signals.pre_save.receivers),
-... len(models.signals.post_save.receivers),
-... len(models.signals.pre_delete.receivers),
-... len(models.signals.post_delete.receivers))
-
->>> pre_signals == post_signals
-True
-
-"""}
View
126 tests/modeltests/signals/tests.py
@@ -1,6 +1,18 @@
from django.db.models import signals
from django.test import TestCase
-from modeltests.signals.models import Person
+
+from models import Person
+
+
+# #8285: signals can be any callable
+class PostDeleteHandler(object):
+ def __init__(self, data):
+ self.data = data
+
+ def __call__(self, signal, sender, instance, **kwargs):
+ self.data.append(
+ (instance, instance.id is None)
+ )
class MyReceiver(object):
def __init__(self, param):
@@ -12,6 +24,115 @@ def __call__(self, signal, sender, **kwargs):
signal.disconnect(receiver=self, sender=sender)
class SignalTests(TestCase):
+ def test_basic(self):
+ # Save up the number of connected signals so that we can check at the
+ # end that all the signals we register get properly unregistered (#9989)
+ pre_signals = (
+ len(signals.pre_save.receivers),
+ len(signals.post_save.receivers),
+ len(signals.pre_delete.receivers),
+ len(signals.post_delete.receivers),
+ )
+
+ data = []
+
+ def pre_save_test(signal, sender, instance, **kwargs):
+ data.append(
+ (instance, kwargs.get("raw", False))
+ )
+ signals.pre_save.connect(pre_save_test)
+
+ def post_save_test(signal, sender, instance, **kwargs):
+ data.append(
+ (instance, kwargs.get("created"), kwargs.get("raw", False))
+ )
+ signals.post_save.connect(post_save_test)
+
+ def pre_delete_test(signal, sender, instance, **kwargs):
+ data.append(
+ (instance, instance.id is None)
+ )
+ signals.pre_delete.connect(pre_delete_test)
+
+ post_delete_test = PostDeleteHandler(data)
+ signals.post_delete.connect(post_delete_test)
+
+ p1 = Person(first_name="John", last_name="Smith")
+ self.assertEqual(data, [])
+ p1.save()
+ self.assertEqual(data, [
+ (p1, False),
+ (p1, True, False),
+ ])
+ data[:] = []
+
+ p1.first_name = "Tom"
+ p1.save()
+ self.assertEqual(data, [
+ (p1, False),
+ (p1, False, False),
+ ])
+ data[:] = []
+
+ # Calling an internal method purely so that we can trigger a "raw" save.
+ p1.save_base(raw=True)
+ self.assertEqual(data, [
+ (p1, True),
+ (p1, False, True),
+ ])
+ data[:] = []
+
+ p1.delete()
+ self.assertEqual(data, [
+ (p1, False),
+ (p1, False),
+ ])
+ data[:] = []
+
+ p2 = Person(first_name="James", last_name="Jones")
+ p2.id = 99999
+ p2.save()
+ self.assertEqual(data, [
+ (p2, False),
+ (p2, True, False),
+ ])
+ data[:] = []
+
+ p2.id = 99998
+ p2.save()
+ self.assertEqual(data, [
+ (p2, False),
+ (p2, True, False),
+ ])
+ data[:] = []
+
+ p2.delete()
+ self.assertEqual(data, [
+ (p2, False),
+ (p2, False)
+ ])
+
+ self.assertQuerysetEqual(
+ Person.objects.all(), [
+ "James Jones",
+ ],
+ unicode
+ )
+
+ signals.post_delete.disconnect(post_delete_test)
+ signals.pre_delete.disconnect(pre_delete_test)
+ signals.post_save.disconnect(post_save_test)
+ signals.pre_save.disconnect(pre_save_test)
+
+ # Check that all our signals got disconnected properly.
+ post_signals = (
+ len(signals.pre_save.receivers),
+ len(signals.post_save.receivers),
+ len(signals.pre_delete.receivers),
+ len(signals.post_delete.receivers),
+ )
+ self.assertEqual(pre_signals, post_signals)
+
def test_disconnect_in_dispatch(self):
"""
Test that signals that disconnect when being called don't mess future
@@ -21,8 +142,7 @@ def test_disconnect_in_dispatch(self):
signals.post_save.connect(sender=Person, receiver=a)
signals.post_save.connect(sender=Person, receiver=b)
p = Person.objects.create(first_name='John', last_name='Smith')
-
+
self.failUnless(a._run)
self.failUnless(b._run)
self.assertEqual(signals.post_save.receivers, [])
-
Please sign in to comment.
Something went wrong with that request. Please try again.