From 6eaff8dd819168d8406379c344f793b779963d92 Mon Sep 17 00:00:00 2001
From: Todd Dembrey <todd.dembrey@torchbox.com>
Date: Tue, 12 Dec 2017 10:29:04 +0000
Subject: [PATCH] Add stage basic tests and model
---
opentech/apply/tests.py | 13 ++++++++++---
opentech/apply/workflow.py | 5 +++++
2 files changed, 15 insertions(+), 3 deletions(-)
diff --git a/opentech/apply/tests.py b/opentech/apply/tests.py
index 00d7eef50..1d23399c6 100644
--- a/opentech/apply/tests.py
+++ b/opentech/apply/tests.py
@@ -1,10 +1,17 @@
-from unittest import TestCase
+from django.test import SimpleTestCase
-from .workflow import Workflow
+from .workflow import Workflow, Stage
-class TestWorkflowCreation(TestCase):
+class TestWorkflowCreation(SimpleTestCase):
def test_can_create_workflow(self):
name = 'single_stage'
workflow = Workflow(name)
self.assertEqual(workflow.name, name)
+
+
+class TestStageCreation(SimpleTestCase):
+ def test_can_create_stage(self):
+ name = 'the_stage'
+ stage = Stage(name)
+ self.assertEqual(stage.name, name)
diff --git a/opentech/apply/workflow.py b/opentech/apply/workflow.py
index 95b4dbeb3..069be6123 100644
--- a/opentech/apply/workflow.py
+++ b/opentech/apply/workflow.py
@@ -1,3 +1,8 @@
class Workflow:
def __init__(self, name):
self.name = name
+
+
+class Stage:
+ def __init__(self, name):
+ self.name = name
--
GitLab