Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-27 14:31:13 +00:00
Code Activity

Use DeviceClass Enums in directv tests (#62118)

Browse Source
This commit is contained in:
Dave T
2021-12-16 22:34:07 +00:00
committed by GitHub
parent 61f2f9d9ac
commit f02e9eb70f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/components/directv/test_media_player.py
View File

@@ -12,7 +12,7 @@ from homeassistant.components.directv.media_player import (
ATTR_MEDIA_RECORDED,
ATTR_MEDIA_START_TIME,
)
from homeassistant.components.media_player import DEVICE_CLASS_RECEIVER
from homeassistant.components.media_player import MediaPlayerDeviceClass
from homeassistant.components.media_player.const import (
ATTR_INPUT_SOURCE,
ATTR_MEDIA_ALBUM_NAME,
@@ -160,15 +160,15 @@ async def test_unique_id(
entity_registry = er.async_get(hass)
main = entity_registry.async_get(MAIN_ENTITY_ID)
assert main.original_device_class == DEVICE_CLASS_RECEIVER
assert main.original_device_class == MediaPlayerDeviceClass.RECEIVER
assert main.unique_id == "028877455858"
client = entity_registry.async_get(CLIENT_ENTITY_ID)
assert client.original_device_class == DEVICE_CLASS_RECEIVER
assert client.original_device_class == MediaPlayerDeviceClass.RECEIVER
assert client.unique_id == "2CA17D1CD30X"
unavailable_client = entity_registry.async_get(UNAVAILABLE_ENTITY_ID)
assert unavailable_client.original_device_class == DEVICE_CLASS_RECEIVER
assert unavailable_client.original_device_class == MediaPlayerDeviceClass.RECEIVER
assert unavailable_client.unique_id == "9XXXXXXXXXX9"