"""Helfer fuer die Home-Assistant-Tests."""

from __future__ import annotations

from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from custom_components.ninstrip.const import DOMAIN
from tests_unit.samples import MAC_K1Y


def entity_id_for(hass: HomeAssistant, platform: str, key: str, mac: str = MAC_K1Y) -> str:
    """Entitaeten ueber ihre unique_id suchen, statt Namen zu raten."""
    entity_id = er.async_get(hass).async_get_entity_id(platform, DOMAIN, f"{mac}_{key}")
    assert entity_id is not None, f"keine {platform}-Entitaet mit dem Schluessel {key}"
    return entity_id


async def abfragen(hass: HomeAssistant, freezer) -> None:
    """Einen Abfragezyklus vorspulen (Vorgabe: 30 s)."""
    from datetime import timedelta

    from pytest_homeassistant_custom_component.common import async_fire_time_changed

    freezer.tick(timedelta(seconds=31))
    async_fire_time_changed(hass)
    await hass.async_block_till_done()