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

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats 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"