"""Tests for the diagnostics data provided by the TwenteMilieu integration."""
from aiohttp import ClientSession

from homeassistant.core import HomeAssistant

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


async def test_diagnostics(
    hass: HomeAssistant,
    hass_client: ClientSession,
    init_integration: MockConfigEntry,
):
    """Test diagnostics."""
    assert await get_diagnostics_for_config_entry(
        hass, hass_client, init_integration
    ) == {
        "0": "2021-11-01",
        "1": "2021-11-02",
        "2": None,
        "6": "2022-01-06",
        "10": "2021-11-03",
    }