From 3879c5907454e14dd0128ef33c927580c0fe383c Mon Sep 17 00:00:00 2001
From: Alex Gaynor <alex.gaynor@gmail.com>
Date: Mon, 11 Oct 2010 17:41:44 +0000
Subject: [PATCH] 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
---
 tests/modeltests/save_delete_hooks/models.py | 37 +++++++-------------
 tests/modeltests/save_delete_hooks/tests.py  | 30 ++++++++++++++++
 2 files changed, 43 insertions(+), 24 deletions(-)
 create mode 100644 tests/modeltests/save_delete_hooks/tests.py

diff --git a/tests/modeltests/save_delete_hooks/models.py b/tests/modeltests/save_delete_hooks/models.py
index 54c9defd8b..515c7f6c91 100644
--- a/tests/modeltests/save_delete_hooks/models.py
+++ b/tests/modeltests/save_delete_hooks/models.py
@@ -7,37 +7,26 @@ the methods.
 
 from django.db import models
 
+
 class Person(models.Model):
     first_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):
         return u"%s %s" % (self.first_name, self.last_name)
 
-    def save(self, force_insert=False, force_update=False):
-        print "Before save"
+    def save(self, *args, **kwargs):
+        self.data.append("Before save")
          # Call the "real" save() method
-        super(Person, self).save(force_insert, force_update)
-        print "After save"
+        super(Person, self).save(*args, **kwargs)
+        self.data.append("After save")
 
     def delete(self):
-        print "Before deletion"
-        super(Person, self).delete() # Call the "real" delete() method
-        print "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()
-[]
-"""}
+        self.data.append("Before deletion")
+        # Call the "real" delete() method
+        super(Person, self).delete()
+        self.data.append("After deletion")
diff --git a/tests/modeltests/save_delete_hooks/tests.py b/tests/modeltests/save_delete_hooks/tests.py
new file mode 100644
index 0000000000..dc7b8ee12a
--- /dev/null
+++ b/tests/modeltests/save_delete_hooks/tests.py
@@ -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(), [])