Tests: imports

3 files changed, 73 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())
diff --git a/commorganization/tests/autres.csv b/commorganization/tests/autres.csv
new file mode 100644
index 0000000..e28d938
--- /dev/null
+++ b/commorganization/tests/autres.csv
@@ -0,0 +1,15 @@
+Secteur,Type,Nom,URL,Localisation siège (département),Localisation siège (code postal),Localisation siège (commune)
+Aménagement ,Agence d'urbanisme,Agence d'urbanisme et de développement intercommunal de l'agglomération rennaise ,https://www.audiar.org,35000 RENNES,,
+Aménagement ,Agence d'urbanisme,"CÔTES D'ARMOR DÉVELOPPEMENT, AGENCE DE DÉVELOPPEMENT TERRITORIAL & TOURISTIQUE DES CÔTES D’ARMOR",www.cad22.com,22000 Saint-Brieuc,,
+Environnement,Agences Locales de l’Energie et du Climat (ALEC) ,Agences Locales de l’Energie et du Climat (ALEC) de Bretagne Sud,http://www.aloen.fr,56000 VANNES,,
+Environnement,Agences Locales de l’Energie et du Climat (ALEC) ,Agences Locales de l’Energie et du Climat (ALEC) du Centre Ouest Bretagne,http://www.centre-ouest-bretagne.org/Agir-pour-le-developpement-durable/Alecob-maitrisons-notre-energie,29270 Carhaix,,
+Environnement,Agences Locales de l’Energie et du Climat (ALEC) ,Agences Locales de l’Energie et du Climat (ALEC) du Pays de Brest,http://www.energence.net,29200 BREST,,
+Environnement,Agences Locales de l’Energie et du Climat (ALEC) ,Agences Locales de l’Energie et du Climat (ALEC) du Pays de Morlaix,http://www.heol-energies.org,29600 MORLAIX,,
+Tourisme,Association de promotion du tourisme,Canaux de Bretagne ,http://www.canauxdebretagne.org/,56300 PONTIVY,,
+Aménagement,Association développement local,BRUDED,https://www.bruded.fr,35630 Langouët,,
+Aménagement,Associations de Promotion de d’Habitat Participatif (APHP),Habitat Participatif Ouest,http://www.habitatparticipatif-ouest.net,35000 RENNES,,
+Aménagement,Associations de Promotion de d’Habitat Participatif (APHP),réseau d‘entraide pour votre projet d’habitat participatif en Ille-et-vilaine,http://www.parasol35.org,35000 RENNES,,
+État,Institution nationale,"Economie, Finances, Industrie",www.finances.gouv.fr,75000 PARIS,,
+État,Institution nationale,"Fonction Publique, Réforme de l'Etat, Décentralisation",www.fonction-publique.gouv.fr,75000 PARIS,,
+État,Institution nationale,Justice,www.justice.gouv.fr,75000 PARIS,,
+État,Institution nationale,"Economie, Finances, Industrie",www.minefi.gouv.fr,75000 PARIS,,
diff --git a/commorganization/tests/communes.csv b/commorganization/tests/communes.csv
new file mode 100644
index 0000000..468c807
--- /dev/null
+++ b/commorganization/tests/communes.csv
@@ -0,0 +1,8 @@
+DÉPARTEMENT,NOM EPCI OU MAIRIE,CODE POSTAL,COMMUNE,C.C. / C.A. / C.U.,SITE INTERNET,POPULATION COMMUNAUTAIRE,Type de collectivité,Code INSEE
+CÔTES D'ARMOR,Saint-Brieuc Agglomération Baie d'Armor (SBABA) ,22000,SAINT-BRIEUC,Communauté d'Agglomération,www.saintbrieuc-agglo.fr,119 804,EPCI,NA
+CÔTES D'ARMOR,Lannion-Trégor Communauté,22307,LANNION Cedex,Communauté d'Agglomération,www.lannion-tregor.com,78 990,EPCI,NA
+CÔTES D'ARMOR,Communauté de Communes de la Presqu'île de Lézardrieux,22740,PLEUDANIEL,Communauté de Communes,www.cclezardrieux.fr,8 318,EPCI,NA
+FINISTÈRE,Communauté d'Agglomération de Quimper Communauté,29107,QUIMPER Cedex,Communauté d'Agglomération,www.quimper-communaute.fr,91 357,EPCI,NA
+ILLE-ET-VILAINE,Communauté d'Agglomération Rennes Métropole,35207,RENNES Cedex,Communauté d'Agglomération,http://metropole.rennes.fr/,432 841,EPCI,NA
+,Mairie d’ALLINEUC,22460,ALLINEUC,,0,567,Commune,22001
+,Mairie d’ANDEL,22400,ANDEL,,www.andel22.com,1 121,Commune,22002