def test_execute_with_action(self):
context = Context('name', initial_user_context={'foo': 'bar'})
self.assertEqual(None, context.current_state)
action = mock.Mock()
t = Transition('name', 'target', action=action)
t.execute(context, 'obj')
self.assertEqual('target', context.current_state)
action.execute.assert_called_with({'foo': 'bar'}, 'obj')
def test_execute_no_action(self):
context = Context('name')
self.assertEqual(None, context.current_state)
t = Transition('name', 'target')
t.execute(context, None)
self.assertEqual('target', context.current_state)