class test_AppLoader(unittest.TestCase):
def setUp(self):
self.app = app_or_default()
self.loader = AppLoader(app=self.app)
def test_config_from_envvar(self, key="CELERY_HARNESS_CFG1"):
self.assertFalse(self.loader.config_from_envvar("HDSAJIHWIQHEWQU",
silent=True))
self.assertRaises(ImproperlyConfigured,
self.loader.config_from_envvar, "HDSAJIHWIQHEWQU",
silent=False)
os.environ[key] = __name__ + ".object_config"
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf["FOO"], 1)
self.assertEqual(self.loader.conf["BAR"], 2)
os.environ[key] = "unknown_asdwqe.asdwqewqe"
self.assertRaises(ImportError,
self.loader.config_from_envvar, key, silent=False)
self.assertFalse(self.loader.config_from_envvar(key, silent=True))
os.environ[key] = __name__ + ".dict_config"
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf["FOO"], 10)
self.assertEqual(self.loader.conf["BAR"], 20)
def test_on_worker_init(self):
self.loader.conf["CELERY_IMPORTS"] = ("subprocess", )
sys.modules.pop("subprocess", None)
self.loader.on_worker_init()
self.assertIn("subprocess", sys.modules)
class test_AppLoader(Case):
def setUp(self):
self.app = app_or_default()
self.loader = AppLoader(app=self.app)
def test_config_from_envvar(self, key='CELERY_HARNESS_CFG1'):
self.assertFalse(self.loader.config_from_envvar('HDSAJIHWIQHEWQU',
silent=True))
with self.assertRaises(ImproperlyConfigured):
self.loader.config_from_envvar('HDSAJIHWIQHEWQU', silent=False)
os.environ[key] = __name__ + '.object_config'
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf['FOO'], 1)
self.assertEqual(self.loader.conf['BAR'], 2)
os.environ[key] = 'unknown_asdwqe.asdwqewqe'
with self.assertRaises(ImportError):
self.loader.config_from_envvar(key, silent=False)
self.assertFalse(self.loader.config_from_envvar(key, silent=True))
os.environ[key] = __name__ + '.dict_config'
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf['FOO'], 10)
self.assertEqual(self.loader.conf['BAR'], 20)
def test_on_worker_init(self):
prev, self.app.conf.CELERY_IMPORTS = \
self.app.conf.CELERY_IMPORTS, ('subprocess', )
try:
sys.modules.pop('subprocess', None)
self.loader.init_worker()
self.assertIn('subprocess', sys.modules)
finally:
self.app.conf.CELERY_IMPORTS = prev
class test_AppLoader(Case):
def setUp(self):
self.app = app_or_default()
self.loader = AppLoader(app=self.app)
def test_config_from_envvar(self, key='CELERY_HARNESS_CFG1'):
self.assertFalse(
self.loader.config_from_envvar('HDSAJIHWIQHEWQU', silent=True))
with self.assertRaises(ImproperlyConfigured):
self.loader.config_from_envvar('HDSAJIHWIQHEWQU', silent=False)
os.environ[key] = __name__ + '.object_config'
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf['FOO'], 1)
self.assertEqual(self.loader.conf['BAR'], 2)
os.environ[key] = 'unknown_asdwqe.asdwqewqe'
with self.assertRaises(ImportError):
self.loader.config_from_envvar(key, silent=False)
self.assertFalse(self.loader.config_from_envvar(key, silent=True))
os.environ[key] = __name__ + '.dict_config'
self.assertTrue(self.loader.config_from_envvar(key))
self.assertEqual(self.loader.conf['FOO'], 10)
self.assertEqual(self.loader.conf['BAR'], 20)
def test_on_worker_init(self):
prev, self.app.conf.CELERY_IMPORTS = (self.app.conf.CELERY_IMPORTS,
('subprocess', ))
try:
sys.modules.pop('subprocess', None)
self.loader.init_worker()
self.assertIn('subprocess', sys.modules)
finally:
self.app.conf.CELERY_IMPORTS = prev
