def test_proc_title(self):
conf = WorkerConfig(self.mkargs('--name=foo'))
self.assertEqual('test-script [init]', conf.get_proc_title())
conf.prepare_worker()
self.assertEqual('test-script [init] queue=foo', conf.get_proc_title())
conf.worker.status = 'waiting'
self.assertEqual('test-script [waiting] queue=foo loop=0/1000 waiting-jobs=0', conf.get_proc_title())
conf.worker.end_forced = True
self.assertEqual('test-script [waiting - ending] queue=foo loop=0/1000 waiting-jobs=0', conf.get_proc_title())
def test_prepare_worker(self):
conf = WorkerConfig(self.mkargs('--name=foo'))
self.assertIsNone(getattr(conf, 'worker', None))
conf.prepare_worker()
self.assertIsInstance(conf.worker, Worker)
conf = WorkerConfig(self.mkargs('--name=bar --worker-class=tests.workers.WorkerConfigRunTest.WorkerClass'))
conf.prepare_worker()
self.assertIsInstance(conf.worker, WorkerConfigRunTest.WorkerClass)
self.assertFalse(conf.worker.end_forced)
conf = WorkerConfig(self.mkargs('--name=baz --dry-run'))
conf.prepare_worker()
self.assertTrue(conf.worker.end_forced)
