Skip to content

Commit

Permalink
Merge c67cb7f into 02076d3
Browse files Browse the repository at this point in the history
  • Loading branch information
facundofc committed Oct 30, 2019
2 parents 02076d3 + c67cb7f commit 8064db4
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 3 deletions.
7 changes: 7 additions & 0 deletions tests/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -774,6 +774,13 @@ def trigger(self, value):
m = Machine(model=model)
self.assertEqual(model.trigger(5), 5)

def raise_key_error():
raise KeyError

self.stuff.machine.add_transition('do_raises_keyerror', '*', 'C', before=raise_key_error)
with self.assertRaises(KeyError):
self.stuff.trigger('do_raises_keyerror')

def test_get_triggers(self):
states = ['A', 'B', 'C']
transitions = [['a2b', 'A', 'B'],
Expand Down
6 changes: 3 additions & 3 deletions transitions/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,10 +62,10 @@ def _get_trigger(model, machine, trigger_name, *args, **kwargs):
bool: True if a transitions has been conducted or the trigger event has been queued.
"""
try:
return machine.events[trigger_name].trigger(model, *args, **kwargs)
event = machine.events[trigger_name]
except KeyError:
pass
raise AttributeError("Do not know event named '%s'." % trigger_name)
raise AttributeError("Do not know event named '%s'." % trigger_name)
return event.trigger(model, *args, **kwargs)


def _prep_ordered_arg(desired_length, arguments=None):
Expand Down

0 comments on commit 8064db4

Please sign in to comment.