Import: fix unit import on MCC context record

1 files changed, 11 insertions, 8 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 72144b528..2900b8786 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -48,7 +48,8 @@ class ImportContextRecordTest(ImportOperationTest):
         old_nb = models.ContextRecord.objects.count()
         MCC = ImporterType.objects.get(name=u"MCC - UE")
         mcc_file = open(settings.ROOT_PATH + \
-         '../archaeological_context_records/tests/MCC-context-records-example.csv', 'rb')
+                    '../archaeological_context_records/tests/'\
+                    'MCC-context-records-example.csv', 'rb')
         file_dict = {'imported_file': SimpleUploadedFile(mcc_file.name,
                                                          mcc_file.read())}
         post_dict = {'importer_type':MCC.pk, 'skip_lines':1,
@@ -61,14 +62,12 @@ class ImportContextRecordTest(ImportOperationTest):
         impt = form.save(self.ishtar_user)
         impt.initialize()
 
-        # doing manualy connections
+        # doing manual connections
         hc = models.Unit.objects.get(txt_idx='not_in_context').pk
-        #for tg in TargetKey.objects.all():
-        #    print(tg.target, tg.key)
-        tg = TargetKey.objects.get(target__target='unit', key='Hors contexte')
-        tg.value = hc
-        tg.is_set = True
-        tg.save()
+        self.setTargetKey('unit', 'hc', hc)
+        self.setTargetKey('unit', 'hors-contexte', hc)
+        layer = models.Unit.objects.get(txt_idx='layer').pk
+        self.setTargetKey('unit', 'couche', layer)
 
         impt.importation()
         if not test:
@@ -77,6 +76,10 @@ class ImportContextRecordTest(ImportOperationTest):
         # new ues has now been imported
         current_nb = models.ContextRecord.objects.count()
         self.assertTrue(current_nb == (old_nb + 4))
+        self.assertEqual(
+            models.ContextRecord.objects.filter(unit__pk=hc).count(), 3)
+        self.assertEqual(
+            models.ContextRecord.objects.filter(unit__pk=layer).count(), 1)
 
 
 class ContextRecordInit(OperationInitTest):