"""Check that the launch command rejects a mismatched model action length."""

import os
from pathlib import Path
import subprocess
import pytest


ROOT = Path(__file__).parents[1]
SCRIPT = ROOT / "scripts" / "run_serving.sh"


def _launch(horizon):
    if "WALLX_TEST_CHECKPOINT" not in os.environ:
        pytest.skip("set WALLX_TEST_CHECKPOINT to run checkpoint-specific contract tests")
    checkpoint = Path(os.environ["WALLX_TEST_CHECKPOINT"])
    env = os.environ.copy()
    env["PYTHON_BIN"] = os.environ.get("WALLX_TEST_PYTHON", "python")
    return subprocess.run(
        [
            "bash", str(SCRIPT),
            "--checkpoint-path", str(checkpoint),
            "--train-config-path", str(checkpoint / "config.yml"),
            "--action-horizon", str(horizon),
            "--dry-run",
        ],
        cwd=ROOT, env=env, text=True, capture_output=True, check=False,
    )


def test_launch_rejects_mismatch_to_training_horizon():
    result = _launch(10)
    assert result.returncode != 0
    assert "action horizon" in (result.stderr + result.stdout).lower()


def test_launch_accepts_training_horizon():
    result = _launch(32)
    assert result.returncode == 0, result.stderr
    assert "--model-config.action-horizon  32" in result.stdout