def test_task_locking(self): ret = locked_task(1, 2) task = test_huey.dequeue() self.worker(task) self.assertEqual(ret.get(), 3) ret = locked_task(2, 3) task = test_huey.dequeue() with test_huey.lock_task('test-lock'): self.worker(task) self.assertTrue(ret.get() is None)
def test_task_locking(self): ret = locked_task(1, 2) task = test_huey.dequeue() self.worker(task) self.assertEqual(ret.get(), 3) ret = locked_task(2, 3) task = test_huey.dequeue() with test_huey.lock_task('test-lock'): self.worker(task) self.assertRaises(TaskException, ret.get)