mirror of
https://github.com/home-assistant/core.git
synced 2025-07-11 23:37:18 +00:00
Use enums in roku tests (#62196)
This commit is contained in:
parent
d874eb261d
commit
5e0eb0eb4d
@ -4,7 +4,7 @@ from unittest.mock import patch
|
|||||||
|
|
||||||
from rokuecp import RokuError
|
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 (
|
from homeassistant.components.media_player.const import (
|
||||||
ATTR_APP_ID,
|
ATTR_APP_ID,
|
||||||
ATTR_APP_NAME,
|
ATTR_APP_NAME,
|
||||||
@ -89,7 +89,7 @@ async def test_setup(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -
|
|||||||
|
|
||||||
assert hass.states.get(MAIN_ENTITY_ID)
|
assert hass.states.get(MAIN_ENTITY_ID)
|
||||||
assert main
|
assert main
|
||||||
assert main.original_device_class == DEVICE_CLASS_RECEIVER
|
assert main.original_device_class is MediaPlayerDeviceClass.RECEIVER
|
||||||
assert main.unique_id == UPNP_SERIAL
|
assert main.unique_id == UPNP_SERIAL
|
||||||
|
|
||||||
|
|
||||||
@ -121,7 +121,7 @@ async def test_tv_setup(
|
|||||||
|
|
||||||
assert hass.states.get(TV_ENTITY_ID)
|
assert hass.states.get(TV_ENTITY_ID)
|
||||||
assert tv
|
assert tv
|
||||||
assert tv.original_device_class == DEVICE_CLASS_TV
|
assert tv.original_device_class is MediaPlayerDeviceClass.TV
|
||||||
assert tv.unique_id == TV_SERIAL
|
assert tv.unique_id == TV_SERIAL
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user