import pytest
from unittest.mock import patch, MagicMock
import nomad_client
import subprocess

@patch("subprocess.run")
@patch("nomad_client.get_node_map")
def test_get_allocation_id(mock_node_map, mock_run):
    """Test getting allocation ID for a node."""
    mock_node_map.return_value = {"node_id1": "node1"}
    
    # Mock 'nomad job status navidrome-litefs' output
    mock_job_status = MagicMock()
    mock_job_status.stdout = """
Allocations
ID        Node ID   Task Group  Version  Desired  Status   Created    Modified
abc12345  node_id1  navidrome   1        run      running  1h ago     1h ago
"""
    
    # Mock 'nomad alloc status abc12345' output
    mock_alloc_status = MagicMock()
    mock_alloc_status.stdout = "ID = abc12345-full-id"
    
    mock_run.side_effect = [mock_job_status, mock_alloc_status]
    
    alloc_id = nomad_client.get_allocation_id("node1", "navidrome-litefs")
    assert alloc_id == "abc12345-full-id"

@patch("subprocess.run")
def test_get_logs(mock_run):
    """Test fetching logs for an allocation."""
    mock_stderr = "Error: database is locked\nSome other error"
    m = MagicMock()
    m.stdout = mock_stderr
    m.return_code = 0
    mock_run.return_value = m
    
    logs = nomad_client.get_allocation_logs("abc12345", tail=20)
    assert "database is locked" in logs
    # It should have tried with -task navidrome first
    mock_run.assert_any_call(
        ["nomad", "alloc", "logs", "-stderr", "-task", "navidrome", "-n", "20", "abc12345"],
        capture_output=True, text=True, check=True
    )

@patch("subprocess.run")
def test_restart_allocation(mock_run):
    """Test restarting an allocation."""
    m = MagicMock()
    m.return_code = 0
    mock_run.return_value = m
    
    success = nomad_client.restart_allocation("abc12345")
    assert success is True
    mock_run.assert_called_with(
        ["nomad", "alloc", "restart", "abc12345"],
        capture_output=True, text=True, check=True
    )