diff options
author | Florian Apolloner <florian@apolloner.eu> | 2013-02-26 09:53:47 +0100 |
---|---|---|
committer | Florian Apolloner <florian@apolloner.eu> | 2013-02-26 14:36:57 +0100 |
commit | 89f40e36246100df6a11316c31a76712ebc6c501 (patch) | |
tree | 6e65639683ddaf2027908d1ecb1739e0e2ff853b /tests/save_delete_hooks | |
parent | b3d2ccb5bfbaf6e7fe1f98843baaa48c35a70950 (diff) | |
download | django-89f40e36246100df6a11316c31a76712ebc6c501.tar.gz |
Merged regressiontests and modeltests into the test root.
Diffstat (limited to 'tests/save_delete_hooks')
-rw-r--r-- | tests/save_delete_hooks/__init__.py | 0 | ||||
-rw-r--r-- | tests/save_delete_hooks/models.py | 35 | ||||
-rw-r--r-- | tests/save_delete_hooks/tests.py | 33 |
3 files changed, 68 insertions, 0 deletions
diff --git a/tests/save_delete_hooks/__init__.py b/tests/save_delete_hooks/__init__.py new file mode 100644 index 0000000000..e69de29bb2 --- /dev/null +++ b/tests/save_delete_hooks/__init__.py diff --git a/tests/save_delete_hooks/models.py b/tests/save_delete_hooks/models.py new file mode 100644 index 0000000000..a6e1abfb77 --- /dev/null +++ b/tests/save_delete_hooks/models.py @@ -0,0 +1,35 @@ +""" +13. Adding hooks before/after saving and deleting + +To execute arbitrary code around ``save()`` and ``delete()``, just subclass +the methods. +""" +from __future__ import unicode_literals + +from django.db import models +from django.utils.encoding import python_2_unicode_compatible + + +@python_2_unicode_compatible +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 __str__(self): + return "%s %s" % (self.first_name, self.last_name) + + def save(self, *args, **kwargs): + self.data.append("Before save") + # Call the "real" save() method + super(Person, self).save(*args, **kwargs) + self.data.append("After save") + + def delete(self): + self.data.append("Before deletion") + # Call the "real" delete() method + super(Person, self).delete() + self.data.append("After deletion") diff --git a/tests/save_delete_hooks/tests.py b/tests/save_delete_hooks/tests.py new file mode 100644 index 0000000000..42e0d4a80e --- /dev/null +++ b/tests/save_delete_hooks/tests.py @@ -0,0 +1,33 @@ +from __future__ import absolute_import + +from django.test import TestCase +from django.utils import six + +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", + ], + six.text_type + ) + + p.delete() + self.assertEqual(p.data, [ + "Before save", + "After save", + "Before deletion", + "After deletion", + ]) + self.assertQuerysetEqual(Person.objects.all(), []) |