"""Tests for the LocationClient."""

import inspect
from unittest.mock import Mock, patch

import pytest
import requests
from madsci.client.location_client import LocationClient
from madsci.common.types.location_types import Location
from madsci.common.types.resource_types.server_types import ResourceHierarchy
from madsci.common.utils import new_ulid_str


def test_location_client_initialization():
    """Test LocationClient initialization."""
    client = LocationClient("http://localhost:8006")
    assert str(client.location_server_url) == "http://localhost:8006/"


def test_location_client_initialization_with_trailing_slash():
    """Test LocationClient initialization with URL that already has trailing slash."""
    client = LocationClient("http://localhost:8006/")
    assert str(client.location_server_url) == "http://localhost:8006/"


def test_location_client_headers():
    """Test that headers are properly formatted."""
    client = LocationClient("http://localhost:8006")
    headers = client._get_headers()
    assert headers["Content-Type"] == "application/json"


@pytest.fixture
def sample_location():
    """Create a sample location for testing."""
    return Location(
        location_id=new_ulid_str(),
        name="Test Location",
        description="A test location",
    )


@pytest.fixture
def location_client():
    """Create a LocationClient for testing."""
    return LocationClient("http://localhost:8006")


def test_transfer_methods_exist(location_client):
    """Test that transfer methods exist and are callable."""
    # These methods should exist and be callable
    assert hasattr(location_client, "get_transfer_graph")
    assert callable(location_client.get_transfer_graph)

    assert hasattr(location_client, "plan_transfer")
    assert callable(location_client.plan_transfer)

    assert hasattr(location_client, "get_location_resources")
    assert callable(location_client.get_location_resources)


def test_transfer_method_signatures(location_client):
    """Test that transfer methods have correct signatures."""
    # Check get_transfer_graph signature
    sig = inspect.signature(location_client.get_transfer_graph)
    params = list(sig.parameters.keys())
    assert "timeout" in params

    # Check plan_transfer signature
    sig = inspect.signature(location_client.plan_transfer)
    params = list(sig.parameters.keys())
    assert "source_location_id" in params
    assert "target_location_id" in params
    assert "resource_id" in params
    assert "timeout" in params

    # Check get_location_resources signature
    sig = inspect.signature(location_client.get_location_resources)
    params = list(sig.parameters.keys())
    assert "location_id" in params
    assert "timeout" in params


def test_get_location_by_name_method_exists(location_client):
    """Test that get_location_by_name method exists and is callable."""
    assert hasattr(location_client, "get_location_by_name")
    assert callable(location_client.get_location_by_name)


def test_get_location_by_name_method_signature(location_client):
    """Test that get_location_by_name method has correct signature."""
    sig = inspect.signature(location_client.get_location_by_name)
    params = list(sig.parameters.keys())
    assert "location_name" in params
    assert "timeout" in params

    # Check return type annotation
    assert sig.return_annotation.__name__ == "Location"


@patch("madsci.client.location_client.create_http_session")
def test_get_location_by_name_method_request(mock_create_session):
    """Test that get_location_by_name makes correct HTTP request."""

    # Mock successful response
    mock_response = Mock()
    test_location_id = new_ulid_str()  # Generate valid ULID
    mock_location_data = {
        "location_id": test_location_id,
        "location_name": "test_location_name",
        "description": "Test location description",
    }
    mock_response.json.return_value = mock_location_data
    mock_response.raise_for_status.return_value = None

    mock_session = Mock()
    mock_session.get.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create a new client to use mocked session
    client = LocationClient(location_server_url="http://test/")

    # Call the method
    result = client.get_location_by_name("test_location_name")

    # Verify the request was made correctly
    mock_session.get.assert_called_once()
    call_args = mock_session.get.call_args
    # Check that the URL is correct and query parameters are used
    assert call_args[0][0].endswith("/location")
    assert call_args[1]["params"] == {"name": "test_location_name"}

    # Verify the result is a Location object
    assert isinstance(result, Location)
    assert result.location_name == "test_location_name"
    assert result.location_id == test_location_id


@patch("madsci.client.location_client.create_http_session")
def test_get_location_by_name_method_error_handling(mock_create_session):
    """Test that get_location_by_name handles errors correctly."""

    # Mock 404 response
    mock_response = Mock()
    mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError(
        "404 Not Found"
    )

    mock_session = Mock()
    mock_session.get.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    location_client = LocationClient("http://localhost:8006")

    # Call the method and expect an exception
    with pytest.raises(requests.exceptions.HTTPError):
        location_client.get_location_by_name("nonexistent_location")


@patch("madsci.client.location_client.create_http_session")
def test_get_location_resources_empty_hierarchy(mock_create_session):
    """Test get_location_resources returns empty hierarchy when no resources attached."""
    # Mock successful response with empty hierarchy
    mock_response = Mock()
    mock_hierarchy_data = {
        "ancestor_ids": [],
        "resource_id": "",
        "descendant_ids": {},
    }
    mock_response.json.return_value = mock_hierarchy_data
    mock_response.raise_for_status.return_value = None

    mock_session = Mock()
    mock_session.get.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    client = LocationClient(location_server_url="http://test/")

    # Call the method
    test_location_id = new_ulid_str()
    result = client.get_location_resources(test_location_id)

    # Verify the request was made correctly
    mock_session.get.assert_called_once()
    call_args = mock_session.get.call_args
    assert call_args[0][0].endswith(f"/location/{test_location_id}/resources")

    # Verify the result is a ResourceHierarchy object
    assert isinstance(result, ResourceHierarchy)
    assert result.ancestor_ids == []
    assert result.resource_id == ""
    assert result.descendant_ids == {}


@patch("madsci.client.location_client.create_http_session")
def test_get_location_resources_with_resource(mock_create_session):
    """Test get_location_resources returns proper hierarchy with attached resource."""
    # Mock successful response with resource hierarchy
    test_resource_id = new_ulid_str()
    test_child_id = new_ulid_str()
    mock_response = Mock()
    mock_hierarchy_data = {
        "ancestor_ids": [],
        "resource_id": test_resource_id,
        "descendant_ids": {test_resource_id: [test_child_id]},
    }
    mock_response.json.return_value = mock_hierarchy_data
    mock_response.raise_for_status.return_value = None

    mock_session = Mock()
    mock_session.get.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    client = LocationClient(location_server_url="http://test/")

    # Call the method
    test_location_id = new_ulid_str()
    result = client.get_location_resources(test_location_id)

    # Verify the request was made correctly
    mock_session.get.assert_called_once()
    call_args = mock_session.get.call_args
    assert call_args[0][0].endswith(f"/location/{test_location_id}/resources")

    # Verify the result is a ResourceHierarchy object with correct data
    assert isinstance(result, ResourceHierarchy)
    assert result.ancestor_ids == []
    assert result.resource_id == test_resource_id
    assert result.descendant_ids == {test_resource_id: [test_child_id]}


@patch("madsci.client.location_client.create_http_session")
def test_get_location_resources_error_handling(mock_create_session):
    """Test that get_location_resources handles errors correctly."""
    # Mock 404 response
    mock_response = Mock()
    mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError(
        "404 Not Found"
    )

    mock_session = Mock()
    mock_session.get.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    location_client = LocationClient("http://localhost:8006")

    # Call the method and expect an exception
    test_location_id = new_ulid_str()
    with pytest.raises(requests.exceptions.HTTPError):
        location_client.get_location_resources(test_location_id)


def test_get_location_resources_return_type_annotation(location_client):
    """Test that get_location_resources has correct return type annotation."""
    sig = inspect.signature(location_client.get_location_resources)
    assert sig.return_annotation == ResourceHierarchy


def test_remove_representation_method_exists(location_client):
    """Test that remove_representation method exists and is callable."""
    assert hasattr(location_client, "remove_representation")
    assert callable(location_client.remove_representation)


def test_remove_representation_method_signature(location_client):
    """Test that remove_representation method has correct signature."""
    sig = inspect.signature(location_client.remove_representation)
    params = list(sig.parameters.keys())
    assert "location_id" in params
    assert "node_name" in params
    assert "timeout" in params

    # Check return type annotation
    assert sig.return_annotation.__name__ == "Location"


@patch("madsci.client.location_client.create_http_session")
def test_remove_representation_method_request(mock_create_session):
    """Test that remove_representation makes correct HTTP request."""
    # Mock successful response
    mock_response = Mock()
    test_location_id = new_ulid_str()
    mock_location_data = {
        "location_id": test_location_id,
        "location_name": "test_location",
        "description": "Test location",
        "representations": {"robot_2": {"position": [1, 1, 1]}},
    }
    mock_response.json.return_value = mock_location_data
    mock_response.raise_for_status.return_value = None

    mock_session = Mock()
    mock_session.delete.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    client = LocationClient(location_server_url="http://test/")

    # Call the method
    result = client.remove_representation(test_location_id, "robot_1")

    # Verify the request was made correctly
    mock_session.delete.assert_called_once()
    call_args = mock_session.delete.call_args
    assert call_args[0][0].endswith(
        f"/location/{test_location_id}/remove_representation/robot_1"
    )

    # Verify the result is a Location object
    assert isinstance(result, Location)
    assert result.location_id == test_location_id
    assert "robot_1" not in result.representations
    assert "robot_2" in result.representations


@patch("madsci.client.location_client.create_http_session")
def test_remove_representation_error_handling(mock_create_session):
    """Test that remove_representation handles errors correctly."""
    # Mock 404 response
    mock_response = Mock()
    mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError(
        "404 Not Found"
    )

    mock_session = Mock()
    mock_session.delete.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    location_client = LocationClient("http://localhost:8006")

    # Call the method and expect an exception
    test_location_id = new_ulid_str()
    with pytest.raises(requests.exceptions.HTTPError):
        location_client.remove_representation(test_location_id, "nonexistent_node")


def test_detach_resource_method_exists(location_client):
    """Test that detach_resource method exists and is callable."""
    assert hasattr(location_client, "detach_resource")
    assert callable(location_client.detach_resource)


def test_detach_resource_method_signature(location_client):
    """Test that detach_resource method has correct signature."""
    sig = inspect.signature(location_client.detach_resource)
    params = list(sig.parameters.keys())
    assert "location_id" in params
    assert "timeout" in params

    # Check return type annotation
    assert sig.return_annotation.__name__ == "Location"


@patch("madsci.client.location_client.create_http_session")
def test_detach_resource_method_request(mock_create_session):
    """Test that detach_resource makes correct HTTP request."""
    # Mock successful response
    mock_response = Mock()
    test_location_id = new_ulid_str()
    mock_location_data = {
        "location_id": test_location_id,
        "location_name": "test_location",
        "description": "Test location",
        "resource_id": None,
    }
    mock_response.json.return_value = mock_location_data
    mock_response.raise_for_status.return_value = None

    mock_session = Mock()
    mock_session.delete.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    client = LocationClient(location_server_url="http://test/")

    # Call the method
    result = client.detach_resource(test_location_id)

    # Verify the request was made correctly
    mock_session.delete.assert_called_once()
    call_args = mock_session.delete.call_args
    assert call_args[0][0].endswith(f"/location/{test_location_id}/detach_resource")

    # Verify the result is a Location object
    assert isinstance(result, Location)
    assert result.location_id == test_location_id
    assert result.resource_id is None


@patch("madsci.client.location_client.create_http_session")
def test_detach_resource_error_handling(mock_create_session):
    """Test that detach_resource handles errors correctly."""
    # Mock 404 response
    mock_response = Mock()
    mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError(
        "404 Not Found"
    )

    mock_session = Mock()
    mock_session.delete.return_value = mock_response
    mock_create_session.return_value = mock_session

    # Create client with mocked session
    location_client = LocationClient("http://localhost:8006")

    # Call the method and expect an exception
    test_location_id = new_ulid_str()
    with pytest.raises(requests.exceptions.HTTPError):
        location_client.detach_resource(test_location_id)
