core/tests/components/airnow/test_diagnostics.py

"""Test AirNow diagnostics."""

from unittest.mock import patch

import pytest
from syrupy import SnapshotAssertion
from syrupy.filters import props

from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry
from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator


@pytest.mark.usefixtures("setup_airnow")
async def test_entry_diagnostics(
    hass: HomeAssistant,
    config_entry: MockConfigEntry,
    hass_client: ClientSessionGenerator,
    snapshot: SnapshotAssertion,
) -> None:
    """Test config entry diagnostics."""

    # Fake LocalTimeZoneInfo
    with patch(
        "homeassistant.util.dt.async_get_time_zone",
        return_value="PST",
    ):
        assert await hass.config_entries.async_setup(config_entry.entry_id)
        assert await get_diagnostics_for_config_entry(
            hass, hass_client, config_entry
        ) == snapshot(exclude=props("created_at", "modified_at"))