def test_generators(self): def handler(cmd): yield 1 yield 2 yield 3 invoker = HandlerInvoker(FakeCommand, handler) rv = invoker.execute(FakeCommand()) assert rv == [1, 2, 3]
def test_can_invoke_function(self): called = [] def handler(cmd): called.append(cmd) handler = HandlerInvoker(FakeCommand, handler) cmd = FakeCommand() handler.execute(cmd) assert len(called) > 0