From b3bda3945437c0023206dfb6e233686a36dd2ad1 Mon Sep 17 00:00:00 2001
From: Todd Dembrey <todd.dembrey@torchbox.com>
Date: Tue, 23 Oct 2018 15:41:22 +0100
Subject: [PATCH] fixup! update function to return for the correct number of
stages
---
opentech/apply/funds/tests/test_admin_form.py | 4 ++--
opentech/apply/funds/tests/test_admin_views.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/opentech/apply/funds/tests/test_admin_form.py b/opentech/apply/funds/tests/test_admin_form.py
index cafd805c2..93413519a 100644
--- a/opentech/apply/funds/tests/test_admin_form.py
+++ b/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())
diff --git a/opentech/apply/funds/tests/test_admin_views.py b/opentech/apply/funds/tests/test_admin_views.py
index b065ad10d..3469b0448 100644
--- a/opentech/apply/funds/tests/test_admin_views.py
+++ b/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)
--
GitLab