Tests: improve wizard test management

1 files changed, 15 insertions, 15 deletions

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}