Use DeviceClass Enums in directv tests (#62118)

This commit is contained in:
Dave T 2021-12-16 22:34:07 +00:00 committed by GitHub
parent 61f2f9d9ac
commit f02e9eb70f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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