From d82e8b6cc0d62b8d3185815b38b2cefbe0d9b155 Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Tue, 21 Dec 2021 21:30:18 +0000
Subject: [PATCH] Use new enums in mobile_app tests (#62517)

---
 tests/components/mobile_app/test_sensor.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/components/mobile_app/test_sensor.py b/tests/components/mobile_app/test_sensor.py
index 295e37ee7d9..7eb99df8d8f 100644
--- a/tests/components/mobile_app/test_sensor.py
+++ b/tests/components/mobile_app/test_sensor.py
@@ -3,7 +3,7 @@ from http import HTTPStatus
 
 import pytest
 
-from homeassistant.components.sensor import DEVICE_CLASS_DATE, DEVICE_CLASS_TIMESTAMP
+from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.const import PERCENTAGE, STATE_UNAVAILABLE, STATE_UNKNOWN
 from homeassistant.helpers import device_registry as dr, entity_registry as er
 
@@ -284,24 +284,24 @@ async def test_update_sensor_no_state(hass, create_registrations, webhook_client
 @pytest.mark.parametrize(
     "device_class,native_value,state_value",
     [
-        (DEVICE_CLASS_DATE, "2021-11-18", "2021-11-18"),
+        (SensorDeviceClass.DATE, "2021-11-18", "2021-11-18"),
         (
-            DEVICE_CLASS_TIMESTAMP,
+            SensorDeviceClass.TIMESTAMP,
             "2021-11-18T20:25:00+00:00",
             "2021-11-18T20:25:00+00:00",
         ),
         (
-            DEVICE_CLASS_TIMESTAMP,
+            SensorDeviceClass.TIMESTAMP,
             "2021-11-18 20:25:00+01:00",
             "2021-11-18T19:25:00+00:00",
         ),
         (
-            DEVICE_CLASS_TIMESTAMP,
+            SensorDeviceClass.TIMESTAMP,
             "unavailable",
             STATE_UNAVAILABLE,
         ),
         (
-            DEVICE_CLASS_TIMESTAMP,
+            SensorDeviceClass.TIMESTAMP,
             "unknown",
             STATE_UNKNOWN,
         ),