Use enums in roku tests (#62196)

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

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