github
/
ha-core
mirror of https://github.com/home-assistant/core
1
Fork 0
Code Releases Activity

Use new constants in dlna_dmr media player (#78045)

This commit is contained in:
epenet 2022-09-09 09:03:59 +02:00 committed by GitHub
parent 7ff23506fe
commit 8b3ce8c58c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 179 additions and 183 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

140
homeassistant/components/dlna_dmr/const.py
View File

@ -7,7 +7,7 @@ from typing import Final
from async_upnp_client.profiles.dlna import PlayMode as _PlayMode from async_upnp_client.profiles.dlna import PlayMode as _PlayMode
from homeassistant.components.media_player import const as _mp_const from homeassistant.components.media_player import MediaType, RepeatMode
LOGGER = logging.getLogger(__package__) LOGGER = logging.getLogger(__package__)
@ -28,66 +28,66 @@ PROTOCOL_ANY: Final = "*"
STREAMABLE_PROTOCOLS: Final = [PROTOCOL_HTTP, PROTOCOL_RTSP, PROTOCOL_ANY] STREAMABLE_PROTOCOLS: Final = [PROTOCOL_HTTP, PROTOCOL_RTSP, PROTOCOL_ANY]
# Map UPnP class to media_player media_content_type # Map UPnP class to media_player media_content_type
MEDIA_TYPE_MAP: Mapping[str, str] = { MEDIA_TYPE_MAP: Mapping[str, MediaType] = {
"object": _mp_const.MEDIA_TYPE_URL, "object": MediaType.URL,
"object.item": _mp_const.MEDIA_TYPE_URL, "object.item": MediaType.URL,
"object.item.imageItem": _mp_const.MEDIA_TYPE_IMAGE, "object.item.imageItem": MediaType.IMAGE,
"object.item.imageItem.photo": _mp_const.MEDIA_TYPE_IMAGE, "object.item.imageItem.photo": MediaType.IMAGE,
"object.item.audioItem": _mp_const.MEDIA_TYPE_MUSIC, "object.item.audioItem": MediaType.MUSIC,
"object.item.audioItem.musicTrack": _mp_const.MEDIA_TYPE_MUSIC, "object.item.audioItem.musicTrack": MediaType.MUSIC,
"object.item.audioItem.audioBroadcast": _mp_const.MEDIA_TYPE_MUSIC, "object.item.audioItem.audioBroadcast": MediaType.MUSIC,
"object.item.audioItem.audioBook": _mp_const.MEDIA_TYPE_PODCAST, "object.item.audioItem.audioBook": MediaType.PODCAST,
"object.item.videoItem": _mp_const.MEDIA_TYPE_VIDEO, "object.item.videoItem": MediaType.VIDEO,
"object.item.videoItem.movie": _mp_const.MEDIA_TYPE_MOVIE, "object.item.videoItem.movie": MediaType.MOVIE,
"object.item.videoItem.videoBroadcast": _mp_const.MEDIA_TYPE_TVSHOW, "object.item.videoItem.videoBroadcast": MediaType.TVSHOW,
"object.item.videoItem.musicVideoClip": _mp_const.MEDIA_TYPE_VIDEO, "object.item.videoItem.musicVideoClip": MediaType.VIDEO,
"object.item.playlistItem": _mp_const.MEDIA_TYPE_PLAYLIST, "object.item.playlistItem": MediaType.PLAYLIST,
"object.item.textItem": _mp_const.MEDIA_TYPE_URL, "object.item.textItem": MediaType.URL,
"object.item.bookmarkItem": _mp_const.MEDIA_TYPE_URL, "object.item.bookmarkItem": MediaType.URL,
"object.item.epgItem": _mp_const.MEDIA_TYPE_EPISODE, "object.item.epgItem": MediaType.EPISODE,
"object.item.epgItem.audioProgram": _mp_const.MEDIA_TYPE_EPISODE, "object.item.epgItem.audioProgram": MediaType.EPISODE,
"object.item.epgItem.videoProgram": _mp_const.MEDIA_TYPE_EPISODE, "object.item.epgItem.videoProgram": MediaType.EPISODE,
"object.container": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container": MediaType.PLAYLIST,
"object.container.person": _mp_const.MEDIA_TYPE_ARTIST, "object.container.person": MediaType.ARTIST,
"object.container.person.musicArtist": _mp_const.MEDIA_TYPE_ARTIST, "object.container.person.musicArtist": MediaType.ARTIST,
"object.container.playlistContainer": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container.playlistContainer": MediaType.PLAYLIST,
"object.container.album": _mp_const.MEDIA_TYPE_ALBUM, "object.container.album": MediaType.ALBUM,
"object.container.album.musicAlbum": _mp_const.MEDIA_TYPE_ALBUM, "object.container.album.musicAlbum": MediaType.ALBUM,
"object.container.album.photoAlbum": _mp_const.MEDIA_TYPE_ALBUM, "object.container.album.photoAlbum": MediaType.ALBUM,
"object.container.genre": _mp_const.MEDIA_TYPE_GENRE, "object.container.genre": MediaType.GENRE,
"object.container.genre.musicGenre": _mp_const.MEDIA_TYPE_GENRE, "object.container.genre.musicGenre": MediaType.GENRE,
"object.container.genre.movieGenre": _mp_const.MEDIA_TYPE_GENRE, "object.container.genre.movieGenre": MediaType.GENRE,
"object.container.channelGroup": _mp_const.MEDIA_TYPE_CHANNELS, "object.container.channelGroup": MediaType.CHANNELS,
"object.container.channelGroup.audioChannelGroup": _mp_const.MEDIA_TYPE_CHANNELS, "object.container.channelGroup.audioChannelGroup": MediaType.CHANNELS,
"object.container.channelGroup.videoChannelGroup": _mp_const.MEDIA_TYPE_CHANNELS, "object.container.channelGroup.videoChannelGroup": MediaType.CHANNELS,
"object.container.epgContainer": _mp_const.MEDIA_TYPE_TVSHOW, "object.container.epgContainer": MediaType.TVSHOW,
"object.container.storageSystem": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container.storageSystem": MediaType.PLAYLIST,
"object.container.storageVolume": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container.storageVolume": MediaType.PLAYLIST,
"object.container.storageFolder": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container.storageFolder": MediaType.PLAYLIST,
"object.container.bookmarkFolder": _mp_const.MEDIA_TYPE_PLAYLIST, "object.container.bookmarkFolder": MediaType.PLAYLIST,
} }
# Map media_player media_content_type to UPnP class. Not everything will map # Map media_player media_content_type to UPnP class. Not everything will map
# directly, in which case it's not specified and other defaults will be used. # directly, in which case it's not specified and other defaults will be used.
MEDIA_UPNP_CLASS_MAP: Mapping[str, str] = { MEDIA_UPNP_CLASS_MAP: Mapping[MediaType | str, str] = {
_mp_const.MEDIA_TYPE_ALBUM: "object.container.album.musicAlbum", MediaType.ALBUM: "object.container.album.musicAlbum",
_mp_const.MEDIA_TYPE_ARTIST: "object.container.person.musicArtist", MediaType.ARTIST: "object.container.person.musicArtist",
_mp_const.MEDIA_TYPE_CHANNEL: "object.item.videoItem.videoBroadcast", MediaType.CHANNEL: "object.item.videoItem.videoBroadcast",
_mp_const.MEDIA_TYPE_CHANNELS: "object.container.channelGroup", MediaType.CHANNELS: "object.container.channelGroup",
_mp_const.MEDIA_TYPE_COMPOSER: "object.container.person.musicArtist", MediaType.COMPOSER: "object.container.person.musicArtist",
_mp_const.MEDIA_TYPE_CONTRIBUTING_ARTIST: "object.container.person.musicArtist", MediaType.CONTRIBUTING_ARTIST: "object.container.person.musicArtist",
_mp_const.MEDIA_TYPE_EPISODE: "object.item.epgItem.videoProgram", MediaType.EPISODE: "object.item.epgItem.videoProgram",
_mp_const.MEDIA_TYPE_GENRE: "object.container.genre", MediaType.GENRE: "object.container.genre",
_mp_const.MEDIA_TYPE_IMAGE: "object.item.imageItem", MediaType.IMAGE: "object.item.imageItem",
_mp_const.MEDIA_TYPE_MOVIE: "object.item.videoItem.movie", MediaType.MOVIE: "object.item.videoItem.movie",
_mp_const.MEDIA_TYPE_MUSIC: "object.item.audioItem.musicTrack", MediaType.MUSIC: "object.item.audioItem.musicTrack",
_mp_const.MEDIA_TYPE_PLAYLIST: "object.item.playlistItem", MediaType.PLAYLIST: "object.item.playlistItem",
_mp_const.MEDIA_TYPE_PODCAST: "object.item.audioItem.audioBook", MediaType.PODCAST: "object.item.audioItem.audioBook",
_mp_const.MEDIA_TYPE_SEASON: "object.item.epgItem.videoProgram", MediaType.SEASON: "object.item.epgItem.videoProgram",
_mp_const.MEDIA_TYPE_TRACK: "object.item.audioItem.musicTrack", MediaType.TRACK: "object.item.audioItem.musicTrack",
_mp_const.MEDIA_TYPE_TVSHOW: "object.item.videoItem.videoBroadcast", MediaType.TVSHOW: "object.item.videoItem.videoBroadcast",
_mp_const.MEDIA_TYPE_URL: "object.item.bookmarkItem", MediaType.URL: "object.item.bookmarkItem",
_mp_const.MEDIA_TYPE_VIDEO: "object.item.videoItem", MediaType.VIDEO: "object.item.videoItem",
} }
# Translation of MediaMetadata keys to DIDL-Lite keys. # Translation of MediaMetadata keys to DIDL-Lite keys.
@ -109,32 +109,32 @@ MEDIA_METADATA_DIDL: Mapping[str, str] = {
# of play modes in order of suitability. Fall back to _PlayMode.NORMAL in any # of play modes in order of suitability. Fall back to _PlayMode.NORMAL in any
# case. NOTE: This list is slightly different to that in SHUFFLE_PLAY_MODES, # case. NOTE: This list is slightly different to that in SHUFFLE_PLAY_MODES,
# due to fallback behaviour when turning on repeat modes. # due to fallback behaviour when turning on repeat modes.
REPEAT_PLAY_MODES: Mapping[tuple[bool, str], list[_PlayMode]] = { REPEAT_PLAY_MODES: Mapping[tuple[bool, RepeatMode], list[_PlayMode]] = {
(False, _mp_const.REPEAT_MODE_OFF): [ (False, RepeatMode.OFF): [
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(False, _mp_const.REPEAT_MODE_ONE): [ (False, RepeatMode.ONE): [
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(False, _mp_const.REPEAT_MODE_ALL): [ (False, RepeatMode.ALL): [
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_OFF): [ (True, RepeatMode.OFF): [
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_ONE): [ (True, RepeatMode.ONE): [
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_ALL): [ (True, RepeatMode.ALL): [
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
@ -146,31 +146,31 @@ REPEAT_PLAY_MODES: Mapping[tuple[bool, str], list[_PlayMode]] = {
# of play modes in order of suitability. Fall back to _PlayMode.NORMAL in any # of play modes in order of suitability. Fall back to _PlayMode.NORMAL in any
# case. # case.
SHUFFLE_PLAY_MODES: Mapping[tuple[bool, str], list[_PlayMode]] = { SHUFFLE_PLAY_MODES: Mapping[tuple[bool, str], list[_PlayMode]] = {
(False, _mp_const.REPEAT_MODE_OFF): [ (False, RepeatMode.OFF): [
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(False, _mp_const.REPEAT_MODE_ONE): [ (False, RepeatMode.ONE): [
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(False, _mp_const.REPEAT_MODE_ALL): [ (False, RepeatMode.ALL): [
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_OFF): [ (True, RepeatMode.OFF): [
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_ONE): [ (True, RepeatMode.ONE): [
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
_PlayMode.REPEAT_ONE, _PlayMode.REPEAT_ONE,
_PlayMode.NORMAL, _PlayMode.NORMAL,
], ],
(True, _mp_const.REPEAT_MODE_ALL): [ (True, RepeatMode.ALL): [
_PlayMode.RANDOM, _PlayMode.RANDOM,
_PlayMode.SHUFFLE, _PlayMode.SHUFFLE,
_PlayMode.REPEAT_ALL, _PlayMode.REPEAT_ALL,

49
homeassistant/components/dlna_dmr/media_player.py
View File

@ -19,27 +19,16 @@ from typing_extensions import Concatenate, ParamSpec
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import media_source, ssdp from homeassistant.components import media_source, ssdp
from homeassistant.components.media_player import ( from homeassistant.components.media_player import (
ATTR_MEDIA_EXTRA,
BrowseMedia, BrowseMedia,
MediaPlayerEntity, MediaPlayerEntity,
MediaPlayerEntityFeature, MediaPlayerEntityFeature,
MediaPlayerState,
MediaType,
RepeatMode,
async_process_play_media_url, async_process_play_media_url,
) )
from homeassistant.components.media_player.const import ( from homeassistant.const import CONF_DEVICE_ID, CONF_TYPE, CONF_URL
ATTR_MEDIA_EXTRA,
REPEAT_MODE_ALL,
REPEAT_MODE_OFF,
REPEAT_MODE_ONE,
)
from homeassistant.const import (
CONF_DEVICE_ID,
CONF_TYPE,
CONF_URL,
STATE_IDLE,
STATE_OFF,
STATE_ON,
STATE_PAUSED,
STATE_PLAYING,
)
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry, entity_registry from homeassistant.helpers import device_registry, entity_registry
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -466,27 +455,27 @@ class DlnaDmrEntity(MediaPlayerEntity):
return f"{self.udn}::{self.device_type}" return f"{self.udn}::{self.device_type}"
@property @property
def state(self) -> str | None: def state(self) -> MediaPlayerState | None:
"""State of the player.""" """State of the player."""
if not self._device or not self.available: if not self._device or not self.available:
return STATE_OFF return MediaPlayerState.OFF
if self._device.transport_state is None: if self._device.transport_state is None:
return STATE_ON return MediaPlayerState.ON
if self._device.transport_state in ( if self._device.transport_state in (
TransportState.PLAYING, TransportState.PLAYING,
TransportState.TRANSITIONING, TransportState.TRANSITIONING,
): ):
return STATE_PLAYING return MediaPlayerState.PLAYING
if self._device.transport_state in ( if self._device.transport_state in (
TransportState.PAUSED_PLAYBACK, TransportState.PAUSED_PLAYBACK,
TransportState.PAUSED_RECORDING, TransportState.PAUSED_RECORDING,
): ):
return STATE_PAUSED return MediaPlayerState.PAUSED
if self._device.transport_state == TransportState.VENDOR_DEFINED: if self._device.transport_state == TransportState.VENDOR_DEFINED:
# Unable to map this state to anything reasonable, so it's "Unknown" # Unable to map this state to anything reasonable, so it's "Unknown"
return None return None
return STATE_IDLE return MediaPlayerState.IDLE
@property @property
def supported_features(self) -> int: def supported_features(self) -> int:
@ -586,7 +575,7 @@ class DlnaDmrEntity(MediaPlayerEntity):
@catch_request_errors @catch_request_errors
async def async_play_media( async def async_play_media(
self, media_type: str, media_id: str, **kwargs: Any self, media_type: MediaType | str, media_id: str, **kwargs: Any
) -> None: ) -> None:
"""Play a piece of media.""" """Play a piece of media."""
_LOGGER.debug("Playing media: %s, %s, %s", media_type, media_id, kwargs) _LOGGER.debug("Playing media: %s, %s, %s", media_type, media_id, kwargs)
@ -683,7 +672,7 @@ class DlnaDmrEntity(MediaPlayerEntity):
"""Enable/disable shuffle mode.""" """Enable/disable shuffle mode."""
assert self._device is not None assert self._device is not None
repeat = self.repeat or REPEAT_MODE_OFF repeat = self.repeat or RepeatMode.OFF
potential_play_modes = SHUFFLE_PLAY_MODES[(shuffle, repeat)] potential_play_modes = SHUFFLE_PLAY_MODES[(shuffle, repeat)]
valid_play_modes = self._device.valid_play_modes valid_play_modes = self._device.valid_play_modes
@ -698,7 +687,7 @@ class DlnaDmrEntity(MediaPlayerEntity):
) )
@property @property
def repeat(self) -> str | None: def repeat(self) -> RepeatMode | None:
"""Return current repeat mode.""" """Return current repeat mode."""
if not self._device: if not self._device:
return None return None
@ -710,15 +699,15 @@ class DlnaDmrEntity(MediaPlayerEntity):
return None return None
if play_mode == PlayMode.REPEAT_ONE: if play_mode == PlayMode.REPEAT_ONE:
return REPEAT_MODE_ONE return RepeatMode.ONE
if play_mode in (PlayMode.REPEAT_ALL, PlayMode.RANDOM): if play_mode in (PlayMode.REPEAT_ALL, PlayMode.RANDOM):
return REPEAT_MODE_ALL return RepeatMode.ALL
return REPEAT_MODE_OFF return RepeatMode.OFF
@catch_request_errors @catch_request_errors
async def async_set_repeat(self, repeat: str) -> None: async def async_set_repeat(self, repeat: RepeatMode) -> None:
"""Set repeat mode.""" """Set repeat mode."""
assert self._device is not None assert self._device is not None
@ -839,7 +828,7 @@ class DlnaDmrEntity(MediaPlayerEntity):
return self._device.current_track_uri return self._device.current_track_uri
@property @property
def media_content_type(self) -> str | None: def media_content_type(self) -> MediaType | None:
"""Content type of current playing media.""" """Content type of current playing media."""
if not self._device or not self._device.media_class: if not self._device or not self._device.media_class:
return None return None

173
tests/components/dlna_dmr/test_media_player.py
View File

@ -21,7 +21,6 @@ import pytest
from homeassistant import const as ha_const from homeassistant import const as ha_const
from homeassistant.components import ssdp from homeassistant.components import ssdp
from homeassistant.components.dlna_dmr import media_player
from homeassistant.components.dlna_dmr.const import ( from homeassistant.components.dlna_dmr.const import (
CONF_BROWSE_UNFILTERED, CONF_BROWSE_UNFILTERED,
CONF_CALLBACK_URL_OVERRIDE, CONF_CALLBACK_URL_OVERRIDE,
@ -30,10 +29,17 @@ from homeassistant.components.dlna_dmr.const import (
DOMAIN as DLNA_DOMAIN, DOMAIN as DLNA_DOMAIN,
) )
from homeassistant.components.dlna_dmr.data import EventListenAddr from homeassistant.components.dlna_dmr.data import EventListenAddr
from homeassistant.components.media_player import ATTR_TO_PROPERTY, const as mp_const from homeassistant.components.dlna_dmr.media_player import DlnaDmrEntity
from homeassistant.components.media_player.const import DOMAIN as MP_DOMAIN from homeassistant.components.media_player import (
from homeassistant.components.media_source.const import DOMAIN as MS_DOMAIN ATTR_TO_PROPERTY,
from homeassistant.components.media_source.models import PlayMedia DOMAIN as MP_DOMAIN,
MediaPlayerEntityFeature,
MediaPlayerState,
MediaType,
RepeatMode,
const as mp_const,
)
from homeassistant.components.media_source import DOMAIN as MS_DOMAIN, PlayMedia
from homeassistant.const import ATTR_ENTITY_ID from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import async_get as async_get_dr from homeassistant.helpers.device_registry import async_get as async_get_dr
@ -223,7 +229,7 @@ async def test_setup_entry_no_options(
ANY, {"_udn": MOCK_DEVICE_UDN, "NTS": "ssdp:byebye"} ANY, {"_udn": MOCK_DEVICE_UDN, "NTS": "ssdp:byebye"}
) )
# Quick check of the state to verify the entity has a connected DmrDevice # Quick check of the state to verify the entity has a connected DmrDevice
assert mock_state.state == media_player.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Check the name matches that supplied # Check the name matches that supplied
assert mock_state.name == MOCK_DEVICE_NAME assert mock_state.name == MOCK_DEVICE_NAME
@ -292,7 +298,7 @@ async def test_setup_entry_with_options(
ANY, {"_udn": MOCK_DEVICE_UDN, "NTS": "ssdp:byebye"} ANY, {"_udn": MOCK_DEVICE_UDN, "NTS": "ssdp:byebye"}
) )
# Quick check of the state to verify the entity has a connected DmrDevice # Quick check of the state to verify the entity has a connected DmrDevice
assert mock_state.state == media_player.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Check the name matches that supplied # Check the name matches that supplied
assert mock_state.name == MOCK_DEVICE_NAME assert mock_state.name == MOCK_DEVICE_NAME
@ -359,7 +365,7 @@ async def test_event_subscribe_rejected(
assert mock_state is not None assert mock_state is not None
# Device should be connected # Device should be connected
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Device should not be unsubscribed # Device should not be unsubscribed
dmr_device_mock.async_unsubscribe_services.assert_not_awaited() dmr_device_mock.async_unsubscribe_services.assert_not_awaited()
@ -385,14 +391,14 @@ async def test_available_device(
# Check entity state gets updated when device changes state # Check entity state gets updated when device changes state
for (dev_state, ent_state) in [ for (dev_state, ent_state) in [
(None, ha_const.STATE_ON), (None, MediaPlayerState.ON),
(TransportState.STOPPED, ha_const.STATE_IDLE), (TransportState.STOPPED, MediaPlayerState.IDLE),
(TransportState.PLAYING, ha_const.STATE_PLAYING), (TransportState.PLAYING, MediaPlayerState.PLAYING),
(TransportState.TRANSITIONING, ha_const.STATE_PLAYING), (TransportState.TRANSITIONING, MediaPlayerState.PLAYING),
(TransportState.PAUSED_PLAYBACK, ha_const.STATE_PAUSED), (TransportState.PAUSED_PLAYBACK, MediaPlayerState.PAUSED),
(TransportState.PAUSED_RECORDING, ha_const.STATE_PAUSED), (TransportState.PAUSED_RECORDING, MediaPlayerState.PAUSED),
(TransportState.RECORDING, ha_const.STATE_IDLE), (TransportState.RECORDING, MediaPlayerState.IDLE),
(TransportState.NO_MEDIA_PRESENT, ha_const.STATE_IDLE), (TransportState.NO_MEDIA_PRESENT, MediaPlayerState.IDLE),
(TransportState.VENDOR_DEFINED, ha_const.STATE_UNKNOWN), (TransportState.VENDOR_DEFINED, ha_const.STATE_UNKNOWN),
]: ]:
dmr_device_mock.profile_device.available = True dmr_device_mock.profile_device.available = True
@ -416,16 +422,19 @@ async def test_feature_flags(
"""Test feature flags of a connected DlnaDmrEntity.""" """Test feature flags of a connected DlnaDmrEntity."""
# Check supported feature flags, one at a time. # Check supported feature flags, one at a time.
FEATURE_FLAGS: list[tuple[str, int]] = [ FEATURE_FLAGS: list[tuple[str, int]] = [
("has_volume_level", mp_const.SUPPORT_VOLUME_SET), ("has_volume_level", MediaPlayerEntityFeature.VOLUME_SET),
("has_volume_mute", mp_const.SUPPORT_VOLUME_MUTE), ("has_volume_mute", MediaPlayerEntityFeature.VOLUME_MUTE),
("can_play", mp_const.SUPPORT_PLAY), ("can_play", MediaPlayerEntityFeature.PLAY),
("can_pause", mp_const.SUPPORT_PAUSE), ("can_pause", MediaPlayerEntityFeature.PAUSE),
("can_stop", mp_const.SUPPORT_STOP), ("can_stop", MediaPlayerEntityFeature.STOP),
("can_previous", mp_const.SUPPORT_PREVIOUS_TRACK), ("can_previous", MediaPlayerEntityFeature.PREVIOUS_TRACK),
("can_next", mp_const.SUPPORT_NEXT_TRACK), ("can_next", MediaPlayerEntityFeature.NEXT_TRACK),
("has_play_media", mp_const.SUPPORT_PLAY_MEDIA | mp_const.SUPPORT_BROWSE_MEDIA), (
("can_seek_rel_time", mp_const.SUPPORT_SEEK), "has_play_media",
("has_presets", mp_const.SUPPORT_SELECT_SOUND_MODE), MediaPlayerEntityFeature.PLAY_MEDIA | MediaPlayerEntityFeature.BROWSE_MEDIA,
),
("can_seek_rel_time", MediaPlayerEntityFeature.SEEK),
("has_presets", MediaPlayerEntityFeature.SELECT_SOUND_MODE),
] ]
# Clear all feature properties # Clear all feature properties
@ -446,10 +455,10 @@ async def test_feature_flags(
# shuffle and repeat features depend on the available play modes # shuffle and repeat features depend on the available play modes
PLAY_MODE_FEATURE_FLAGS: list[tuple[PlayMode, int]] = [ PLAY_MODE_FEATURE_FLAGS: list[tuple[PlayMode, int]] = [
(PlayMode.NORMAL, 0), (PlayMode.NORMAL, 0),
(PlayMode.SHUFFLE, mp_const.SUPPORT_SHUFFLE_SET), (PlayMode.SHUFFLE, MediaPlayerEntityFeature.SHUFFLE_SET),
(PlayMode.REPEAT_ONE, mp_const.SUPPORT_REPEAT_SET), (PlayMode.REPEAT_ONE, MediaPlayerEntityFeature.REPEAT_SET),
(PlayMode.REPEAT_ALL, mp_const.SUPPORT_REPEAT_SET), (PlayMode.REPEAT_ALL, MediaPlayerEntityFeature.REPEAT_SET),
(PlayMode.RANDOM, mp_const.SUPPORT_SHUFFLE_SET), (PlayMode.RANDOM, MediaPlayerEntityFeature.SHUFFLE_SET),
(PlayMode.DIRECT_1, 0), (PlayMode.DIRECT_1, 0),
(PlayMode.INTRO, 0), (PlayMode.INTRO, 0),
(PlayMode.VENDOR_DEFINED, 0), (PlayMode.VENDOR_DEFINED, 0),
@ -497,13 +506,13 @@ async def test_attributes(
# media_content_type is mapped from UPnP class to MediaPlayer type # media_content_type is mapped from UPnP class to MediaPlayer type
dmr_device_mock.media_class = "object.item.audioItem.musicTrack" dmr_device_mock.media_class = "object.item.audioItem.musicTrack"
attrs = await get_attrs(hass, mock_entity_id) attrs = await get_attrs(hass, mock_entity_id)
assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == mp_const.MEDIA_TYPE_MUSIC assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == MediaType.MUSIC
dmr_device_mock.media_class = "object.item.videoItem.movie" dmr_device_mock.media_class = "object.item.videoItem.movie"
attrs = await get_attrs(hass, mock_entity_id) attrs = await get_attrs(hass, mock_entity_id)
assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == mp_const.MEDIA_TYPE_MOVIE assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == MediaType.MOVIE
dmr_device_mock.media_class = "object.item.videoItem.videoBroadcast" dmr_device_mock.media_class = "object.item.videoItem.videoBroadcast"
attrs = await get_attrs(hass, mock_entity_id) attrs = await get_attrs(hass, mock_entity_id)
assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == mp_const.MEDIA_TYPE_TVSHOW assert attrs[mp_const.ATTR_MEDIA_CONTENT_TYPE] == MediaType.TVSHOW
# media_season & media_episode have a special case # media_season & media_episode have a special case
dmr_device_mock.media_season_number = "0" dmr_device_mock.media_season_number = "0"
@ -519,13 +528,13 @@ async def test_attributes(
# shuffle and repeat is based on device's play mode # shuffle and repeat is based on device's play mode
for play_mode, shuffle, repeat in [ for play_mode, shuffle, repeat in [
(PlayMode.NORMAL, False, mp_const.REPEAT_MODE_OFF), (PlayMode.NORMAL, False, RepeatMode.OFF),
(PlayMode.SHUFFLE, True, mp_const.REPEAT_MODE_OFF), (PlayMode.SHUFFLE, True, RepeatMode.OFF),
(PlayMode.REPEAT_ONE, False, mp_const.REPEAT_MODE_ONE), (PlayMode.REPEAT_ONE, False, RepeatMode.ONE),
(PlayMode.REPEAT_ALL, False, mp_const.REPEAT_MODE_ALL), (PlayMode.REPEAT_ALL, False, RepeatMode.ALL),
(PlayMode.RANDOM, True, mp_const.REPEAT_MODE_ALL), (PlayMode.RANDOM, True, RepeatMode.ALL),
(PlayMode.DIRECT_1, False, mp_const.REPEAT_MODE_OFF), (PlayMode.DIRECT_1, False, RepeatMode.OFF),
(PlayMode.INTRO, False, mp_const.REPEAT_MODE_OFF), (PlayMode.INTRO, False, RepeatMode.OFF),
]: ]:
dmr_device_mock.play_mode = play_mode dmr_device_mock.play_mode = play_mode
attrs = await get_attrs(hass, mock_entity_id) attrs = await get_attrs(hass, mock_entity_id)
@ -620,7 +629,7 @@ async def test_play_media_stopped(
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_MUSIC, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
}, },
@ -652,7 +661,7 @@ async def test_play_media_playing(
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_MUSIC, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
}, },
@ -685,7 +694,7 @@ async def test_play_media_no_autoplay(
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_MUSIC, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
mp_const.ATTR_MEDIA_EXTRA: {"autoplay": False}, mp_const.ATTR_MEDIA_EXTRA: {"autoplay": False},
@ -716,7 +725,7 @@ async def test_play_media_metadata(
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_MUSIC, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
mp_const.ATTR_MEDIA_EXTRA: { mp_const.ATTR_MEDIA_EXTRA: {
@ -746,7 +755,7 @@ async def test_play_media_metadata(
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_TVSHOW, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.TVSHOW,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/123.mkv", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/123.mkv",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
mp_const.ATTR_MEDIA_EXTRA: { mp_const.ATTR_MEDIA_EXTRA: {
@ -878,21 +887,21 @@ async def test_shuffle_repeat_modes(
# Test repeat with all variations of existing play mode # Test repeat with all variations of existing play mode
for init_mode, repeat_set, expect_mode in [ for init_mode, repeat_set, expect_mode in [
(PlayMode.NORMAL, mp_const.REPEAT_MODE_OFF, PlayMode.NORMAL), (PlayMode.NORMAL, RepeatMode.OFF, PlayMode.NORMAL),
(PlayMode.SHUFFLE, mp_const.REPEAT_MODE_OFF, PlayMode.SHUFFLE), (PlayMode.SHUFFLE, RepeatMode.OFF, PlayMode.SHUFFLE),
(PlayMode.REPEAT_ONE, mp_const.REPEAT_MODE_OFF, PlayMode.NORMAL), (PlayMode.REPEAT_ONE, RepeatMode.OFF, PlayMode.NORMAL),
(PlayMode.REPEAT_ALL, mp_const.REPEAT_MODE_OFF, PlayMode.NORMAL), (PlayMode.REPEAT_ALL, RepeatMode.OFF, PlayMode.NORMAL),
(PlayMode.RANDOM, mp_const.REPEAT_MODE_OFF, PlayMode.SHUFFLE), (PlayMode.RANDOM, RepeatMode.OFF, PlayMode.SHUFFLE),
(PlayMode.NORMAL, mp_const.REPEAT_MODE_ONE, PlayMode.REPEAT_ONE), (PlayMode.NORMAL, RepeatMode.ONE, PlayMode.REPEAT_ONE),
(PlayMode.SHUFFLE, mp_const.REPEAT_MODE_ONE, PlayMode.REPEAT_ONE), (PlayMode.SHUFFLE, RepeatMode.ONE, PlayMode.REPEAT_ONE),
(PlayMode.REPEAT_ONE, mp_const.REPEAT_MODE_ONE, PlayMode.REPEAT_ONE), (PlayMode.REPEAT_ONE, RepeatMode.ONE, PlayMode.REPEAT_ONE),
(PlayMode.REPEAT_ALL, mp_const.REPEAT_MODE_ONE, PlayMode.REPEAT_ONE), (PlayMode.REPEAT_ALL, RepeatMode.ONE, PlayMode.REPEAT_ONE),
(PlayMode.RANDOM, mp_const.REPEAT_MODE_ONE, PlayMode.REPEAT_ONE), (PlayMode.RANDOM, RepeatMode.ONE, PlayMode.REPEAT_ONE),
(PlayMode.NORMAL, mp_const.REPEAT_MODE_ALL, PlayMode.REPEAT_ALL), (PlayMode.NORMAL, RepeatMode.ALL, PlayMode.REPEAT_ALL),
(PlayMode.SHUFFLE, mp_const.REPEAT_MODE_ALL, PlayMode.RANDOM), (PlayMode.SHUFFLE, RepeatMode.ALL, PlayMode.RANDOM),
(PlayMode.REPEAT_ONE, mp_const.REPEAT_MODE_ALL, PlayMode.REPEAT_ALL), (PlayMode.REPEAT_ONE, RepeatMode.ALL, PlayMode.REPEAT_ALL),
(PlayMode.REPEAT_ALL, mp_const.REPEAT_MODE_ALL, PlayMode.REPEAT_ALL), (PlayMode.REPEAT_ALL, RepeatMode.ALL, PlayMode.REPEAT_ALL),
(PlayMode.RANDOM, mp_const.REPEAT_MODE_ALL, PlayMode.RANDOM), (PlayMode.RANDOM, RepeatMode.ALL, PlayMode.RANDOM),
]: ]:
dmr_device_mock.play_mode = init_mode dmr_device_mock.play_mode = init_mode
await hass.services.async_call( await hass.services.async_call(
@ -926,7 +935,7 @@ async def test_shuffle_repeat_modes(
ha_const.SERVICE_REPEAT_SET, ha_const.SERVICE_REPEAT_SET,
{ {
ATTR_ENTITY_ID: mock_entity_id, ATTR_ENTITY_ID: mock_entity_id,
mp_const.ATTR_MEDIA_REPEAT: mp_const.REPEAT_MODE_OFF, mp_const.ATTR_MEDIA_REPEAT: RepeatMode.OFF,
}, },
blocking=True, blocking=True,
) )
@ -1226,7 +1235,7 @@ async def test_unavailable_device(
( (
mp_const.SERVICE_PLAY_MEDIA, mp_const.SERVICE_PLAY_MEDIA,
{ {
mp_const.ATTR_MEDIA_CONTENT_TYPE: mp_const.MEDIA_TYPE_MUSIC, mp_const.ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3", mp_const.ATTR_MEDIA_CONTENT_ID: "http://192.88.99.20:8200/MediaItems/17621.mp3",
mp_const.ATTR_MEDIA_ENQUEUE: False, mp_const.ATTR_MEDIA_ENQUEUE: False,
}, },
@ -1318,7 +1327,7 @@ async def test_become_available(
# Quick check of the state to verify the entity has a connected DmrDevice # Quick check of the state to verify the entity has a connected DmrDevice
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Check hass device information is now filled in # Check hass device information is now filled in
dev_reg = async_get_dr(hass) dev_reg = async_get_dr(hass)
device = dev_reg.async_get_device(identifiers={(DLNA_DOMAIN, MOCK_DEVICE_UDN)}) device = dev_reg.async_get_device(identifiers={(DLNA_DOMAIN, MOCK_DEVICE_UDN)})
@ -1496,7 +1505,7 @@ async def test_multiple_ssdp_alive(
# Device should be available # Device should be available
mock_state = hass.states.get(mock_disconnected_entity_id) mock_state = hass.states.get(mock_disconnected_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == media_player.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
async def test_ssdp_byebye( async def test_ssdp_byebye(
@ -1592,7 +1601,7 @@ async def test_ssdp_update_seen_bootid(
# Device was not reconnected, even with a new boot ID # Device was not reconnected, even with a new boot ID
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 0 assert dmr_device_mock.async_unsubscribe_services.await_count == 0
assert dmr_device_mock.async_subscribe_services.await_count == 1 assert dmr_device_mock.async_subscribe_services.await_count == 1
@ -1617,7 +1626,7 @@ async def test_ssdp_update_seen_bootid(
# Nothing should change # Nothing should change
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 0 assert dmr_device_mock.async_unsubscribe_services.await_count == 0
assert dmr_device_mock.async_subscribe_services.await_count == 1 assert dmr_device_mock.async_subscribe_services.await_count == 1
@ -1642,7 +1651,7 @@ async def test_ssdp_update_seen_bootid(
# Nothing should change # Nothing should change
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 0 assert dmr_device_mock.async_unsubscribe_services.await_count == 0
assert dmr_device_mock.async_subscribe_services.await_count == 1 assert dmr_device_mock.async_subscribe_services.await_count == 1
@ -1662,7 +1671,7 @@ async def test_ssdp_update_seen_bootid(
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 0 assert dmr_device_mock.async_unsubscribe_services.await_count == 0
assert dmr_device_mock.async_subscribe_services.await_count == 1 assert dmr_device_mock.async_subscribe_services.await_count == 1
@ -1719,7 +1728,7 @@ async def test_ssdp_update_missed_bootid(
# Device should not reconnect yet # Device should not reconnect yet
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 0 assert dmr_device_mock.async_unsubscribe_services.await_count == 0
assert dmr_device_mock.async_subscribe_services.await_count == 1 assert dmr_device_mock.async_subscribe_services.await_count == 1
@ -1739,7 +1748,7 @@ async def test_ssdp_update_missed_bootid(
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_unsubscribe_services.await_count == 1 assert dmr_device_mock.async_unsubscribe_services.await_count == 1
assert dmr_device_mock.async_subscribe_services.await_count == 2 assert dmr_device_mock.async_subscribe_services.await_count == 2
@ -1778,7 +1787,7 @@ async def test_ssdp_bootid(
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_subscribe_services.call_count == 1 assert dmr_device_mock.async_subscribe_services.call_count == 1
assert dmr_device_mock.async_unsubscribe_services.call_count == 0 assert dmr_device_mock.async_unsubscribe_services.call_count == 0
@ -1798,7 +1807,7 @@ async def test_ssdp_bootid(
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_subscribe_services.call_count == 1 assert dmr_device_mock.async_subscribe_services.call_count == 1
assert dmr_device_mock.async_unsubscribe_services.call_count == 0 assert dmr_device_mock.async_unsubscribe_services.call_count == 0
@ -1818,7 +1827,7 @@ async def test_ssdp_bootid(
mock_state = hass.states.get(entity_id) mock_state = hass.states.get(entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
assert dmr_device_mock.async_subscribe_services.call_count == 2 assert dmr_device_mock.async_subscribe_services.call_count == 2
assert dmr_device_mock.async_unsubscribe_services.call_count == 1 assert dmr_device_mock.async_unsubscribe_services.call_count == 1
@ -1849,14 +1858,14 @@ async def test_become_unavailable(
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# With a working connection, the state should be restored # With a working connection, the state should be restored
await async_update_entity(hass, mock_entity_id) await async_update_entity(hass, mock_entity_id)
dmr_device_mock.async_update.assert_any_call(do_ping=True) dmr_device_mock.async_update.assert_any_call(do_ping=True)
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Break the service again, and the connection too. An update will cause the # Break the service again, and the connection too. An update will cause the
# device to be disconnected # device to be disconnected
@ -1918,7 +1927,7 @@ async def test_poll_availability(
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
# Clean up # Clean up
assert await hass.config_entries.async_remove(config_entry_mock.entry_id) == { assert await hass.config_entries.async_remove(config_entry_mock.entry_id) == {
@ -1938,9 +1947,7 @@ async def test_disappearing_device(
directly to skip the availability check. directly to skip the availability check.
""" """
# Retrieve entity directly. # Retrieve entity directly.
entity: media_player.DlnaDmrEntity = hass.data[MP_DOMAIN].get_entity( entity: DlnaDmrEntity = hass.data[MP_DOMAIN].get_entity(mock_disconnected_entity_id)
mock_disconnected_entity_id
)
# Test attribute access # Test attribute access
for attr in ATTR_TO_PROPERTY: for attr in ATTR_TO_PROPERTY:
@ -1964,7 +1971,7 @@ async def test_disappearing_device(
await entity.async_media_previous_track() await entity.async_media_previous_track()
await entity.async_media_next_track() await entity.async_media_next_track()
await entity.async_set_shuffle(True) await entity.async_set_shuffle(True)
await entity.async_set_repeat(mp_const.REPEAT_MODE_ALL) await entity.async_set_repeat(RepeatMode.ALL)
await entity.async_select_sound_mode("Default") await entity.async_select_sound_mode("Default")
@ -2023,7 +2030,7 @@ async def test_config_update_listen_port(
# Check that its still connected # Check that its still connected
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
async def test_config_update_connect_failure( async def test_config_update_connect_failure(
@ -2097,7 +2104,7 @@ async def test_config_update_callback_url(
# Check that its still connected # Check that its still connected
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE
async def test_config_update_poll_availability( async def test_config_update_poll_availability(
@ -2138,4 +2145,4 @@ async def test_config_update_poll_availability(
# Check that its still connected # Check that its still connected
mock_state = hass.states.get(mock_entity_id) mock_state = hass.states.get(mock_entity_id)
assert mock_state is not None assert mock_state is not None
assert mock_state.state == ha_const.STATE_IDLE assert mock_state.state == MediaPlayerState.IDLE