Disable polling Sonos switches by default (#58705)

pull/58716/head
jjlawren 2021-10-29 14:43:59 -05:00 committed by GitHub
parent a4a5a2e782
commit 6e7fe13d51
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 39 additions and 0 deletions

View File

@ -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:

View File

@ -1,7 +1,10 @@
"""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