def test_prepare_prog_name(self):
x = CeleryCommand(app=self.app)
main = Mock(name='__main__')
main.__file__ = '/opt/foo.py'
with patch.dict(sys.modules, __main__=main):
assert x.prepare_prog_name('__main__.py') == '/opt/foo.py'
assert x.prepare_prog_name('celery') == 'celery'
def test_pool_restart_reload_modules(self):
consumer = Consumer(self.app)
consumer.controller = _WC(app=self.app)
consumer.controller.consumer = consumer
consumer.controller.pool.restart = Mock()
consumer.reset_rate_limits = Mock(name="reset_rate_limits()")
consumer.update_strategies = Mock(name="update_strategies()")
panel = self.create_panel(consumer=consumer)
panel.app = self.app
_import = panel.app.loader.import_from_cwd = Mock()
_reload = Mock()
self.app.conf.worker_pool_restarts = True
with patch.dict(sys.modules, {"foo": None}):
panel.handle("pool_restart", {"modules": ["foo"], "reload": False, "reloader": _reload})
consumer.controller.pool.restart.assert_called()
_reload.assert_not_called()
_import.assert_not_called()
_import.reset_mock()
_reload.reset_mock()
consumer.controller.pool.restart.reset_mock()
panel.handle("pool_restart", {"modules": ["foo"], "reload": True, "reloader": _reload})
consumer.controller.pool.restart.assert_called()
_reload.assert_called()
_import.assert_not_called()
def test_find_module_pep420_namespace_package(tmp_path, monkeypatch):
monkeypatch.chdir(str(tmp_path))
(tmp_path / 'pkg' / 'foo').mkdir(parents=True)
(tmp_path / 'pkg' / 'foo' / '__init__.py').write_text('')
(tmp_path / 'pkg' / 'foo' / 'bar.py').write_text('')
with patch.dict(sys.modules):
for modname in list(sys.modules):
if modname == 'pkg' or modname.startswith('pkg.'):
del sys.modules[modname]
with pytest.raises(ImportError):
find_module('pkg.missing')
with pytest.raises(ImportError):
find_module('pkg.foo.missing')
assert find_module('pkg.foo.bar')
with pytest.raises(NotAPackage) as exc_info:
find_module('pkg.foo.bar.missing')
assert exc_info.value.args[0] == 'pkg.foo.bar'
def test_pool_restart_reload_modules(self):
consumer = Consumer(self.app)
consumer.controller = _WC(app=self.app)
consumer.controller.consumer = consumer
consumer.controller.pool.restart = Mock()
consumer.reset_rate_limits = Mock(name='reset_rate_limits()')
consumer.update_strategies = Mock(name='update_strategies()')
panel = self.create_panel(consumer=consumer)
panel.app = self.app
_import = panel.app.loader.import_from_cwd = Mock()
_reload = Mock()
self.app.conf.worker_pool_restarts = True
with patch.dict(sys.modules, {'foo': None}):
panel.handle('pool_restart', {
'modules': ['foo'],
'reload': False,
'reloader': _reload,
})
consumer.controller.pool.restart.assert_called()
_reload.assert_not_called()
_import.assert_not_called()
_import.reset_mock()
_reload.reset_mock()
consumer.controller.pool.restart.reset_mock()
panel.handle('pool_restart', {
'modules': ['foo'],
'reload': True,
'reloader': _reload,
})
consumer.controller.pool.restart.assert_called()
_reload.assert_called()
_import.assert_not_called()