diff options
Diffstat (limited to 'ishtar_common/tests.py')
-rw-r--r-- | ishtar_common/tests.py | 18 |
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 |