Exemplo n.º 1
0
    def test_with_other_functions(self):
        def myfunc1():
            return myfunc2

        def myfunc2():
            return myfunc3

        def myfunc3():
            return 1

        self.assertEqual(func.trampoline(myfunc1), 1)
Exemplo n.º 2
0
do_start = State(name='start',
                 function=setup,
                 on_success='do_first_step',
                 on_failure='do_stop')
do_first_step = State(name='first_step',
                      function=first_step,
                      on_success='do_second_step',
                      on_failure='do_stop')
do_second_step = State(name='second_step',
                       function=second_step,
                       on_success='do_wait',
                       on_failure='do_stop')
do_wait = State(name='wait',
                function=wait,
                on_success='do_first_step',
                on_failure='do_stop')
do_stop = State(name='stop', function=stop, on_success=None, on_failure=None)


def state_machine(state):
    if state.function():
        new_state = globals().get(state.on_success, None)
    else:
        new_state = globals().get(state.on_failure, None)
    if new_state:
        return partial(state_machine, new_state)


if __name__ == '__main__':
    trampoline(state_machine, do_start)
Exemplo n.º 3
0
 def test_with_args(self):
     def myfunc(a):
         return a
     self.assertEqual(func.trampoline(myfunc, 1), 1)
Exemplo n.º 4
0
 def test_no_func(self):
     # Test that if no function is given, trampoline is basically a glorified
     # apply.
     def myfunc():
         return 1
     self.assertEqual(func.trampoline(myfunc), 1)
Exemplo n.º 5
0
    print 'First step...'
    return True
def second_step():
    print 'Second step...'
    return True
def wait():
    print 'Waiting...'
    return True
def stop():
    print 'Stopping...'
    return True

State = collections.namedtuple('State', 'name function on_success on_failure')

do_start = State(name='start', function=setup, on_success='do_first_step', on_failure='do_stop')
do_first_step = State(name='first_step', function=first_step, on_success='do_second_step', on_failure='do_stop')
do_second_step = State(name='second_step', function=second_step, on_success='do_wait', on_failure='do_stop')
do_wait = State(name='wait', function=wait, on_success='do_first_step', on_failure='do_stop')
do_stop = State(name='stop', function=stop, on_success=None, on_failure=None)

def state_machine(state):
    if state.function():
        new_state = globals().get(state.on_success, None)
    else:
        new_state = globals().get(state.on_failure, None)
    if new_state:
        return partial(state_machine, new_state)

if __name__ == '__main__':        
    trampoline(state_machine, do_start)