103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
"""Fixtures for component testing."""
|
|
from collections.abc import Generator
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def patch_zeroconf_multiple_catcher() -> Generator[None, None, None]:
|
|
"""Patch zeroconf wrapper that detects if multiple instances are used."""
|
|
with patch(
|
|
"homeassistant.components.zeroconf.install_multiple_zeroconf_catcher",
|
|
side_effect=lambda zc: None,
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def prevent_io() -> Generator[None, None, None]:
|
|
"""Fixture to prevent certain I/O from happening."""
|
|
with patch(
|
|
"homeassistant.components.http.ban.load_yaml_config_file",
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def entity_registry_enabled_by_default() -> Generator[None, None, None]:
|
|
"""Test fixture that ensures all entities are enabled in the registry."""
|
|
with patch(
|
|
"homeassistant.helpers.entity.Entity.entity_registry_enabled_default",
|
|
return_value=True,
|
|
):
|
|
yield
|
|
|
|
|
|
# Blueprint test fixtures
|
|
@pytest.fixture(name="stub_blueprint_populate")
|
|
def stub_blueprint_populate_fixture() -> Generator[None, Any, None]:
|
|
"""Stub copying the blueprints to the config folder."""
|
|
from tests.components.blueprint.common import stub_blueprint_populate_fixture_helper
|
|
|
|
yield from stub_blueprint_populate_fixture_helper()
|
|
|
|
|
|
# TTS test fixtures
|
|
@pytest.fixture(name="mock_tts_get_cache_files")
|
|
def mock_tts_get_cache_files_fixture():
|
|
"""Mock the list TTS cache function."""
|
|
from tests.components.tts.common import mock_tts_get_cache_files_fixture_helper
|
|
|
|
yield from mock_tts_get_cache_files_fixture_helper()
|
|
|
|
|
|
@pytest.fixture(name="mock_tts_init_cache_dir")
|
|
def mock_tts_init_cache_dir_fixture(
|
|
init_tts_cache_dir_side_effect: Any,
|
|
) -> Generator[MagicMock, None, None]:
|
|
"""Mock the TTS cache dir in memory."""
|
|
from tests.components.tts.common import mock_tts_init_cache_dir_fixture_helper
|
|
|
|
yield from mock_tts_init_cache_dir_fixture_helper(init_tts_cache_dir_side_effect)
|
|
|
|
|
|
@pytest.fixture(name="init_tts_cache_dir_side_effect")
|
|
def init_tts_cache_dir_side_effect_fixture() -> Any:
|
|
"""Return the cache dir."""
|
|
from tests.components.tts.common import (
|
|
init_tts_cache_dir_side_effect_fixture_helper,
|
|
)
|
|
|
|
return init_tts_cache_dir_side_effect_fixture_helper()
|
|
|
|
|
|
@pytest.fixture(name="mock_tts_cache_dir")
|
|
def mock_tts_cache_dir_fixture(
|
|
tmp_path, mock_tts_init_cache_dir, mock_tts_get_cache_files, request
|
|
):
|
|
"""Mock the TTS cache dir with empty dir."""
|
|
from tests.components.tts.common import mock_tts_cache_dir_fixture_helper
|
|
|
|
yield from mock_tts_cache_dir_fixture_helper(
|
|
tmp_path, mock_tts_init_cache_dir, mock_tts_get_cache_files, request
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="tts_mutagen_mock")
|
|
def tts_mutagen_mock_fixture():
|
|
"""Mock writing tags."""
|
|
from tests.components.tts.common import tts_mutagen_mock_fixture_helper
|
|
|
|
yield from tts_mutagen_mock_fixture_helper()
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def prevent_ffmpeg_subprocess() -> Generator[None, None, None]:
|
|
"""Prevent ffmpeg from creating a subprocess."""
|
|
with patch(
|
|
"homeassistant.components.ffmpeg.FFVersion.get_version", return_value="6.0"
|
|
):
|
|
yield
|