Feat: move action to worker with a dispatcher

This commit is contained in:
2022-04-10 14:37:19 +02:00
parent 3f1464f3f6
commit 08411bd42d
5 changed files with 36 additions and 9 deletions

View File

@@ -1,10 +1,14 @@
from bopytex.planner import fake_planner
from bopytex.service import orcherstrator
from bopytex.tasks import Task
from bopytex.worker import Dispatcher, fake_worker
def test_service():
options = {"quantity": 3, "template": "tpl_src.tex"}
service = orcherstrator(options, fake_planner.simple)
for i, task in enumerate(service):
assert task == Task("Do", args={"number": i}, deps=[], output=f"{i}")
dispatcher = Dispatcher(actions={"DO": fake_worker})
service = orcherstrator(options, fake_planner.simple, dispatcher)
for i, task_output in enumerate(service):
assert task_output == f"FAKE - {{'number': {i}}} - [] - {i}"