Tests: organization from warehouse

1 files changed, 16 insertions, 0 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index bdd48536b..73cc166ac 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -170,6 +170,22 @@ class ContainerWizardCreationTest(WizardTest, FindInit, TestCase):
                          self.localisation_detail_number + 1)
 
 
+class WarehouseTest(TestCase):
+    fixtures = WAREHOUSE_FIXTURES
+
+    def test_orga_from_warehouse(self):
+        w = models.Warehouse.objects.create(
+            name="Hophop", warehouse_type=models.WarehouseType.objects.all()[0],
+            address="Adresse"
+        )
+        w.create_attached_organization()
+        w = models.Warehouse.objects.get(pk=w.pk)
+        self.assertIsNotNone(w.organization)
+        self.assertEqual(w.organization.name, "Hophop")
+        self.assertEqual(w.organization.address, "Adresse")
+        self.assertIsNone(w.address)
+
+
 class ContainerTest(FindInit, TestCase):
     fixtures = WAREHOUSE_FIXTURES