From 532231d46a09a933ef93421bcf5e4eee1fe1fe05 Mon Sep 17 00:00:00 2001
From: Todd Dembrey <todd.dembrey@torchbox.com>
Date: Tue, 23 Oct 2018 15:05:39 +0100
Subject: [PATCH] update function to return for the correct number of stages
---
opentech/apply/funds/tests/factories/models.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/opentech/apply/funds/tests/factories/models.py b/opentech/apply/funds/tests/factories/models.py
index 36e233f24..e678dd8d7 100644
--- a/opentech/apply/funds/tests/factories/models.py
+++ b/opentech/apply/funds/tests/factories/models.py
@@ -21,6 +21,8 @@ from opentech.apply.funds.models.forms import (
RoundBaseForm,
RoundBaseReviewForm,
)
+from opentech.apply.funds.workflow import ConceptProposal, Request
+
from opentech.apply.users.tests.factories import StaffFactory, UserFactory
from opentech.apply.stream_forms.testing.factories import FormDataFactory
from opentech.apply.home.factories import ApplyHomePageFactory
@@ -49,7 +51,10 @@ __all__ = [
def workflow_for_stages(stages):
- return list(FundType.WORKFLOW_CHOICES.keys())[stages - 1]
+ return {
+ 1: Request.admin_name,
+ 2: ConceptProposal.admin_name,
+ }[stages]
class AbstractApplicationFactory(wagtail_factories.PageFactory):
--
GitLab