From a49f75003e664d622ccc0d11707e5b28275198e7 Mon Sep 17 00:00:00 2001
From: Todd Dembrey <todd.dembrey@torchbox.com>
Date: Thu, 14 Dec 2017 09:27:10 +0000
Subject: [PATCH] Add basic phase model
---
opentech/apply/tests/test_workflow.py | 9 ++++++++-
opentech/apply/workflow.py | 5 +++++
2 files changed, 13 insertions(+), 1 deletion(-)
diff --git a/opentech/apply/tests/test_workflow.py b/opentech/apply/tests/test_workflow.py
index 1fda36a31..45c152052 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 60d4fedf4..521f1b5f2 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
--
GitLab