diff --git a/tests/components/google_translate/test_tts.py b/tests/components/google_translate/test_tts.py index 4d14c7e28cb..41cecd8cd98 100644 --- a/tests/components/google_translate/test_tts.py +++ b/tests/components/google_translate/test_tts.py @@ -13,17 +13,13 @@ import pytest from homeassistant.components import tts from homeassistant.components.google_translate.const import CONF_TLD, DOMAIN -from homeassistant.components.media_player import ( - ATTR_MEDIA_CONTENT_ID, - DOMAIN as DOMAIN_MP, - SERVICE_PLAY_MEDIA, -) +from homeassistant.components.media_player import ATTR_MEDIA_CONTENT_ID from homeassistant.config import async_process_ha_core_config from homeassistant.const import ATTR_ENTITY_ID, CONF_PLATFORM from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.setup import async_setup_component -from tests.common import MockConfigEntry, async_mock_service +from tests.common import MockConfigEntry from tests.components.tts.common import retrieve_media from tests.typing import ClientSessionGenerator @@ -39,12 +35,6 @@ def mock_tts_cache_dir_autouse(mock_tts_cache_dir: Path) -> Path: return mock_tts_cache_dir -@pytest.fixture -def calls(hass: HomeAssistant) -> list[ServiceCall]: - """Mock media player calls.""" - return async_mock_service(hass, DOMAIN_MP, SERVICE_PLAY_MEDIA) - - @pytest.fixture(autouse=True) async def setup_internal_url(hass: HomeAssistant) -> None: """Set up internal url.""" @@ -126,7 +116,7 @@ async def test_tts_service( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -139,9 +129,11 @@ async def test_tts_service( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -181,7 +173,7 @@ async def test_service_say_german_config( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -194,9 +186,11 @@ async def test_service_say_german_config( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -236,7 +230,7 @@ async def test_service_say_german_service( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -249,9 +243,11 @@ async def test_service_say_german_service( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -290,7 +286,7 @@ async def test_service_say_en_uk_config( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -303,9 +299,11 @@ async def test_service_say_en_uk_config( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -345,7 +343,7 @@ async def test_service_say_en_uk_service( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -358,9 +356,11 @@ async def test_service_say_en_uk_service( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -400,7 +400,7 @@ async def test_service_say_en_couk( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -413,9 +413,11 @@ async def test_service_say_en_couk( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.OK ) assert len(mock_gtts.mock_calls) == 2 @@ -454,7 +456,7 @@ async def test_service_say_error( hass: HomeAssistant, mock_gtts: MagicMock, hass_client: ClientSessionGenerator, - calls: list[ServiceCall], + service_calls: list[ServiceCall], setup: str, tts_service: str, service_data: dict[str, Any], @@ -469,9 +471,11 @@ async def test_service_say_error( blocking=True, ) - assert len(calls) == 1 + assert len(service_calls) == 2 assert ( - await retrieve_media(hass, hass_client, calls[0].data[ATTR_MEDIA_CONTENT_ID]) + await retrieve_media( + hass, hass_client, service_calls[1].data[ATTR_MEDIA_CONTENT_ID] + ) == HTTPStatus.NOT_FOUND ) assert len(mock_gtts.mock_calls) == 2