from pathlib import Path

from pam_deploy_graph.agent import PamDeployAgent
from pam_deploy_graph.fake_runner import FakeActionRunner
from pam_deploy_graph.langgraph_runtime import LangGraphDeploymentRuntime


PARAMS = {
    "HOME_BASE_URL": "https://pam.home.example.com",
    "CLIENT_ID": "client",
    "CLIENT_SECRET": "secret",
    "AIRPORT_CODE": "HET",
    "APP_NAME": "PAM",
    "MODULE_NAME": "Node",
    "VERSION_NUMBER": "2.0.5",
    "ZIP_FILE_PATH": "C:/pkg.zip",
}


def test_langgraph_runtime_pauses_failure_and_resume_retries(tmp_path: Path):
    fake = FakeActionRunner(
        {
            "verify-ip:192.168.1.10": {
                "ACTION": "verify-ip",
                "IP": "192.168.1.10",
                "SUCCESS": "false",
                "MESSAGE": "health check failed",
            }
        }
    )
    agent = PamDeployAgent(fake_runner=fake)
    state = agent.create_state(
        params=PARAMS,
        execution_strategy="fake",
        config_path=str(tmp_path / "config.txt"),
        checkpoint_path=str(tmp_path / "checkpoint.json"),
    )
    runtime = LangGraphDeploymentRuntime(agent=agent)

    first = runtime.start(state)

    assert first.interrupted is False
    assert runtime.waiting_confirmation is False
    assert first.confirmation == {}
    assert first.state is not None
    assert first.state.paused is True
    assert first.state.pending_confirmation == ""
    assert first.state.ip_states["192.168.1.10"]["failed_stage"] == "verify-ip"

    fake.fixtures = {}
    agent.resume_state(first.state)
    second = runtime.start(first.state)

    assert second.interrupted is False
    assert runtime.waiting_confirmation is False
    assert second.state is not None
    assert second.state.pending_confirmation == ""
    assert second.state.paused is False
    assert second.state.ip_states["192.168.1.10"]["rollback_status"] == "ROLLBACK_NOT_RUN"
    assert second.state.ip_states["192.168.1.11"]["status"] == "SUCCESS"