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'))