diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index a4014db18..1d211c91b 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -2197,11 +2197,8 @@ class ShortMenuTest(TestCase): class ImportTest(TestCase): - def test_delete_related(self): - town = models.Town.objects.create(name='my-test') - self.assertEqual(models.Town.objects.filter(name='my-test').count(), 1) - # create an import, fields are not relevant... + def create_import(self): create_user() imp_model = models.ImporterModel.objects.create( klass='ishtar_common.models.Person', name='Person') @@ -2220,12 +2217,19 @@ class ImportTest(TestCase): user=models.IshtarUser.objects.all()[0], importer_type=importer_type, imported_file=mcc_operation_file) + return imprt + + def test_delete_related(self): + town = models.Town.objects.create(name='my-test') + self.assertEqual(models.Town.objects.filter(name='my-test').count(), 1) + + imprt = self.create_import() - town.imports.add(imprt) - imprt.delete() - # town should be deleted - self.assertEqual(models.Town.objects.filter(name='my-test').count(), - 0) + town.imports.add(imprt) + imprt.delete() + # town should be deleted + self.assertEqual(models.Town.objects.filter(name='my-test').count(), + 0) def test_keys(self): content_type = ContentType.objects.get_for_model( |