Fix type hints in miele tests (#156657)

This commit is contained in:
Åke Strandberg
2025-11-16 12:12:45 +01:00
committed by GitHub
parent 6f0de3071a
commit 11d5d314cc

View File

@@ -1,6 +1,7 @@
"""Tests for the Miele integration."""
from collections.abc import Awaitable, Callable
from typing import Any
from unittest.mock import AsyncMock
from homeassistant.core import HomeAssistant
@@ -16,11 +17,15 @@ async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry)
await hass.async_block_till_done()
def get_data_callback(mock: AsyncMock) -> Callable[[int], Awaitable[None]]:
def get_data_callback(
mock: AsyncMock,
) -> Callable[[dict[str, Any]], Awaitable[None]]:
"""Get registered callback for api data push."""
return mock.listen_events.call_args_list[0].kwargs.get("data_callback")
def get_actions_callback(mock: AsyncMock) -> Callable[[int], Awaitable[None]]:
def get_actions_callback(
mock: AsyncMock,
) -> Callable[[dict[str, Any]], Awaitable[None]]:
"""Get registered callback for api data push."""
return mock.listen_events.call_args_list[0].kwargs.get("actions_callback")