jeans/esphome
jeans/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-08-06 18:37:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

tweaks

Browse Source
This commit is contained in:
J. Nick Koston 2025-07-05 13:44:42 -05:00
parent e99b8d2daf
commit 294bd4d042
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration/test_light_calls.py
View File

@ -24,7 +24,7 @@ async def test_light_calls(
state_futures: dict[int, asyncio.Future[Any]] = {} state_futures: dict[int, asyncio.Future[Any]] = {}
states: dict[int, Any] = {} states: dict[int, Any] = {}
def on_state(state): def on_state(state: Any) -> None:
states[state.key] = state states[state.key] = state
if state.key in state_futures and not state_futures[state.key].done(): if state.key in state_futures and not state_futures[state.key].done():
state_futures[state.key].set_result(state) state_futures[state.key].set_result(state)
@ -39,7 +39,7 @@ async def test_light_calls(
rgbcw_light = next(light for light in lights if "RGBCW" in light.name) rgbcw_light = next(light for light in lights if "RGBCW" in light.name)
rgb_light = next(light for light in lights if "RGB Light" in light.name) rgb_light = next(light for light in lights if "RGB Light" in light.name)
async def wait_for_state_change(key, timeout=1.0): async def wait_for_state_change(key: int, timeout: float = 1.0) -> Any:
"""Wait for a state change for the given entity key.""" """Wait for a state change for the given entity key."""
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
state_futures[key] = loop.create_future() state_futures[key] = loop.create_future()