Use freezer.tick once more in SamsungTV (#143970)

This commit is contained in:
epenet 2025-04-30 17:50:13 +02:00 committed by GitHub
parent 837592381a
commit 70133da025
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 16 deletions

View File

@ -3,24 +3,13 @@
from __future__ import annotations
from collections.abc import Mapping
from datetime import timedelta
from typing import Any
from homeassistant.components.samsungtv.const import DOMAIN, ENTRY_RELOAD_COOLDOWN
from homeassistant.components.samsungtv.const import DOMAIN
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.util import dt as dt_util
from tests.common import MockConfigEntry, async_fire_time_changed
async def async_wait_config_entry_reload(hass: HomeAssistant) -> None:
"""Wait for the config entry to reload."""
await hass.async_block_till_done()
async_fire_time_changed(
hass, dt_util.utcnow() + timedelta(seconds=ENTRY_RELOAD_COOLDOWN)
)
await hass.async_block_till_done()
from tests.common import MockConfigEntry
async def setup_samsungtv_entry(

View File

@ -41,6 +41,7 @@ from homeassistant.components.samsungtv.const import (
CONF_SSDP_RENDERING_CONTROL_LOCATION,
DOMAIN,
ENCRYPTED_WEBSOCKET_PORT,
ENTRY_RELOAD_COOLDOWN,
METHOD_ENCRYPTED_WEBSOCKET,
METHOD_WEBSOCKET,
TIMEOUT_WEBSOCKET,
@ -79,7 +80,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceNotSupported
from homeassistant.setup import async_setup_component
from . import async_wait_config_entry_reload, setup_samsungtv_entry
from . import setup_samsungtv_entry
from .const import (
MOCK_CONFIG,
MOCK_ENTRY_WS_WITH_MAC,
@ -1154,7 +1155,10 @@ async def test_select_source_app(hass: HomeAssistant, remotews: Mock) -> None:
@pytest.mark.usefixtures("rest_api")
async def test_websocket_unsupported_remote_control(
hass: HomeAssistant, remotews: Mock, caplog: pytest.LogCaptureFixture
hass: HomeAssistant,
remotews: Mock,
freezer: FrozenDateTimeFactory,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test for turn_off."""
entry = await setup_samsungtv_entry(hass, MOCK_ENTRY_WS)
@ -1188,7 +1192,12 @@ async def test_websocket_unsupported_remote_control(
"'unrecognized method value : ms.remote.control'" in caplog.text
)
await async_wait_config_entry_reload(hass)
# Wait config_entry reload
await hass.async_block_till_done()
freezer.tick(timedelta(seconds=ENTRY_RELOAD_COOLDOWN))
async_fire_time_changed(hass)
await hass.async_block_till_done()
# ensure reauth triggered, and method/port updated
assert [
flow