1 files changed, 50 insertions, 1 deletions
diff --git a/commorganization/tests.py b/commorganization/tests.py
index 7ce503c..84c2e75 100644
--- a/commorganization/tests.py
+++ b/commorganization/tests.py
@@ -1,3 +1,52 @@
+from io import StringIO
+
+from django.core.management import call_command
 from django.test import TestCase
 
-# Create your tests here.
+from . import models
+
+
+class CommandsTestCase(TestCase):
+    def test_import_commune(self):
+        out = StringIO()
+        nb_area_type = models.AreaType.objects.count()
+        nb_area = models.Area.objects.count()
+        nb_org_type = models.OrganizationType.objects.count()
+        nb_org = models.Organization.objects.count()
+        nb_target = models.Target.objects.count()
+        call_command('import_csv_communes',
+                     'commorganization/tests/communes.csv', '--quiet',
+                     stdout=out)
+        nb_area_type_expected = nb_area_type + 4
+        nb_area_expected = nb_area + 10
+        nb_org_type_expected = nb_org_type + 4
+        nb_org_expected = nb_org + 7
+        nb_target_expected = nb_target + 7
+        self.assertEqual(nb_area_type_expected, models.AreaType.objects.count())
+        self.assertEqual(nb_area_expected, models.Area.objects.count())
+        self.assertEqual(nb_org_type_expected,
+                         models.OrganizationType.objects.count())
+        self.assertEqual(nb_org_expected, models.Organization.objects.count())
+        self.assertEqual(nb_target_expected, models.Target.objects.count())
+
+    def test_import_autre(self):
+        out = StringIO()
+        nb_area_type = models.AreaType.objects.count()
+        nb_area = models.Area.objects.count()
+        nb_org_type = models.OrganizationType.objects.count()
+        nb_org = models.Organization.objects.count()
+        nb_target = models.Target.objects.count()
+        call_command('import_csv_autres',
+                     'commorganization/tests/autres.csv', '--quiet',
+                     stdout=out)
+        nb_area_type_expected = nb_area_type + 0
+        nb_area_expected = nb_area + 0
+        nb_org_type_expected = nb_org_type + 10
+        nb_org_expected = nb_org + 13
+        nb_target_expected = nb_target + 14
+        self.assertEqual(nb_area_type_expected, models.AreaType.objects.count())
+        self.assertEqual(nb_area_expected, models.Area.objects.count())
+        self.assertEqual(nb_org_type_expected,
+                         models.OrganizationType.objects.count())
+        self.assertEqual(nb_org_expected, models.Organization.objects.count())
+        self.assertEqual(nb_target_expected, models.Target.objects.count())