from pathlib import Path


ROOT = Path(__file__).resolve().parents[2]


def test_demo_workflow_contract_includes_raster_fixture_dataset() -> None:
    schema = (ROOT / "backend" / "app" / "schemas" / "demo.py").read_text(encoding="utf-8")
    service = (ROOT / "backend" / "app" / "services" / "demo_workflow_service.py").read_text(encoding="utf-8")

    assert "raster_dataset_id: UUID | None = None" in schema
    assert 'RASTER_FILENAME = "demo_context_raster.tif"' in service
    assert "_create_demo_raster_bytes" in service
    assert "_create_raster_dataset" in service
    assert "dataset_type=\"raster\"" in service
    assert "source_name=\"fixture\"" in service
    assert "raster_dataset_id=raster.id" in service


def test_demo_workflow_frontend_uses_raster_fixture_for_ai_labs() -> None:
    types = (ROOT / "frontend" / "src" / "types.ts").read_text(encoding="utf-8")
    app = (ROOT / "frontend" / "src" / "App.tsx").read_text(encoding="utf-8")
    hook = (ROOT / "frontend" / "src" / "hooks" / "useDemoWorkflow.ts").read_text(encoding="utf-8")

    assert "raster_dataset_id?: string | null" in types
    assert "setSelectedDetectionDatasetId" in app
    assert "setSelectedSegmentationDatasetId" in app
    assert "setSelectedDetectionDatasetId(result.raster_dataset_id ?? '')" in hook
    assert "setSelectedSegmentationDatasetId(result.raster_dataset_id ?? '')" in hook
    assert "rasterDataset" in hook