Correctly map repeat mode in Music Assistant (#155777)

Author: OzGav

homeassistant/components/music_assistant/media_player.py

@@ -115,6 +115,13 @@
     MediaPlayerEnqueue.REPLACE: QueueOption.REPLACE,
 }
 
+REPEAT_MODE_MAPPING_TO_HA = {
+    MassRepeatMode.OFF: RepeatMode.OFF,
+    MassRepeatMode.ONE: RepeatMode.ONE,
+    MassRepeatMode.ALL: RepeatMode.ALL,
+    # UNKNOWN is intentionally not mapped - will return None
+}
+
 SERVICE_PLAY_MEDIA_ADVANCED = "play_media"
 SERVICE_PLAY_ANNOUNCEMENT = "play_announcement"
 SERVICE_TRANSFER_QUEUE = "transfer_queue"
@@ -657,7 +664,7 @@ def _update_media_attributes(
         # player has an MA queue active (either its own queue or some group queue)
         self._attr_app_id = DOMAIN
         self._attr_shuffle = queue.shuffle_enabled
-        self._attr_repeat = queue.repeat_mode.value
+        self._attr_repeat = REPEAT_MODE_MAPPING_TO_HA.get(queue.repeat_mode)
         if not (cur_item := queue.current_item):
             # queue is empty
             return

tests/components/music_assistant/fixtures/player_queues.json

@@ -134,7 +134,7 @@
             "links": null,
             "chapters": null,
             "performers": null,
-            "preview": "https://p.scdn.co/mp3-preview/98deb9c370bbaa350be058b3470fbe3bc1e28d9d?cid=2eb96f9b37494be1824999d58028a305",
+            "preview": "https://p.scdn.co/mp3-preview/98deb9c370bbaa350be058b3470fbe3bc1e28d9d",
             "popularity": 77,
             "last_refresh": null
           },

tests/components/music_assistant/snapshots/test_media_player.ambr

@@ -126,7 +126,7 @@
       'media_position': 232,
       'media_position_updated_at': datetime.datetime(2024, 10, 30, 18, 31, 49, 565951, tzinfo=datetime.timezone.utc),
       'media_title': 'November Rain',
-      'repeat': 'all',
+      'repeat': <RepeatMode.ALL: 'all'>,
       'shuffle': True,
       'supported_features': <MediaPlayerEntityFeature: 8320959>,
       'volume_level': 0.06,