From c0de6e0608a068084156302c9bdabefbcc482cd8 Mon Sep 17 00:00:00 2001 From: Todd Dembrey <todd.dembrey@torchbox.com> Date: Tue, 23 Oct 2018 15:23:31 +0100 Subject: [PATCH] fixup! update function to return for the correct number of stages --- opentech/apply/funds/tests/test_admin_form.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opentech/apply/funds/tests/test_admin_form.py b/opentech/apply/funds/tests/test_admin_form.py index 8c4336f10..cafd805c2 100644 --- a/opentech/apply/funds/tests/test_admin_form.py +++ b/opentech/apply/funds/tests/test_admin_form.py @@ -34,13 +34,13 @@ def formset_base(field, total, delete, factory, same=False): return base_data -def form_data(number_forms=0, delete=0, stages=None, same_forms=False): +def form_data(number_forms=0, delete=0, stages=1, same_forms=False): form_data = formset_base('forms', number_forms, delete, same=same_forms, factory=ApplicationFormFactory) review_form_data = formset_base('review_forms', number_forms, False, same=same_forms, factory=ReviewFormFactory) form_data.update(review_form_data) fund_data = factory.build(dict, FACTORY_CLASS=FundTypeFactory) - fund_data['workflow_name'] = workflow_for_stages(stages or number_forms) + fund_data['workflow_name'] = workflow_for_stages(stages) form_data.update(fund_data) return form_data -- GitLab