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

Use enums in roku tests (#62196)

Browse source
This commit is contained in:
Robert Hillis 2021-12-17 10:25:37 -05:00 • committed by GitHub
parent d874eb261d
commit 5e0eb0eb4d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/components/roku/test_media_player.py
View file

@ -4,7 +4,7 @@ from unittest.mock import patch
from rokuecp import RokuError
from homeassistant.components.media_player import DEVICE_CLASS_RECEIVER, DEVICE_CLASS_TV
from homeassistant.components.media_player import MediaPlayerDeviceClass
from homeassistant.components.media_player.const import (
ATTR_APP_ID,
ATTR_APP_NAME,
@ -89,7 +89,7 @@ async def test_setup(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -
assert hass.states.get(MAIN_ENTITY_ID)
assert main
assert main.original_device_class == DEVICE_CLASS_RECEIVER
assert main.original_device_class is MediaPlayerDeviceClass.RECEIVER
assert main.unique_id == UPNP_SERIAL
@ -121,7 +121,7 @@ async def test_tv_setup(
assert hass.states.get(TV_ENTITY_ID)
assert tv
assert tv.original_device_class == DEVICE_CLASS_TV
assert tv.original_device_class is MediaPlayerDeviceClass.TV
assert tv.unique_id == TV_SERIAL