Added loop automaton test.

nazavode · Oct 12, 2015 · 44a351d · 44a351d
1 parent 713fcb5
commit 44a351d
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/tests/test_automaton.py b/tests/test_automaton.py
@@ -301,3 +301,29 @@ class Sink(Automaton):
     assert auto.state == "sink1"
     auto.event3()
     assert auto.state == "sink2"
+
+
+def test_sink_state_loop():
+    class Sink(Automaton):
+        event1 = Event("state_a", "state_b")
+        event2 = Event(("state_a", "state_b", "state_c", "state_d"), "sink1")
+        event3 = Event(("state_a", "state_b", "state_c", "state_d", "sink1"), "sink2")
+        event4 = Event("sink2", "state_a")
+
+    auto = Sink(initial_state="state_a")
+    auto.event1()
+    assert auto.state == "state_b"
+    auto.event2()
+    assert auto.state == "sink1"
+    auto.event3()
+    assert auto.state == "sink2"
+    auto.event4()
+    assert auto.state == "state_a"
+    auto.event1()
+    assert auto.state == "state_b"
+    auto.event2()
+    assert auto.state == "sink1"
+    auto.event3()
+    assert auto.state == "sink2"
+    auto.event4()
+    assert auto.state == "state_a"