Add a command to clean Ishtar DB.

1 files changed, 18 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index f22e42e38..10584e4f2 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -17,6 +17,7 @@
 
 # See the file COPYING for details.
 
+from StringIO import StringIO
 from bs4 import BeautifulSoup as Soup
 
 from django.conf import settings
@@ -25,6 +26,7 @@ from django.contrib.contenttypes.models import ContentType
 from django.core.cache import cache
 from django.core.exceptions import ValidationError
 from django.core.files.base import File as DjangoFile
+from django.core.management import call_command
 from django.core.urlresolvers import reverse
 from django.template.defaultfilters import slugify
 from django.test import TestCase
@@ -76,6 +78,22 @@ def create_user():
     return username, password, user
 
 
+class CommandsTestCase(TestCase):
+    def test_clean_ishtar(self):
+        """
+        Clean ishtar db
+        """
+        from archaeological_operations.models import Parcel
+        p = Parcel.objects.create(
+            town=models.Town.objects.create(name='test', numero_insee='25000'))
+        parcel_nb = Parcel.objects.count()
+        out = StringIO()
+        call_command('clean_ishtar', stdout=out)
+        # no operation or file attached - the parcel should have disappear
+        self.assertEqual(parcel_nb - 1, Parcel.objects.count())
+        self.assertEqual(Parcel.objects.filter(pk=p.pk).count(), 0)
+
+
 class WizardTestFormData(object):
     """
     Test set to simulate wizard steps