From f02e9eb70f9ed8babaf2485baf35773559572ee1 Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Thu, 16 Dec 2021 22:34:07 +0000
Subject: [PATCH] Use DeviceClass Enums in directv tests (#62118)

---
 tests/components/directv/test_media_player.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/components/directv/test_media_player.py b/tests/components/directv/test_media_player.py
index 41431748d36..cfade8bc4e7 100644
--- a/tests/components/directv/test_media_player.py
+++ b/tests/components/directv/test_media_player.py
@@ -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"