1 files changed, 48 insertions, 0 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 3e1f16433..4597060da 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -2414,6 +2414,54 @@ class ImportTest(TestCase):
             )
         return imprt
 
+    def test_edit_import(self):
+        username = "username666"
+        password = "dcbqj7xnjkxnjsknx!@%"
+        user = User.objects.create_superuser(
+            username, "nomail@nomail.com", password
+        )
+        imprt = self.create_import()
+        c = Client()
+        c.login(username=username, password=password)
+
+
+        response = c.get(reverse("edit_import", kwargs={"pk": imprt.pk}))
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, imprt.importer_type)
+        self.assertContains(response, imprt.imported_file)
+        self.assertContains(response, imprt.imported_images)
+        self.assertContains(response, imprt.encoding)
+        self.assertContains(response, imprt.csv_sep)
+
+        imp_model = models.ImporterModel.objects.create(
+            klass="ishtar_common.models.Parcel", name="Parcel"
+        )
+        importer_type = models.ImporterType.objects.create(associated_models=imp_model)
+
+        dest = os.path.join(settings.MEDIA_ROOT, "MCC-operations-example.csv")
+        shutil.copy(
+            settings.ROOT_PATH
+            + "../archaeological_operations/tests/MCC-operations-example.csv",
+            dest,
+        )
+
+        data = {
+            "id_name": "Test Name",
+            "id_importer_type": importer_type,
+        }
+
+        response = c.post(reverse("edit_import", kwargs={"pk": imprt.pk}), data)
+        self.assertEqual(response.status_code, 302)
+
+        response = c.get(reverse("edit_import", kwargs={"pk": imprt.pk}))
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, imprt.name)
+        self.assertContains(response, imprt.importer_type)
+        self.assertContains(response, imprt.imported_file)
+        self.assertContains(response, imprt.imported_images)
+        self.assertContains(response, imprt.encoding)
+        self.assertContains(response, imprt.csv_sep)
+
     def test_archive_import(self):
         imprt = self.create_import()
         with open(imprt.imported_file.path, "r") as f: