import pytest
from pathlib import Path
from client.workflow import find_bridge_inputs, patch_inputs, load_workflow

FIXTURE_PATH = Path(__file__).parent / "fixtures" / "sample_workflow.json"


def test_find_bridge_inputs():
    wf = load_workflow(FIXTURE_PATH)
    bridge_map = find_bridge_inputs(wf)
    assert "prompt" in bridge_map
    assert "steps" in bridge_map
    assert "cfg" in bridge_map
    assert bridge_map["prompt"] == ("1", "STRING")
    assert bridge_map["steps"] == ("2", "INT")


def test_patch_inputs():
    wf = load_workflow(FIXTURE_PATH)
    patched = patch_inputs(wf, {"prompt": "hello world", "steps": 30})
    assert patched["1"]["inputs"]["value"] == "hello world"
    assert patched["2"]["inputs"]["value"] == 30
    # 원본 불변 확인
    original = load_workflow(FIXTURE_PATH)
    assert original["1"]["inputs"]["value"] == "default prompt"


def test_patch_unknown_input_raises():
    wf = load_workflow(FIXTURE_PATH)
    with pytest.raises(ValueError, match="not found in workflow"):
        patch_inputs(wf, {"nonexistent": "value"})