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)