16 lines
637 B
Python
16 lines
637 B
Python
import worker
|
|
|
|
|
|
def test_dispatch_routes_to_provider(monkeypatch):
|
|
called = {}
|
|
monkeypatch.setattr(worker, "run_gpt_image_generation", lambda tid, p: called.setdefault("gpt", (tid, p)))
|
|
worker._dispatch({"job_type": "gpt_image_generation", "task_id": "t1", "params": {"prompt": "x"}})
|
|
assert called["gpt"][0] == "t1"
|
|
|
|
|
|
def test_dispatch_unknown_job_type_reports_failed(monkeypatch):
|
|
calls = []
|
|
monkeypatch.setattr(worker, "webhook_update_task", lambda *a, **k: calls.append((a, k)))
|
|
worker._dispatch({"job_type": "midjourney_generation", "task_id": "t9", "params": {}})
|
|
assert calls[-1][0][1] == "failed"
|