37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
"""Common fixtures for the AccuWeather tests."""
|
||
|
|
|
||
|
|
from collections.abc import Generator
|
||
|
|
from unittest.mock import AsyncMock, patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from homeassistant.components.accuweather.const import DOMAIN
|
||
|
|
|
||
|
|
from tests.common import load_json_array_fixture, load_json_object_fixture
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_accuweather_client() -> Generator[AsyncMock, None, None]:
|
||
|
|
"""Mock a AccuWeather client."""
|
||
|
|
current = load_json_object_fixture("current_conditions_data.json", DOMAIN)
|
||
|
|
forecast = load_json_array_fixture("forecast_data.json", DOMAIN)
|
||
|
|
location = load_json_object_fixture("location_data.json", DOMAIN)
|
||
|
|
|
||
|
|
with (
|
||
|
|
patch(
|
||
|
|
"homeassistant.components.accuweather.AccuWeather", autospec=True
|
||
|
|
) as mock_client,
|
||
|
|
patch(
|
||
|
|
"homeassistant.components.accuweather.config_flow.AccuWeather",
|
||
|
|
new=mock_client,
|
||
|
|
),
|
||
|
|
):
|
||
|
|
client = mock_client.return_value
|
||
|
|
client.async_get_location.return_value = location
|
||
|
|
client.async_get_current_conditions.return_value = current
|
||
|
|
client.async_get_daily_forecast.return_value = forecast
|
||
|
|
client.location_key = "0123456"
|
||
|
|
client.requests_remaining = 10
|
||
|
|
|
||
|
|
yield client
|