def test_load_extensions_commands(self):
with patch('celery.bin.celery.Extensions') as Ext:
prev, mod.command_classes = list(mod.command_classes), Mock()
try:
ext = Ext.return_value = Mock(name='Extension')
ext.load.return_value = ['foo', 'bar']
x = CeleryCommand(app=self.app)
x.load_extension_commands()
mod.command_classes.append.assert_called_with(
('Extensions', ['foo', 'bar'], 'magenta'), )
finally:
mod.command_classes = prev
def test_load_extensions_commands(self):
with patch('celery.bin.celery.Extensions') as Ext:
prev, mod.command_classes = list(mod.command_classes), Mock()
try:
ext = Ext.return_value = Mock(name='Extension')
ext.load.return_value = ['foo', 'bar']
x = CeleryCommand(app=self.app)
x.load_extension_commands()
mod.command_classes.append.assert_called_with(
('Extensions', ['foo', 'bar'], 'magenta'),
)
finally:
mod.command_classes = prev
def test_load_extensions_no_commands(self):
with patch('celery.bin.celery.Extensions') as Ext:
ext = Ext.return_value = Mock(name='Extension')
ext.load.return_value = None
x = CeleryCommand(app=self.app)
x.load_extension_commands()
