1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
from io import StringIO
from django.core.management import call_command
from django.test import TestCase
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())
|
