Document -> Town/Area: models, admin, forms

1 files changed, 32 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 7328a5f95..4e450c3f8 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -3821,9 +3821,41 @@ class DocumentTest(TestCase):
         self.find2.base_finds.add(bf2)
         self.st1 = models.SourceType.objects.create(label="Report", code="REP")
         self.st2 = models.SourceType.objects.create(label="Illustration", code="ILL")
+        self.town = models.Town.objects.create(name="Daisy town", numero_insee="59134")
 
         self.username, self.password, self.user = create_superuser()
 
+    def test_create_form(self):
+        nb_doc = models.Document.objects.count()
+        c = Client()
+        url = reverse("create-document")
+        response = c.get(url)
+        self.assertEqual(response.status_code, 302)
+
+        c.login(username=self.username, password=self.password)
+        response = c.get(url)
+        self.assertEqual(response.status_code, 200)
+
+        posted = {
+            "authors": [],
+            "title": "A document",
+            "operations": [str(self.ope1.pk)],
+            "towns": [str(self.town.pk)],
+        }
+        response = c.post(url, posted)
+        new_child_document = self.ope1.documents.order_by("-pk").all()[0]
+        self.assertEqual(nb_doc + 1, models.Document.objects.count())
+        self.assertRedirects(
+            response,
+            "/document/edit/?open_item={}".format(
+                new_child_document.pk
+            ),
+        )
+        self.assertIn(
+            self.town.pk,
+            list(new_child_document.towns.values_list("pk", flat=True).all())
+        )
+
     def test_custom_index(self):
         profile, created = models.IshtarSiteProfile.objects.get_or_create(
             slug="default", active=True