Skip to content
Snippets Groups Projects
Unverified Commit 6e7fe13d authored by jjlawren's avatar jjlawren Committed by GitHub
Browse files

Disable polling Sonos switches by default (#58705)

parent a4a5a2e7
No related branches found
No related tags found
No related merge requests found
......@@ -136,6 +136,7 @@ class SonosSwitchEntity(SonosEntity, SwitchEntity):
self._attr_icon = FEATURE_ICONS.get(feature_type)
if feature_type in POLL_REQUIRED:
self._attr_entity_registry_enabled_default = False
self._attr_should_poll = True
async def _async_poll(self) -> None:
......
"""Tests for the Sonos Alarm switch platform."""
from copy import copy
from datetime import timedelta
from unittest.mock import patch
from homeassistant.components.sonos import DOMAIN
from homeassistant.components.sonos.const import DATA_SONOS_DISCOVERY_MANAGER
from homeassistant.components.sonos.switch import (
ATTR_DURATION,
ATTR_ID,
......@@ -10,9 +13,15 @@ from homeassistant.components.sonos.switch import (
ATTR_RECURRENCE,
ATTR_VOLUME,
)
from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY
from homeassistant.const import ATTR_TIME, STATE_ON
from homeassistant.helpers.entity_registry import async_get as async_get_entity_registry
from homeassistant.setup import async_setup_component
from homeassistant.util import dt
from .conftest import SonosMockEvent
from tests.common import async_fire_time_changed
async def setup_platform(hass, config_entry, config):
......@@ -67,7 +76,36 @@ async def test_switch_attributes(hass, config_entry, config, soco):
crossfade_state = hass.states.get(crossfade.entity_id)
assert crossfade_state.state == STATE_ON
# Ensure switches are disabled
status_light = entity_registry.entities["switch.sonos_zone_a_status_light"]
assert hass.states.get(status_light.entity_id) is None
touch_controls = entity_registry.entities["switch.sonos_zone_a_touch_controls"]
assert hass.states.get(touch_controls.entity_id) is None
# Enable disabled switches
for entity in (status_light, touch_controls):
entity_registry.async_update_entity(
entity_id=entity.entity_id, disabled_by=None
)
await hass.async_block_till_done()
# Fire event to cancel poll timer and avoid triggering errors during time jump
service = soco.contentDirectory
empty_event = SonosMockEvent(soco, service, {})
subscription = service.subscribe.return_value
subscription.callback(event=empty_event)
await hass.async_block_till_done()
# Mock shutdown calls during config entry reload
with patch.object(hass.data[DATA_SONOS_DISCOVERY_MANAGER], "async_shutdown") as m:
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()
assert m.called
status_light_state = hass.states.get(status_light.entity_id)
assert status_light_state.state == STATE_ON
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment