diff --git a/opentech/apply/tests/test_workflow.py b/opentech/apply/tests/test_workflow.py index 1fda36a3184b2665b960ef115fafc2bfe66409eb..45c152052f660579455176d4becfcba9dd97691c 100644 --- a/opentech/apply/tests/test_workflow.py +++ b/opentech/apply/tests/test_workflow.py @@ -1,7 +1,7 @@ from django.test import SimpleTestCase from django.forms import Form -from opentech.apply.workflow import Stage, Workflow +from opentech.apply.workflow import Phase, Stage, Workflow from .factories import StageFactory, WorkflowFactory @@ -40,3 +40,10 @@ class TestStageCreation(SimpleTestCase): stage = Stage(name, form) self.assertEqual(stage.name, name) self.assertEqual(stage.form, form) + + +class TestPhaseCreation(SimpleTestCase): + def test_can_create_phase(self): + name = 'the_phase' + phase = Phase(name) + self.assertEqual(phase.name, name) diff --git a/opentech/apply/workflow.py b/opentech/apply/workflow.py index 60d4fedf4e42b435af1a3997a3f79c3643f22b8a..521f1b5f2a1afccc01f116c71b50351ddb2106a6 100644 --- a/opentech/apply/workflow.py +++ b/opentech/apply/workflow.py @@ -29,3 +29,8 @@ class Stage: def __init__(self, name: str, form: Form) -> None: self.name = name self.form = form + + +class Phase: + def __init__(self, name: str) -> None: + self.name = name