Skip to content

Commit 11d5d31

Browse files
authored
Fix type hints in miele tests (home-assistant#156657)
1 parent 6f0de30 commit 11d5d31

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/components/miele/__init__.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""Tests for the Miele integration."""
22

33
from collections.abc import Awaitable, Callable
4+
from typing import Any
45
from unittest.mock import AsyncMock
56

67
from homeassistant.core import HomeAssistant
@@ -16,11 +17,15 @@ async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry)
1617
await hass.async_block_till_done()
1718

1819

19-
def get_data_callback(mock: AsyncMock) -> Callable[[int], Awaitable[None]]:
20+
def get_data_callback(
21+
mock: AsyncMock,
22+
) -> Callable[[dict[str, Any]], Awaitable[None]]:
2023
"""Get registered callback for api data push."""
2124
return mock.listen_events.call_args_list[0].kwargs.get("data_callback")
2225

2326

24-
def get_actions_callback(mock: AsyncMock) -> Callable[[int], Awaitable[None]]:
27+
def get_actions_callback(
28+
mock: AsyncMock,
29+
) -> Callable[[dict[str, Any]], Awaitable[None]]:
2530
"""Get registered callback for api data push."""
2631
return mock.listen_events.call_args_list[0].kwargs.get("actions_callback")

0 commit comments

Comments
 (0)