evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve integration tests for Cambridge Audio (#125906)

Browse source
This commit is contained in:
Noah Husby 2024-09-13 11:09:33 -04:00 • committed by GitHub
parent d507953c70
commit 2d9c9707e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 105 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
homeassistant/components/cambridge_audio/media_player.py
View file

@ -281,6 +281,6 @@ class CambridgeAudioDevice(CambridgeAudioEntity, MediaPlayerEntity):
async def async_set_repeat(self, repeat: RepeatMode) -> None:
"""Set the repeat mode for the current queue."""
repeat_mode = CambridgeRepeatMode.OFF
if repeat:
if repeat in {RepeatMode.ALL, RepeatMode.ONE}:
repeat_mode = CambridgeRepeatMode.ALL
await self.client.set_repeat(repeat_mode)

105
tests/components/cambridge_audio/test_media_player.py
View file

@ -2,13 +2,21 @@
from unittest.mock import AsyncMock
from aiostreammagic import TransportControl
from aiostreammagic import (
RepeatMode as CambridgeRepeatMode,
ShuffleMode,
TransportControl,
)
from aiostreammagic.models import CallbackType
import pytest
from homeassistant.components.media_player import (
ATTR_MEDIA_REPEAT,
ATTR_MEDIA_SEEK_POSITION,
ATTR_MEDIA_SHUFFLE,
DOMAIN as MP_DOMAIN,
MediaPlayerEntityFeature,
RepeatMode,
)
from homeassistant.const import (
ATTR_ENTITY_ID,
@ -17,9 +25,15 @@ from homeassistant.const import (
SERVICE_MEDIA_PAUSE,
SERVICE_MEDIA_PLAY,
SERVICE_MEDIA_PREVIOUS_TRACK,
SERVICE_MEDIA_SEEK,
SERVICE_REPEAT_SET,
SERVICE_SHUFFLE_SET,
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
STATE_BUFFERING,
STATE_IDLE,
STATE_OFF,
STATE_ON,
STATE_PAUSED,
STATE_PLAYING,
STATE_STANDBY,
@ -124,6 +138,7 @@ async def test_entity_supported_features(
(True, "connecting", STATE_BUFFERING),
(True, "stop", STATE_IDLE),
(True, "ready", STATE_IDLE),
(True, "other", STATE_ON),
],
)
async def test_entity_state(
@ -194,3 +209,91 @@ async def test_media_next_previous_track(
await hass.services.async_call(MP_DOMAIN, SERVICE_MEDIA_PREVIOUS_TRACK, data, True)
mock_stream_magic_client.previous_track.assert_called_once()
async def test_shuffle_repeat(
hass: HomeAssistant,
mock_config_entry: MockConfigEntry,
mock_stream_magic_client: AsyncMock,
) -> None:
"""Test shuffle and repeat service."""
await setup_integration(hass, mock_config_entry)
mock_stream_magic_client.now_playing.controls = [
TransportControl.TOGGLE_SHUFFLE,
TransportControl.TOGGLE_REPEAT,
]
# Test shuffle
await hass.services.async_call(
MP_DOMAIN,
SERVICE_SHUFFLE_SET,
{ATTR_ENTITY_ID: ENTITY_ID, ATTR_MEDIA_SHUFFLE: False},
)
mock_stream_magic_client.set_shuffle.assert_called_with(ShuffleMode.OFF)
await hass.services.async_call(
MP_DOMAIN,
SERVICE_SHUFFLE_SET,
{ATTR_ENTITY_ID: ENTITY_ID, ATTR_MEDIA_SHUFFLE: True},
)
mock_stream_magic_client.set_shuffle.assert_called_with(ShuffleMode.ALL)
# Test repeat
await hass.services.async_call(
MP_DOMAIN,
SERVICE_REPEAT_SET,
{ATTR_ENTITY_ID: ENTITY_ID, ATTR_MEDIA_REPEAT: RepeatMode.OFF},
)
mock_stream_magic_client.set_repeat.assert_called_with(CambridgeRepeatMode.OFF)
await hass.services.async_call(
MP_DOMAIN,
SERVICE_REPEAT_SET,
{ATTR_ENTITY_ID: ENTITY_ID, ATTR_MEDIA_REPEAT: RepeatMode.ALL},
)
mock_stream_magic_client.set_repeat.assert_called_with(CambridgeRepeatMode.ALL)
async def test_power_service(
hass: HomeAssistant,
mock_config_entry: MockConfigEntry,
mock_stream_magic_client: AsyncMock,
) -> None:
"""Test power service."""
await setup_integration(hass, mock_config_entry)
data = {ATTR_ENTITY_ID: ENTITY_ID}
await hass.services.async_call(MP_DOMAIN, SERVICE_TURN_ON, data, True)
mock_stream_magic_client.power_on.assert_called_once()
await hass.services.async_call(MP_DOMAIN, SERVICE_TURN_OFF, data, True)
mock_stream_magic_client.power_off.assert_called_once()
async def test_media_seek(
hass: HomeAssistant,
mock_config_entry: MockConfigEntry,
mock_stream_magic_client: AsyncMock,
) -> None:
"""Test media seek service."""
await setup_integration(hass, mock_config_entry)
mock_stream_magic_client.now_playing.controls = [
TransportControl.SEEK,
]
await hass.services.async_call(
MP_DOMAIN,
SERVICE_MEDIA_SEEK,
{ATTR_ENTITY_ID: ENTITY_ID, ATTR_MEDIA_SEEK_POSITION: 100},
)
mock_stream_magic_client.media_seek.assert_called_once_with(100)