Fix roku select source with app ids (#36191)

This commit is contained in:
Chris Talkington 2020-05-27 10:53:26 -05:00 committed by GitHub
parent 6f4829c390
commit 6fbc3b54bd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 3 deletions

View file

@ -222,7 +222,12 @@ class RokuMediaPlayer(RokuEntity, MediaPlayerEntity):
await self.coordinator.roku.remote("home")
appl = next(
(app for app in self.coordinator.data.apps if app.name == source), None
(
app
for app in self.coordinator.data.apps
if source in (app.name, app.app_id)
),
None,
)
if appl is not None:

View file

@ -332,7 +332,7 @@ async def test_services(
remote_mock.assert_called_once_with("home")
with patch("homeassistant.components.roku.Roku.launch") as remote_mock:
with patch("homeassistant.components.roku.Roku.launch") as launch_mock:
await hass.services.async_call(
MP_DOMAIN,
SERVICE_SELECT_SOURCE,
@ -340,7 +340,17 @@ async def test_services(
blocking=True,
)
remote_mock.assert_called_once_with("12")
launch_mock.assert_called_once_with("12")
with patch("homeassistant.components.roku.Roku.launch") as launch_mock:
await hass.services.async_call(
MP_DOMAIN,
SERVICE_SELECT_SOURCE,
{ATTR_ENTITY_ID: MAIN_ENTITY_ID, ATTR_INPUT_SOURCE: 12},
blocking=True,
)
launch_mock.assert_called_once_with("12")
async def test_tv_services(