Rename fixture function in stt tests (#87630)

This commit is contained in:
epenet 2023-02-07 16:06:53 +01:00 committed by GitHub
parent b478b4fa16
commit 3a83b2f66f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -17,13 +17,15 @@ from homeassistant.components.stt import (
SpeechResultState, SpeechResultState,
async_get_provider, async_get_provider,
) )
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import mock_platform from tests.common import mock_platform
from tests.typing import ClientSessionGenerator
class TestProvider(Provider): class MockProvider(Provider):
"""Test provider.""" """Mock provider."""
fail_process_audio = False fail_process_audio = False
@ -32,7 +34,7 @@ class TestProvider(Provider):
self.calls = [] self.calls = []
@property @property
def supported_languages(self): def supported_languages(self) -> list[str]:
"""Return a list of supported languages.""" """Return a list of supported languages."""
return ["en"] return ["en"]
@ -73,21 +75,23 @@ class TestProvider(Provider):
@pytest.fixture @pytest.fixture
def test_provider(): def mock_provider() -> MockProvider:
"""Test provider fixture.""" """Test provider fixture."""
return TestProvider() return MockProvider()
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def mock_setup(hass, test_provider): async def mock_setup(hass: HomeAssistant, mock_provider: MockProvider) -> None:
"""Set up a test provider.""" """Set up a test provider."""
mock_platform( mock_platform(
hass, "test.stt", Mock(async_get_engine=AsyncMock(return_value=test_provider)) hass, "test.stt", Mock(async_get_engine=AsyncMock(return_value=mock_provider))
) )
assert await async_setup_component(hass, "stt", {"stt": {"platform": "test"}}) assert await async_setup_component(hass, "stt", {"stt": {"platform": "test"}})
async def test_get_provider_info(hass, hass_client): async def test_get_provider_info(
hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
"""Test engine that doesn't exist.""" """Test engine that doesn't exist."""
client = await hass_client() client = await hass_client()
response = await client.get("/api/stt/test") response = await client.get("/api/stt/test")
@ -102,14 +106,18 @@ async def test_get_provider_info(hass, hass_client):
} }
async def test_get_non_existing_provider_info(hass, hass_client): async def test_get_non_existing_provider_info(
hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
"""Test streaming to engine that doesn't exist.""" """Test streaming to engine that doesn't exist."""
client = await hass_client() client = await hass_client()
response = await client.get("/api/stt/not_exist") response = await client.get("/api/stt/not_exist")
assert response.status == HTTPStatus.NOT_FOUND assert response.status == HTTPStatus.NOT_FOUND
async def test_stream_audio(hass, hass_client): async def test_stream_audio(
hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
"""Test streaming audio and getting response.""" """Test streaming audio and getting response."""
client = await hass_client() client = await hass_client()
response = await client.post( response = await client.post(
@ -144,10 +152,16 @@ async def test_stream_audio(hass, hass_client):
), ),
), ),
) )
async def test_metadata_errors(hass, hass_client, header, status, error): async def test_metadata_errors(
hass: HomeAssistant,
hass_client: ClientSessionGenerator,
header: str | None,
status: int,
error: str,
) -> None:
"""Test metadata errors.""" """Test metadata errors."""
client = await hass_client() client = await hass_client()
headers = {} headers: dict[str, str] = {}
if header: if header:
headers["X-Speech-Content"] = header headers["X-Speech-Content"] = header
@ -156,6 +170,6 @@ async def test_metadata_errors(hass, hass_client, header, status, error):
assert await response.text() == error assert await response.text() == error
async def test_get_provider(hass, test_provider): async def test_get_provider(hass: HomeAssistant, mock_provider: MockProvider) -> None:
"""Test we can get STT providers.""" """Test we can get STT providers."""
assert test_provider == async_get_provider(hass, "test") assert mock_provider == async_get_provider(hass, "test")