87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from gyxx_flow.catalog import CatalogError, WorkflowCatalog
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def test_catalog_maps_exactly_21_current_scheduled_tasks() -> None:
|
|
catalog = WorkflowCatalog.load(PROJECT_ROOT / "config")
|
|
scheduled = catalog.scheduled_workflows()
|
|
|
|
assert len(scheduled) == 21
|
|
assert len({item.source_task_name for item in scheduled}) == 21
|
|
assert Counter(item.module for item in scheduled) == {
|
|
"content_marketing": 9,
|
|
"product_commerce": 7,
|
|
"shop_intelligence": 2,
|
|
"supply_chain": 3,
|
|
}
|
|
|
|
|
|
def test_catalog_keeps_known_manual_or_unregistered_workflows_unscheduled() -> None:
|
|
catalog = WorkflowCatalog.load(PROJECT_ROOT / "config")
|
|
|
|
assert {
|
|
"content.mapping.refresh",
|
|
"content.retry_failed",
|
|
"product.backfill",
|
|
"product.market_rank",
|
|
"product.review_collection",
|
|
"supply.purchase_order_update",
|
|
}.issubset({item.workflow_id for item in catalog.manual_workflows()})
|
|
assert "product.weekly_aggregate.documented_missing" in {
|
|
item.workflow_id for item in catalog.unavailable_workflows()
|
|
}
|
|
|
|
|
|
def test_every_scheduled_workflow_has_one_valid_asia_shanghai_schedule() -> None:
|
|
catalog = WorkflowCatalog.load(PROJECT_ROOT / "config")
|
|
|
|
assert catalog.timezone == "Asia/Shanghai"
|
|
assert len(catalog.schedules) == 21
|
|
for workflow in catalog.scheduled_workflows():
|
|
schedule = catalog.schedule_for(workflow.workflow_id)
|
|
assert schedule.workflow_id == workflow.workflow_id
|
|
assert schedule.kind in {"daily", "weekly", "monthly", "interval_days"}
|
|
assert schedule.at.count(":") == 1
|
|
|
|
|
|
def test_catalog_rejects_duplicate_workflow_ids(tmp_path: Path) -> None:
|
|
(tmp_path / "workflows.json").write_text(
|
|
'{"schema_version":2,"workflows":['
|
|
'{"id":"a.one","module":"content_marketing","trigger":"manual",'
|
|
'"execution":{"entry":"run.py"}},'
|
|
'{"id":"a.one","module":"content_marketing","trigger":"manual",'
|
|
'"execution":{"entry":"other.py"}}]}',
|
|
encoding="utf-8",
|
|
)
|
|
(tmp_path / "schedules.json").write_text(
|
|
'{"schema_version":1,"timezone":"Asia/Shanghai","schedules":[]}',
|
|
encoding="utf-8",
|
|
)
|
|
|
|
with pytest.raises(CatalogError, match="duplicate workflow id"):
|
|
WorkflowCatalog.load(tmp_path)
|
|
|
|
|
|
def test_catalog_rejects_absolute_execution_entry(tmp_path: Path) -> None:
|
|
(tmp_path / "workflows.json").write_text(
|
|
'{"schema_version":2,"workflows":['
|
|
'{"id":"a.one","module":"content_marketing","trigger":"manual",'
|
|
'"execution":{"entry":"D:/old/run.py"}}]}',
|
|
encoding="utf-8",
|
|
)
|
|
(tmp_path / "schedules.json").write_text(
|
|
'{"schema_version":1,"timezone":"Asia/Shanghai","schedules":[]}',
|
|
encoding="utf-8",
|
|
)
|
|
|
|
with pytest.raises(CatalogError, match="relative"):
|
|
WorkflowCatalog.load(tmp_path)
|