Pass assumed_state property in universal media player (#87846)

Pass assumed_state property in universal media player
This commit is contained in:
Artem Draft 2023-02-25 09:43:58 +03:00 committed by GitHub
parent f52a5f6965
commit f18c0bf626
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 0 deletions

View file

@ -47,6 +47,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.components.media_player.browse_media import BrowseMedia
from homeassistant.const import (
ATTR_ASSUMED_STATE,
ATTR_ENTITY_ID,
ATTR_ENTITY_PICTURE,
ATTR_SUPPORTED_FEATURES,
@ -291,6 +292,11 @@ class UniversalMediaPlayer(MediaPlayerEntity):
"""Return the name of universal player."""
return self._name
@property
def assumed_state(self) -> bool:
"""Return True if unable to access real state of the entity."""
return self._child_attr(ATTR_ASSUMED_STATE)
@property
def state(self):
"""Return the current state of media player.

View file

@ -492,6 +492,13 @@ async def test_state_children_only(hass: HomeAssistant, mock_states) -> None:
await ump.async_update()
assert ump.state == STATE_PLAYING
mock_states.mock_mp_1._state = STATE_ON
mock_states.mock_mp_1._attr_assumed_state = True
mock_states.mock_mp_1.async_schedule_update_ha_state()
await hass.async_block_till_done()
await ump.async_update()
assert ump.assumed_state is True
async def test_state_with_children_and_attrs(
hass: HomeAssistant, config_children_and_attr, mock_states