jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 01:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix snooz tests (#127468)

Browse Source
This commit is contained in:
Erik Montnemery 2024-10-05 12:13:52 +02:00 committed by GitHub
parent 00df42ba39
commit c104e66964
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/components/snooz/test_fan.py
View File

@ -3,7 +3,7 @@
from __future__ import annotations
from datetime import timedelta
from unittest.mock import Mock
from unittest.mock import Mock, patch
from pysnooz.api import SnoozDeviceState, UnknownSnoozState
from pysnooz.commands import SnoozCommandResult, SnoozCommandResultStatus
@ -32,6 +32,8 @@ from homeassistant.helpers import entity_registry as er
from . import SnoozFixture, create_mock_snooz, create_mock_snooz_config_entry
from tests.components.bluetooth import generate_ble_device
async def test_turn_on(hass: HomeAssistant, snooz_fan_entity_id: str) -> None:
"""Test turning on the device."""
@ -200,7 +202,14 @@ async def test_restore_state(
assert state.state == STATE_UNAVAILABLE
# reload entry
await create_mock_snooz_config_entry(hass, device)
with (
patch("homeassistant.components.snooz.SnoozDevice", return_value=device),
patch(
"homeassistant.components.snooz.async_ble_device_from_address",
return_value=generate_ble_device(device.address, device.name),
),
):
await hass.config_entries.async_setup(entry.entry_id)
# should match last known state
state = hass.states.get(entity_id)
@ -225,7 +234,14 @@ async def test_restore_unknown_state(
assert state.state == STATE_UNAVAILABLE
# reload entry
await create_mock_snooz_config_entry(hass, device)
with (
patch("homeassistant.components.snooz.SnoozDevice", return_value=device),
patch(
"homeassistant.components.snooz.async_ble_device_from_address",
return_value=generate_ble_device(device.address, device.name),
),
):
await hass.config_entries.async_setup(entry.entry_id)
# should match last known state
state = hass.states.get(entity_id)