def setUp(self):
self.am = Mock(ArbiterManager, name='arbiter_manager')
self.zka = Mock(ZookeeperAdaptor, name='zookeeper_adaptor')
self.zka.get_appliction_by_name = Mock(return_value=('COMMAND', 'DATA'))
self.zka.get_applications = Mock(return_value=['APP1'])
self.queue = Mock(Queue, name='queue')
self.queue.get = Mock(side_effect=self.queue_contents + [(STOP_COMMAND, None)])
self.fairground = Fairground(self.am, self.zka, self.queue)
self.fairground.main()
class DescribeFairground(TestCase):
queue_contents = []
def setUp(self):
self.am = Mock(ArbiterManager, name='arbiter_manager')
self.zka = Mock(ZookeeperAdaptor, name='zookeeper_adaptor')
self.zka.get_appliction_by_name = Mock(return_value=('COMMAND', 'DATA'))
self.zka.get_applications = Mock(return_value=['APP1'])
self.queue = Mock(Queue, name='queue')
self.queue.get = Mock(side_effect=self.queue_contents + [(STOP_COMMAND, None)])
self.fairground = Fairground(self.am, self.zka, self.queue)
self.fairground.main()
def should_stop_arbiter_manager(self):
self.am.stop.assert_called_once_with()
def should_stop_zookeeper_adaptor(self):
self.zka.stop.assert_called_once_with()
def should_add_command_to_check_existing_applications(self):
self.queue.put.assert_any_call((CHECK_EXISTING_APPLICATIONS, None))
