Move the factories around

opentech/apply/funds/tests/factories/__init__.py

+from .models import *
+from .blocks import *

opentech/apply/funds/tests/factories/blocks.py

+import wagtail_factories
+
+from opentech.apply.funds.blocks import CustomFormFieldsBlock
+
+
+class CustomFormBlockFactory(wagtail_factories.StructBlockFactory):
+    class Meta:
+        model = CustomFormFieldsBlock

opentech/apply/funds/tests/factories.py → opentech/apply/funds/tests/factories/models.py

@@ -7,6 +7,8 @@ import wagtail_factories
 from opentech.apply.funds.models import ApplicationForm, FundType, FundForm, Round
 from opentech.apply.funds.workflow import Action, Phase, Stage, Workflow
 
+from .blocks import CustomFormBlockFactory
+
 
 class ListSubFactory(factory.SubFactory):
     def __init__(self, *args, count=0, **kwargs):
@@ -137,6 +139,7 @@ class ApplicationFormFactory(factory.DjangoModelFactory):
         model = ApplicationForm
 
     name = factory.Faker('word')
+    form_fields = wagtail_factories.StreamFieldFactory({'field': CustomFormBlockFactory})
 
 
 class RoundFactory(wagtail_factories.PageFactory):

opentech/apply/funds/tests/test_models.py

@@ -124,3 +124,8 @@ class TestRoundModel(TestCase):
 
         with self.assertRaises(ValidationError):
             new_round.clean()
+
+
+class TestFormSubmission(TestCase):
+    def test_can_submit_if_new(self):
+        pass