Refactor media_player test classes (#105630)

tests/components/media_player/test_async_helpers.py

@@ -13,7 +13,7 @@ from homeassistant.const import (
 from homeassistant.core import HomeAssistant
 
 
-class ExtendedMediaPlayer(mp.MediaPlayerEntity):
+class SimpleMediaPlayer(mp.MediaPlayerEntity):
     """Media player test class."""
 
     def __init__(self, hass):
@@ -48,16 +48,6 @@ class ExtendedMediaPlayer(mp.MediaPlayerEntity):
         """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
@@ -66,13 +56,6 @@ class ExtendedMediaPlayer(mp.MediaPlayerEntity):
         """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
@@ -85,68 +68,33 @@ class ExtendedMediaPlayer(mp.MediaPlayerEntity):
         """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):
+class ExtendedMediaPlayer(SimpleMediaPlayer):
     """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 media_play(self):
-        """Play the media player."""
-        self._state = STATE_PLAYING
-
-    def media_pause(self):
-        """Plause the media player."""
-        self._state = STATE_PAUSED
+    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 turn_on(self):
-        """Turn on state."""
-        self._state = STATE_ON
+    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 turn_off(self):
-        """Turn off state."""
-        self._state = STATE_OFF
+    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 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 AttrMediaPlayer(SimpleMediaPlayer):