def test_add_idle_slave_should_not_add_slave_to_queue_if_slave_is_shutdown(self):
mock_slave = Slave('', 10)
mock_slave.kill = Mock(return_value=None)
mock_slave.set_shutdown_mode()
slave_allocator = self._create_slave_allocator()
slave_allocator._idle_slaves.put = Mock()
slave_allocator.add_idle_slave(mock_slave)
self.assertFalse(slave_allocator._idle_slaves.put.called)
def test_add_idle_slave_should_not_add_slave_to_queue_if_slave_is_shutdown(
self):
mock_slave = Slave('', 10)
mock_slave.kill = Mock(return_value=None)
mock_slave.set_shutdown_mode()
slave_allocator = self._create_slave_allocator()
slave_allocator._idle_slaves.put = Mock()
slave_allocator.add_idle_slave(mock_slave)
self.assertFalse(slave_allocator._idle_slaves.put.called)
