From 3929ad5280a3f14be6a2f2e9d49fa397509a1dd1 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Sun, 2 Jul 2017 12:05:25 +0200
Subject: Tests: improve wizard test management

---
 archaeological_context_records/tests.py | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

(limited to 'archaeological_context_records/tests.py')

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 07360f73a..6b15cf080 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -541,26 +541,26 @@ class ContextRecordWizardCreationTest(WizardTest, ContextRecordInit, TestCase):
         FormData(
             "Create a simple context record",
             form_datas={
-                'selec-record_creation': {},
-                'general-record_creation': {
+                'selec': {},
+                'general': {
                     'label': "First"
                 },
+                'relations': []
             },
-            ignored=('datings-record_creation',
-                     'interpretation-record_creation',
-                     'relations-record_creation',)
+            ignored=('datings',
+                     'interpretation',
+                     )
         ),
         FormData(
             "Create a context record with a relation",
             form_datas={
-                'selec-record_creation': {},
-                'general-record_creation': {
+                'selec': {},
+                'general': {
                     'label': "Second"
                 },
-                'relations-record_creation': []
+                'relations': []
             },
-            ignored=('datings-record_creation',
-                     'interpretation-record_creation',)
+            ignored=('datings', 'interpretation',)
         ),
     ]
 
@@ -573,13 +573,13 @@ class ContextRecordWizardCreationTest(WizardTest, ContextRecordInit, TestCase):
         ope = self.get_default_operation()
         parcel = self.get_default_parcel()
         for form_data in self.form_datas:
-            form_data.form_datas['selec-record_creation']['operation_id'] = \
-                ope.pk
-            form_data.form_datas['general-record_creation']['parcel'] = \
-                parcel.pk
+            form_data.set('selec', 'operation_id', ope.pk)
+            form_data.set('general', 'parcel', parcel.pk)
 
         self.related_cr = self.create_context_record(data={'operation': ope})[0]
-        self.form_datas[1].form_datas['relations-record_creation'].append(
+
+        self.form_datas[1].append(
+            'relations',
             {'right_record': self.related_cr.pk,
              'relation_type': models.RelationType.objects.create(
                  label="Test").pk}
-- 
cgit v1.2.3