def test_run(self): job = Job(random_job, 'hi', arg2='there') job.run() self.assertTrue(job.run_time >= 1) self.assertEqual(job.result, 'yo!') job = Job(bad_job) job.run() self.assertIsInstance(job.exception, ZeroDivisionError)
def test_add_job(self): sid = uuid4().hex queue = Queue(sid) queue.add_job(Job(test_job, 'there')) queue.add_job(Job(test_job, arg1='test', arg2='test')) sleep(10) self.assertEqual(queue.num_jobs(), 2) queue.delete()
def test_remove_job(self): sid = uuid4().hex queue = Queue(sid) queue.add_job(Job(test_job, 'there')) queue.add_job(Job(test_job, arg1='test', arg2='test')) for job in queue.jobs: queue.remove_job(job) sleep(10) self.assertEqual(queue.num_jobs(), 0) queue.delete()
def test_work(self): sid = uuid4().hex queue = Queue(sid) queue.add_job(Job(test_job, 'hi', 'there')) sleep(10) self.assertEqual(queue.num_jobs(), 1) worker = Worker([queue]) worker.work(True) sleep(10) self.assertEqual(queue.num_jobs(), 0) queue.delete()
def run(): try: job_id = str(uuid.uuid4()) uid = str(uuid.uuid4()) print("Creating queue") q = Queue("simpleq_test") job = Job(job_id, None, log_stuff, uid) print(f"job_id: {job_id}") print(f"uid: {uid}") for n in range(10): q.add_job(job, delay=5) sleep(10) print(f"q.num_jobs(): {q.num_jobs()}") finally: q.delete()
def test_create_job(self): job = Job(random_job, 'hi', arg2='there') self.assertIsInstance(job, Job)