jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 15:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve Roku media player test coverage (#34403)

Browse Source
This commit is contained in:
Chris Talkington 2020-04-19 09:59:12 -05:00 committed by GitHub
parent 4e095e9fff
commit 75608ce378
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests/components/roku/test_media_player.py
View File

@ -1,5 +1,5 @@
"""Tests for the Roku Media Player platform.""" """Tests for the Roku Media Player platform."""
from asynctest import patch from asynctest import PropertyMock, patch
from requests_mock import Mocker from requests_mock import Mocker
from homeassistant.components.media_player.const import ( from homeassistant.components.media_player.const import (
@ -25,6 +25,7 @@ from homeassistant.components.media_player.const import (
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
SERVICE_MEDIA_NEXT_TRACK, SERVICE_MEDIA_NEXT_TRACK,
SERVICE_MEDIA_PLAY_PAUSE,
SERVICE_MEDIA_PREVIOUS_TRACK, SERVICE_MEDIA_PREVIOUS_TRACK,
SERVICE_TURN_OFF, SERVICE_TURN_OFF,
SERVICE_TURN_ON, SERVICE_TURN_ON,
@ -32,6 +33,7 @@ from homeassistant.const import (
SERVICE_VOLUME_MUTE, SERVICE_VOLUME_MUTE,
SERVICE_VOLUME_UP, SERVICE_VOLUME_UP,
STATE_PLAYING, STATE_PLAYING,
STATE_STANDBY,
) )
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.helpers.typing import HomeAssistantType
@ -47,6 +49,28 @@ TV_SERIAL = "YN00H5555555"
async def test_setup(hass: HomeAssistantType, requests_mock: Mocker) -> None: async def test_setup(hass: HomeAssistantType, requests_mock: Mocker) -> None:
"""Test setup with basic config.""" """Test setup with basic config."""
await setup_integration(hass, requests_mock) await setup_integration(hass, requests_mock)
entity_registry = await hass.helpers.entity_registry.async_get_registry()
main = entity_registry.async_get(MAIN_ENTITY_ID)
assert hass.states.get(MAIN_ENTITY_ID)
assert main.unique_id == UPNP_SERIAL
async def test_idle_setup(hass: HomeAssistantType, requests_mock: Mocker) -> None:
"""Test setup with idle device."""
with patch(
"homeassistant.components.roku.Roku.power_state",
new_callable=PropertyMock(return_value="Off"),
):
await setup_integration(hass, requests_mock)
state = hass.states.get(MAIN_ENTITY_ID)
assert state.state == STATE_STANDBY
async def test_tv_setup(hass: HomeAssistantType, requests_mock: Mocker) -> None:
"""Test Roku TV setup."""
await setup_integration( await setup_integration(
hass, hass,
requests_mock, requests_mock,
@ -58,10 +82,6 @@ async def test_setup(hass: HomeAssistantType, requests_mock: Mocker) -> None:
entity_registry = await hass.helpers.entity_registry.async_get_registry() entity_registry = await hass.helpers.entity_registry.async_get_registry()
main = entity_registry.async_get(MAIN_ENTITY_ID)
assert hass.states.get(MAIN_ENTITY_ID)
assert main.unique_id == UPNP_SERIAL
tv = entity_registry.async_get(TV_ENTITY_ID) tv = entity_registry.async_get(TV_ENTITY_ID)
assert hass.states.get(TV_ENTITY_ID) assert hass.states.get(TV_ENTITY_ID)
assert tv.unique_id == TV_SERIAL assert tv.unique_id == TV_SERIAL
@ -164,6 +184,16 @@ async def test_services(hass: HomeAssistantType, requests_mock: Mocker) -> None:
remote_mock.assert_called_once_with("/keypress/PowerOn") remote_mock.assert_called_once_with("/keypress/PowerOn")
with patch("roku.Roku._post") as remote_mock:
await hass.services.async_call(
MP_DOMAIN,
SERVICE_MEDIA_PLAY_PAUSE,
{ATTR_ENTITY_ID: MAIN_ENTITY_ID},
blocking=True,
)
remote_mock.assert_called_once_with("/keypress/Play")
with patch("roku.Roku._post") as remote_mock: with patch("roku.Roku._post") as remote_mock:
await hass.services.async_call( await hass.services.async_call(
MP_DOMAIN, MP_DOMAIN,