예제 #1
0
 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
예제 #2
0
 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
예제 #3
0
 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()