diff --git a/tests/components/media_player/test_async_helpers.py b/tests/components/media_player/test_async_helpers.py index a24c9cc76b2..f3b70187f33 100644 --- a/tests/components/media_player/test_async_helpers.py +++ b/tests/components/media_player/test_async_helpers.py @@ -13,86 +13,6 @@ from homeassistant.const import ( from homeassistant.core import HomeAssistant -class ExtendedMediaPlayer(mp.MediaPlayerEntity): - """Media player test class.""" - - def __init__(self, hass): - """Initialize the test media player.""" - self.hass = hass - self._volume = 0 - self._state = STATE_OFF - - @property - def state(self): - """State of the player.""" - return self._state - - @property - def volume_level(self): - """Volume level of the media player (0..1).""" - return self._volume - - @property - def supported_features(self): - """Flag media player features that are supported.""" - return ( - mp.const.MediaPlayerEntityFeature.VOLUME_SET - | mp.const.MediaPlayerEntityFeature.VOLUME_STEP - | mp.const.MediaPlayerEntityFeature.PLAY - | mp.const.MediaPlayerEntityFeature.PAUSE - | mp.const.MediaPlayerEntityFeature.TURN_OFF - | mp.const.MediaPlayerEntityFeature.TURN_ON - ) - - def set_volume_level(self, volume): - """Set volume level, range 0..1.""" - self._volume = volume - - def volume_up(self): - """Turn volume up for media player.""" - if self.volume_level < 1: - self.set_volume_level(min(1, self.volume_level + 0.1)) - - def volume_down(self): - """Turn volume down for media player.""" - if self.volume_level > 0: - self.set_volume_level(max(0, self.volume_level - 0.1)) - - def media_play(self): - """Play the media player.""" - self._state = STATE_PLAYING - - def media_pause(self): - """Plause the media player.""" - self._state = STATE_PAUSED - - def media_play_pause(self): - """Play or pause the media player.""" - if self._state == STATE_PLAYING: - self._state = STATE_PAUSED - else: - self._state = STATE_PLAYING - - def turn_on(self): - """Turn on state.""" - self._state = STATE_ON - - def turn_off(self): - """Turn off state.""" - self._state = STATE_OFF - - def standby(self): - """Put device in standby.""" - self._state = STATE_STANDBY - - def toggle(self): - """Toggle the power on the media player.""" - if self._state in [STATE_OFF, STATE_IDLE, STATE_STANDBY]: - self._state = STATE_ON - else: - self._state = STATE_OFF - - class SimpleMediaPlayer(mp.MediaPlayerEntity): """Media player test class.""" @@ -149,6 +69,34 @@ class SimpleMediaPlayer(mp.MediaPlayerEntity): self._state = STATE_STANDBY +class ExtendedMediaPlayer(SimpleMediaPlayer): + """Media player test class.""" + + def volume_up(self): + """Turn volume up for media player.""" + if self.volume_level < 1: + self.set_volume_level(min(1, self.volume_level + 0.1)) + + def volume_down(self): + """Turn volume down for media player.""" + if self.volume_level > 0: + self.set_volume_level(max(0, self.volume_level - 0.1)) + + def media_play_pause(self): + """Play or pause the media player.""" + if self._state == STATE_PLAYING: + self._state = STATE_PAUSED + else: + self._state = STATE_PLAYING + + def toggle(self): + """Toggle the power on the media player.""" + if self._state in [STATE_OFF, STATE_IDLE, STATE_STANDBY]: + self._state = STATE_ON + else: + self._state = STATE_OFF + + class AttrMediaPlayer(SimpleMediaPlayer): """Media player setting properties via _attr_*."""