evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use service_calls fixture in google_translate tests (#120920)

Browse source
This commit is contained in:
epenet 2024-07-01 13:10:39 +02:00 • committed by GitHub
parent 09ff44e59e
commit 4cc414fbf8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 37 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
tests/components/google_translate/test_tts.py
View file

@ -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