fixup! update function to return for the correct number of stages

opentech/apply/funds/tests/test_admin_form.py

@@ -66,7 +66,7 @@ class TestWorkflowFormAdminForm(TestCase):
         self.assertTrue(form.is_valid(), form.errors.as_text())
 
     def test_doesnt_validates_with_two_forms_one_stage(self):
-        form = self.submit_data(form_data(2, stages=1))
+        form = self.submit_data(form_data(2))
         self.assertFalse(form.is_valid())
         self.assertTrue(form.errors['__all__'])
         formset_errors = form.formsets['forms'].errors
@@ -76,5 +76,5 @@ class TestWorkflowFormAdminForm(TestCase):
         self.assertTrue(formset_errors[1]['form'])
 
     def test_can_save_two_forms(self):
-        form = self.submit_data(form_data(2))
+        form = self.submit_data(form_data(2, stages=2))
         self.assertTrue(form.is_valid())

opentech/apply/funds/tests/test_admin_views.py

@@ -17,7 +17,7 @@ class TestFundCreationView(TestCase):
         self.client.force_login(self.user)
         url = reverse('wagtailadmin_pages:add', args=('funds', 'fundtype', self.home.id))
 
-        data = form_data(forms, same_forms=same_forms)
+        data = form_data(forms, same_forms=same_forms, stages=forms)
         data['action-publish'] = True
 
         response = self.client.post(url, data=data, secure=True, follow=True)