from __future__ import annotations

from pathlib import Path
from tests.frontend_contract import read_feature


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


def test_app_uses_detection_and_segmentation_workflow_hooks() -> None:
    app = read_feature("shell")

    assert "useDetectionWorkflow" in app
    assert "useSegmentationWorkflow" in app
    assert "from './hooks/useDetectionWorkflow'" in app
    assert "from './hooks/useSegmentationWorkflow'" in app
    assert "detectionApi" not in app
    assert "segmentationApi" not in app


def test_detection_workflow_hook_owns_detection_api_calls() -> None:
    hook = read_feature("detection")

    assert "detectionApi.listModels" in hook
    assert "detectionApi.listRuns" in hook
    assert "detectionApi.run" in hook
    assert "detectionApi.compareWithReference" in hook
    assert "resetDetectionForProject" in hook


def test_segmentation_workflow_hook_owns_segmentation_api_calls() -> None:
    hook = read_feature("segmentation")

    assert "segmentationApi.listModels" in hook
    assert "segmentationApi.listRuns" in hook
    assert "segmentationApi.run" in hook
    assert "segmentationApi.compareWithReference" in hook
    assert "resetSegmentationForProject" in hook


def test_app_still_wires_detection_and_segmentation_panels() -> None:
    app = read_feature("shell")

    assert "<DetectionLab" in app
    assert "onRunDetection={runDetection}" in app
    assert "onRunQa={runDetectionQa}" in app
    assert "<SegmentationLab" in app
    assert "onRunSegmentation={runSegmentation}" in app
    assert "onRunQa={runSegmentationQa}" in app