core/tests/components/eafm/conftest.py

"""eafm fixtures."""

from unittest.mock import patch

import pytest


@pytest.fixture
def mock_get_stations():
    """Mock aioeafm.get_stations."""
    with patch("homeassistant.components.eafm.config_flow.get_stations") as patched:
        yield patched


@pytest.fixture
def mock_get_station():
    """Mock aioeafm.get_station."""
    with patch("homeassistant.components.eafm.sensor.get_station") as patched:
        yield patched