def setUp(self):
self.waiter_config = mock.Mock()
self.waiter_config.description = ''
self.waiter_config.operation = 'MyOperation'
# Set up the acceptors.
self.success_acceptor = mock.Mock()
self.success_acceptor.state = 'success'
self.fail_acceptor = mock.Mock()
self.fail_acceptor.state = 'failure'
self.error_acceptor = mock.Mock()
self.error_acceptor.state = 'error'
self.waiter_config.acceptors = [
self.fail_acceptor, self.success_acceptor, self.error_acceptor
]
self.doc_builder = WaiterStateDocBuilder(self.waiter_config)
