jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-16 09:47:13 +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 __future__ import annotations
from datetime import timedelta from datetime import timedelta
from unittest.mock import Mock from unittest.mock import Mock, patch
from pysnooz.api import SnoozDeviceState, UnknownSnoozState from pysnooz.api import SnoozDeviceState, UnknownSnoozState
from pysnooz.commands import SnoozCommandResult, SnoozCommandResultStatus 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 . 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: async def test_turn_on(hass: HomeAssistant, snooz_fan_entity_id: str) -> None:
"""Test turning on the device.""" """Test turning on the device."""
@ -200,7 +202,14 @@ async def test_restore_state(
assert state.state == STATE_UNAVAILABLE assert state.state == STATE_UNAVAILABLE
# reload entry # 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 # should match last known state
state = hass.states.get(entity_id) state = hass.states.get(entity_id)
@ -225,7 +234,14 @@ async def test_restore_unknown_state(
assert state.state == STATE_UNAVAILABLE assert state.state == STATE_UNAVAILABLE
# reload entry # 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 # should match last known state
state = hass.states.get(entity_id) state = hass.states.get(entity_id)