Finds: better test with images

1 files changed, 20 insertions, 4 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index 9102a8c1a..c17e440b1 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -20,7 +20,8 @@
 from django.conf import settings
 from django.core.files.uploadedfile import SimpleUploadedFile
 from django.test import TestCase
-from ishtar_common.models import ImporterType, IshtarUser
+from ishtar_common.models import ImporterType, IshtarUser, ImporterColumn,\
+    FormaterType, ImportTarget
 
 from archaeological_finds import models
 
@@ -44,11 +45,24 @@ class ImportFindTest(ImportContextRecordTest):
         old_nb = models.BaseFind.objects.count()
         old_nb_find = models.Find.objects.count()
         MCC = ImporterType.objects.get(name=u"MCC - Mobilier")
+
+        col = ImporterColumn.objects.create(col_number=25,
+                                            importer_type_id=MCC.pk)
+        formater = FormaterType.objects.filter(
+            formater_type='FileFormater').all()[0]
+        ImportTarget.objects.create(target='find__image',
+                                    formater_type_id=formater.pk,
+                                    column_id=col.pk)
         mcc_file = open(
             settings.ROOT_PATH +
             '../archaeological_finds/tests/MCC-finds-example.csv', 'rb')
+        mcc_images = open(
+            settings.ROOT_PATH +
+            '../archaeological_finds/tests/images.zip', 'rb')
         file_dict = {'imported_file': SimpleUploadedFile(mcc_file.name,
-                                                         mcc_file.read())}
+                                                         mcc_file.read()),
+                     'imported_images': SimpleUploadedFile(mcc_images.name,
+                                                           mcc_images.read())}
         post_dict = {'importer_type': MCC.pk, 'skip_lines': 1,
                      "encoding": 'utf-8'}
         form = forms_common.NewImportForm(data=post_dict, files=file_dict,
@@ -69,13 +83,15 @@ class ImportFindTest(ImportContextRecordTest):
             return
         # new finds has now been imported
         current_nb = models.BaseFind.objects.count()
-        self.assertTrue(current_nb == (old_nb + 4))
+        self.assertEqual(current_nb, (old_nb + 4))
         current_nb = models.Find.objects.count()
-        self.assertTrue(current_nb == (old_nb_find + 4))
+        self.assertEqual(current_nb, (old_nb_find + 4))
         self.assertEqual(
             models.Find.objects.filter(material_types__pk=ceram).count(), 4)
         self.assertEqual(
             models.Find.objects.filter(material_types__pk=glass).count(), 1)
+        images = [f.image for f in models.Find.objects.all() if f.image.name]
+        self.assertEqual(len(images), 1)
 
 
 class FindInit(ContextRecordInit):