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