core/tests/components/twentemilieu/test_diagnostics.py

"""Tests for the diagnostics data provided by the TwenteMilieu integration."""

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


async def test_diagnostics(
    hass: HomeAssistant,
    hass_client: ClientSessionGenerator,
    init_integration: MockConfigEntry,
) -> None:
    """Test diagnostics."""
    assert await get_diagnostics_for_config_entry(
        hass, hass_client, init_integration
    ) == {
        "WasteType.NON_RECYCLABLE": ["2021-11-01", "2021-12-01"],
        "WasteType.ORGANIC": ["2021-11-02"],
        "WasteType.PAPER": [],
        "WasteType.TREE": ["2022-01-06"],
        "WasteType.PACKAGES": ["2021-11-03"],
    }