import os
import subprocess
import time


COMPOSE_FILE = os.path.join(os.path.dirname(__file__), "..", "docker-compose.test.yml")


def _compose_exec(cmd: str):
    subprocess.run(
        ["docker", "compose", "-f", COMPOSE_FILE, "exec", "system-under-test",
         "bash", "-c", cmd],
        capture_output=True, timeout=15,
    )


def _wait_for_settled(base_url, client, timeout=30):
    deadline = time.monotonic() + timeout
    while time.monotonic() < deadline:
        resp = client.get(f"{base_url}/unlock/status")
        state = resp.json()["state"]
        if state in ("idle", "error", "ready"):
            return state
        time.sleep(0.5)
    return None


def test_unlock_status_idle(base_url, api_client):
    # Act
    response = api_client.get(f"{base_url}/unlock/status")

    # Assert
    assert response.status_code == 200
    data = response.json()
    assert data["state"] == "idle"
    assert data["error"] is None


def test_unlock_missing_archive(base_url, api_client):
    # Arrange
    payload = {"email": "test@azaion.com", "password": "testpass"}

    # Act
    response = api_client.post(f"{base_url}/unlock", json=payload)

    # Assert
    assert response.status_code == 404


def test_unlock_concurrent_returns_current_state(base_url, api_client):
    # Arrange
    _compose_exec("dd if=/dev/urandom of=/tmp/test.enc bs=1024 count=1 2>/dev/null")
    payload = {"email": "test@azaion.com", "password": "testpass"}

    try:
        # Act
        first = api_client.post(f"{base_url}/unlock", json=payload)
        second = api_client.post(f"{base_url}/unlock", json=payload)

        # Assert
        assert first.status_code == 200
        assert second.status_code == 200
        assert second.json()["state"] != "idle"
    finally:
        _compose_exec("rm -f /tmp/test.enc /tmp/test.tar")
        _wait_for_settled(base_url, api_client)