(self)
| 1486 | |
| 1487 | class TestRunnerClass(BaseTestCase): |
| 1488 | def test_custom_runner_class_single_task(self): |
| 1489 | task = self.tiger.delay(simple_task, runner_class=MyRunnerClass) |
| 1490 | Worker(self.tiger).run(once=True) |
| 1491 | assert self.conn.get("task_id") == task.id |
| 1492 | self.conn.delete("task_id") |
| 1493 | self._ensure_queues() |
| 1494 | |
| 1495 | def test_custom_runner_class_batch_task(self): |
| 1496 | self.tiger.delay(batch_task, args=[1], runner_class=MyRunnerClass) |