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