core/tests/components/radarr/test_services.py

"""Test Radarr services."""

from unittest.mock import patch

from aiopyarr import ArrAuthenticationException, ArrConnectionException
import pytest
from syrupy.assertion import SnapshotAssertion

from homeassistant.components.radarr.const import DOMAIN
from homeassistant.components.radarr.services import (
    ATTR_ENTRY_ID,
    SERVICE_GET_MOVIES,
    SERVICE_GET_QUEUE,
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError

from . import create_entry, setup_integration

from tests.test_util.aiohttp import AiohttpClientMocker


async def test_get_queue_service(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    snapshot: SnapshotAssertion,
) -> None:
    """Test the get_queue service."""
    entry = await setup_integration(hass, aioclient_mock)

    response = await hass.services.async_call(
        DOMAIN,
        SERVICE_GET_QUEUE,
        {ATTR_ENTRY_ID: entry.entry_id},
        blocking=True,
        return_response=True,
    )

    # Explicit assertion for specific behavior
    assert len(response["movies"]) == 2

    # Snapshot for full structure validation
    assert response == snapshot


async def test_get_movies_service(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    snapshot: SnapshotAssertion,
) -> None:
    """Test the get_movies service."""
    entry = await setup_integration(hass, aioclient_mock)

    response = await hass.services.async_call(
        DOMAIN,
        SERVICE_GET_MOVIES,
        {ATTR_ENTRY_ID: entry.entry_id},
        blocking=True,
        return_response=True,
    )

    # Explicit assertion for specific behavior
    assert len(response["movies"]) == 2

    # Snapshot for full structure validation
    assert response == snapshot


@pytest.mark.parametrize(
    ("service", "method"),
    [(SERVICE_GET_QUEUE, "async_get_queue"), (SERVICE_GET_MOVIES, "async_get_movies")],
)
async def test_services_api_connection_error(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    service: str,
    method: str,
) -> None:
    """Test services with API connection error."""
    entry = await setup_integration(hass, aioclient_mock)

    with (
        patch(
            f"homeassistant.components.radarr.coordinator.RadarrClient.{method}",
            side_effect=ArrConnectionException(None, "Connection failed"),
        ),
        pytest.raises(HomeAssistantError, match="Failed to connect to Radarr"),
    ):
        await hass.services.async_call(
            DOMAIN,
            service,
            {ATTR_ENTRY_ID: entry.entry_id},
            blocking=True,
            return_response=True,
        )


@pytest.mark.parametrize(
    ("service", "method"),
    [(SERVICE_GET_QUEUE, "async_get_queue"), (SERVICE_GET_MOVIES, "async_get_movies")],
)
async def test_get_movies_service_api_auth_error(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    service: str,
    method: str,
) -> None:
    """Test services with API authentication error."""
    entry = await setup_integration(hass, aioclient_mock)

    with (
        patch(
            f"homeassistant.components.radarr.coordinator.RadarrClient.{method}",
            side_effect=ArrAuthenticationException(None, "Authentication failed"),
        ),
        pytest.raises(HomeAssistantError, match="Authentication failed for Radarr"),
    ):
        await hass.services.async_call(
            DOMAIN,
            service,
            {ATTR_ENTRY_ID: entry.entry_id},
            blocking=True,
            return_response=True,
        )


@pytest.mark.parametrize(
    "service",
    [SERVICE_GET_QUEUE, SERVICE_GET_MOVIES],
)
async def test_services_invalid_entry(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    service: str,
) -> None:
    """Test get_queue with invalid entry id."""
    # Set up at least one entry so the service gets registered
    await setup_integration(hass, aioclient_mock)

    with pytest.raises(
        ServiceValidationError, match='Config entry for integration "radarr" not found'
    ):
        await hass.services.async_call(
            DOMAIN,
            service,
            {ATTR_ENTRY_ID: "invalid_id"},
            blocking=True,
            return_response=True,
        )


@pytest.mark.parametrize(
    "service",
    [SERVICE_GET_QUEUE, SERVICE_GET_MOVIES],
)
async def test_services_entry_not_loaded(
    hass: HomeAssistant,
    aioclient_mock: AiohttpClientMocker,
    service: str,
) -> None:
    """Test get_queue with entry that's not loaded."""
    # First set up one entry to register the service
    await setup_integration(hass, aioclient_mock)

    # Now create a second entry that isn't loaded
    unloaded_entry = create_entry(hass)

    with pytest.raises(
        ServiceValidationError, match='Config entry "Mock Title" is not loaded'
    ):
        await hass.services.async_call(
            DOMAIN,
            service,
            {ATTR_ENTRY_ID: unloaded_entry.entry_id},
            blocking=True,
            return_response=True,
        )