mirror of https://github.com/django/django.git
Converted save_delete_hooks tests from doctests to unittests. We have always been at war with doctests.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@14145 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
e01bce1bfb
commit
3879c59074
|
@ -7,37 +7,26 @@ the methods.
|
||||||
|
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
|
||||||
|
|
||||||
class Person(models.Model):
|
class Person(models.Model):
|
||||||
first_name = models.CharField(max_length=20)
|
first_name = models.CharField(max_length=20)
|
||||||
last_name = models.CharField(max_length=20)
|
last_name = models.CharField(max_length=20)
|
||||||
|
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super(Person, self).__init__(*args, **kwargs)
|
||||||
|
self.data = []
|
||||||
|
|
||||||
def __unicode__(self):
|
def __unicode__(self):
|
||||||
return u"%s %s" % (self.first_name, self.last_name)
|
return u"%s %s" % (self.first_name, self.last_name)
|
||||||
|
|
||||||
def save(self, force_insert=False, force_update=False):
|
def save(self, *args, **kwargs):
|
||||||
print "Before save"
|
self.data.append("Before save")
|
||||||
# Call the "real" save() method
|
# Call the "real" save() method
|
||||||
super(Person, self).save(force_insert, force_update)
|
super(Person, self).save(*args, **kwargs)
|
||||||
print "After save"
|
self.data.append("After save")
|
||||||
|
|
||||||
def delete(self):
|
def delete(self):
|
||||||
print "Before deletion"
|
self.data.append("Before deletion")
|
||||||
super(Person, self).delete() # Call the "real" delete() method
|
# Call the "real" delete() method
|
||||||
print "After deletion"
|
super(Person, self).delete()
|
||||||
|
self.data.append("After deletion")
|
||||||
__test__ = {'API_TESTS':"""
|
|
||||||
>>> p1 = Person(first_name='John', last_name='Smith')
|
|
||||||
>>> p1.save()
|
|
||||||
Before save
|
|
||||||
After save
|
|
||||||
|
|
||||||
>>> Person.objects.all()
|
|
||||||
[<Person: John Smith>]
|
|
||||||
|
|
||||||
>>> p1.delete()
|
|
||||||
Before deletion
|
|
||||||
After deletion
|
|
||||||
|
|
||||||
>>> Person.objects.all()
|
|
||||||
[]
|
|
||||||
"""}
|
|
||||||
|
|
|
@ -0,0 +1,30 @@
|
||||||
|
from django.test import TestCase
|
||||||
|
|
||||||
|
from models import Person
|
||||||
|
|
||||||
|
|
||||||
|
class SaveDeleteHookTests(TestCase):
|
||||||
|
def test_basic(self):
|
||||||
|
p = Person(first_name="John", last_name="Smith")
|
||||||
|
self.assertEqual(p.data, [])
|
||||||
|
p.save()
|
||||||
|
self.assertEqual(p.data, [
|
||||||
|
"Before save",
|
||||||
|
"After save",
|
||||||
|
])
|
||||||
|
|
||||||
|
self.assertQuerysetEqual(
|
||||||
|
Person.objects.all(), [
|
||||||
|
"John Smith",
|
||||||
|
],
|
||||||
|
unicode
|
||||||
|
)
|
||||||
|
|
||||||
|
p.delete()
|
||||||
|
self.assertEqual(p.data, [
|
||||||
|
"Before save",
|
||||||
|
"After save",
|
||||||
|
"Before deletion",
|
||||||
|
"After deletion",
|
||||||
|
])
|
||||||
|
self.assertQuerysetEqual(Person.objects.all(), [])
|
Loading…
Reference in New Issue