Allow iterating through the stage

opentech/apply/tests/test_workflow.py

@@ -43,6 +43,11 @@ class TestStageCreation(SimpleTestCase):
         self.assertEqual(stage.form, form)
         self.assertEqual(stage.phases, phases)
 
+    def test_can_iterate_through_phases(self):
+        stage = StageFactory()
+        for phase, check in zip(stage, stage.phases):
+            self.assertEqual(phase, check)
+
 
 class TestPhaseCreation(SimpleTestCase):
     def test_can_create_phase(self):

opentech/apply/workflow.py

@@ -31,6 +31,9 @@ class Stage:
         self.form = form
         self.phases = phases
 
+    def __iter__(self) -> Iterator['Phase']:
+        yield from self.phases
+
 
 class Phase:
     def __init__(self, name: str) -> None: