def setUp(self): self.fsm = create_voting_fsm('tracking-id', 'vote', lambda _: 'broadcasted vote', lambda _: 'spawned program', lambda _: 'folded', lambda _: 'cleaned up', )
def setUp(self): self.fsm = create_voting_fsm( 'tracking-id', 'vote', lambda _: 'broadcasted vote', lambda _: 'spawned program', lambda _: 'folded', lambda _: 'cleaned up', )
def test_should_invoke_callbacks_on_lose(self): broadcast_vote = Mock() spawn = Mock() cleanup = Mock() fold = Mock() fsm = create_voting_fsm('tracking-id', 'vote', broadcast_vote, spawn, fold, cleanup) broadcast_vote.assert_called_with(ANY) fsm.fold() fold.assert_called_with(ANY) self.assertEqual(fsm.current, 'finish')