Add workflow and test we can get the class from the fund

opentech/apply/migrations/0003_add_workflow_to_fund.py

+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.7 on 2017-12-19 12:09
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apply', '0002_fund_workflow'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='fund',
+            name='workflow',
+            field=models.CharField(choices=[('single', 'Single Stage'), ('double', 'Two Stage')], default='single', max_length=100),
+        ),
+    ]

opentech/apply/models.py

@@ -8,6 +8,10 @@ from opentech.utils.models import SocialFields, ListingFields
 
 from .workflow import SingleStage, DoubleStage
 
+WORKFLOW_CLASS = {
+    SingleStage.name: SingleStage,
+    DoubleStage.name: DoubleStage,
+}
 
 class ApplyHomePage(Page, SocialFields, ListingFields):
     # Only allow creating HomePages at the root level
@@ -32,9 +36,13 @@ class ApplyHomePage(Page, SocialFields, ListingFields):
 class FundPage(Page):
     parent_page_types = [ApplyHomePage]
     WORKFLOWS = (
-        ('single', SingleStage),
-        ('double', DoubleStage),
+        ('single', SingleStage.name),
+        ('double', DoubleStage.name),
     )
 
     name = models.CharField(max_length=60)
     workflow = models.CharField(choices=WORKFLOWS, max_length=100, default=WORKFLOWS[0][0])
+
+    @property
+    def workflow_class(self):
+        return WORKFLOW_CLASS[self.get_workflow_display()]

opentech/apply/tests/test_models.py

+from django.test import TestCase
+
+from opentech.apply.models import Fund
+from opentech.apply.workflow import SingleStage
+
+
+class TestFundModel(TestCase):
+    def test_can_access_workflow_class(self):
+        fund = Fund.objects.create(name='Internet Freedom Fund')
+        self.assertEqual(fund.workflow, 'single')
+        self.assertEqual(fund.workflow_class, SingleStage)