def test_load(self): with patch('pkg_resources.iter_entry_points') as iterep: with patch('celery.bin.basic.symbol_by_name') as symbyname: ep = Mock() ep.name = 'ep' ep.module_name = 'foo' ep.attrs = ['bar', 'baz'] iterep.return_value = [ep] cls = symbyname.return_value = Mock() register = Mock() e = Extensions('unit', register) e.load() symbyname.assert_called_with('foo:bar') register.assert_called_with(cls, name='ep') with patch('celery.bin.basic.symbol_by_name') as symbyname: symbyname.side_effect = SyntaxError() with patch('warnings.warn') as warn: e.load() self.assertTrue(warn.called) with patch('celery.bin.basic.symbol_by_name') as symbyname: symbyname.side_effect = KeyError('foo') with self.assertRaises(KeyError): e.load()
def test_load(self): with patch('pkg_resources.iter_entry_points') as iterep: with patch('celery.bin.base.symbol_by_name') as symbyname: ep = Mock() ep.name = 'ep' ep.module_name = 'foo' ep.attrs = ['bar', 'baz'] iterep.return_value = [ep] cls = symbyname.return_value = Mock() register = Mock() e = Extensions('unit', register) e.load() symbyname.assert_called_with('foo:bar') register.assert_called_with(cls, name='ep') with patch('celery.bin.base.symbol_by_name') as symbyname: symbyname.side_effect = SyntaxError() with patch('warnings.warn') as warn: e.load() self.assertTrue(warn.called) with patch('celery.bin.base.symbol_by_name') as symbyname: symbyname.side_effect = KeyError('foo') with self.assertRaises(KeyError): e.load()
def test_load(self): with patch("pkg_resources.iter_entry_points") as iterep: with patch("celery.bin.base.symbol_by_name") as symbyname: ep = Mock() ep.name = "ep" ep.module_name = "foo" ep.attrs = ["bar", "baz"] iterep.return_value = [ep] cls = symbyname.return_value = Mock() register = Mock() e = Extensions("unit", register) e.load() symbyname.assert_called_with("foo:bar") register.assert_called_with(cls, name="ep") with patch("celery.bin.base.symbol_by_name") as symbyname: symbyname.side_effect = SyntaxError() with patch("warnings.warn") as warn: e.load() self.assertTrue(warn.called) with patch("celery.bin.base.symbol_by_name") as symbyname: symbyname.side_effect = KeyError("foo") with self.assertRaises(KeyError): e.load()
def load_extension_commands(self): names = Extensions(self.ext_fmt.format(self=self), self.register_command).load() if names: command_classes.append(('Extensions', names, 'magenta'))