"""Neustart-Knopf."""

from __future__ import annotations

from unittest.mock import MagicMock

from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant
from pytest_homeassistant_custom_component.common import MockConfigEntry

from tests.helpers import entity_id_for


async def test_knopf_startet_das_geraet_neu(
    hass: HomeAssistant, init_integration: MockConfigEntry, mock_client: MagicMock
) -> None:
    await hass.services.async_call(
        "button",
        "press",
        {ATTR_ENTITY_ID: entity_id_for(hass, "button", "restart")},
        blocking=True,
    )
    mock_client.reboot.assert_awaited_once()