9a79320861
* Mark executor jobs as background unless created from a tracked task If the current task is not tracked the executor job should not be a background task to avoid delaying startup and shutdown. Currently any executor job created in a untracked task or background task would end up being tracked and delaying startup/shutdown * import exec has the same issue * Avoid tracking import executor jobs There is no reason to track these jobs as they are always awaited and we do not want to support fire and forget import executor jobs * fix xiaomi_miio * lots of fire time changed without background await * revert changes moved to other PR * more * more * more * m * m * p * fix fire and forget tests * scrape * sonos * system * more * capture callback before block * coverage * more * more races * more races * more * missed some * more fixes * missed some more * fix * remove unneeded * one more race * two
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
"""Tests for common SonosSpeaker behavior."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.sonos.const import DATA_SONOS, SCAN_INTERVAL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
|
|
async def test_fallback_to_polling(
|
|
hass: HomeAssistant,
|
|
config_entry,
|
|
soco,
|
|
fire_zgs_event,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test that polling fallback works."""
|
|
config_entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
# Do not wait on background tasks here because the
|
|
# subscription callback will fire an unsub the polling check
|
|
await hass.async_block_till_done()
|
|
await fire_zgs_event()
|
|
|
|
speaker = list(hass.data[DATA_SONOS].discovered.values())[0]
|
|
assert speaker.soco is soco
|
|
assert speaker._subscriptions
|
|
assert not speaker.subscriptions_failed
|
|
|
|
caplog.clear()
|
|
|
|
# Ensure subscriptions are cancelled and polling methods are called when subscriptions time out
|
|
with (
|
|
patch("homeassistant.components.sonos.media.SonosMedia.poll_media"),
|
|
patch(
|
|
"homeassistant.components.sonos.speaker.SonosSpeaker.subscription_address"
|
|
),
|
|
):
|
|
async_fire_time_changed(hass, dt_util.utcnow() + SCAN_INTERVAL)
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
assert not speaker._subscriptions
|
|
assert speaker.subscriptions_failed
|
|
assert "Activity on Zone A from SonosSpeaker.update_volume" in caplog.text
|
|
|
|
|
|
async def test_subscription_creation_fails(
|
|
hass: HomeAssistant, async_setup_sonos
|
|
) -> None:
|
|
"""Test that subscription creation failures are handled."""
|
|
with patch(
|
|
"homeassistant.components.sonos.speaker.SonosSpeaker._subscribe",
|
|
side_effect=ConnectionError("Took too long"),
|
|
):
|
|
await async_setup_sonos()
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
speaker = list(hass.data[DATA_SONOS].discovered.values())[0]
|
|
assert not speaker._subscriptions
|
|
|
|
with patch.object(speaker, "_resub_cooldown_expires_at", None):
|
|
speaker.speaker_activity("discovery")
|
|
await hass.async_block_till_done()
|
|
|
|
assert speaker._subscriptions
|