Admin: fix import edit

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(