Test some more properties of process_transitions

thread · Jan 12, 2018 · 7a6ac4b · 7a6ac4b
1 parent cc6bd78
commit 7a6ac4b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/routemaster/state_machine/tests/test_transitions.py b/routemaster/state_machine/tests/test_transitions.py
@@ -1,11 +1,26 @@
 import mock
 
+from routemaster.state_machine.exceptions import DeletedLabel
 from routemaster.state_machine.transitions import process_transitions
 
 
 def test_cannot_infinite_loop(app_config, create_label, set_metadata):
     label = create_label('foo', 'test_infinite_machine', {})
     set_metadata(label, {'should_progress': True})
+
     with mock.patch('routemaster.state_machine.transitions.logger') as logger:
         process_transitions(app_config, label)
         logger.warn.assert_called_once()
+
+
+def test_stops_on_delete(app_config, create_label, set_metadata):
+    label = create_label('foo', 'test_infinite_machine', {})
+    set_metadata(label, {'should_progress': True})
+
+    with mock.patch(
+        'routemaster.state_machine.transitions.process_gate',
+        side_effect=[mock.DEFAULT, mock.DEFAULT, DeletedLabel],
+    ) as mock_process_gate:
+        process_transitions(app_config, label)
+
+    assert mock_process_gate.call_count == 3