Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Allow string formatting for dispatcher SignalType (#114174)

Browse Source
This commit is contained in:
Marc Mueller
2024-03-26 10:38:29 +01:00
committed by GitHub
parent dd43947ca0
commit eb81a4204e
5 changed files with 58 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/helpers/test_dispatcher.py
View File

@@ -7,6 +7,7 @@ import pytest
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import (
SignalType,
SignalTypeFormat,
async_dispatcher_connect,
async_dispatcher_send,
)
@@ -58,6 +59,27 @@ async def test_signal_type(hass: HomeAssistant) -> None:
assert calls == [("Hello", 2), ("World", 3), ("x", 4)]
async def test_signal_type_format(hass: HomeAssistant) -> None:
"""Test dispatcher with SignalType and format."""
signal: SignalTypeFormat[str, int] = SignalTypeFormat("test-{}")
calls: list[tuple[str, int]] = []
def test_funct(data1: str, data2: int) -> None:
calls.append((data1, data2))
async_dispatcher_connect(hass, signal.format("unique-id"), test_funct)
async_dispatcher_send(hass, signal.format("unique-id"), "Hello", 2)
await hass.async_block_till_done()
assert calls == [("Hello", 2)]
# Test compatibility with string keys
async_dispatcher_send(hass, "test-{}".format("unique-id"), "x", 4)
await hass.async_block_till_done()
assert calls == [("Hello", 2), ("x", 4)]
async def test_simple_function_unsub(hass: HomeAssistant) -> None:
"""Test simple function (executor) and unsub."""
calls1 = []