Python CeleryCommand.register_command Examples

Example #1

File: test_celery.py Project: csunny/celery

 def test_register_command(self):
     prev, CeleryCommand.commands = dict(CeleryCommand.commands), {}
     try:
         fun = Mock(name='fun')
         CeleryCommand.register_command(fun, name='foo')
         self.assertIs(CeleryCommand.commands['foo'], fun)
     finally:
         CeleryCommand.commands = prev