From 043a4b06d0b356f3111a7ea3a19b72a627d66db4 Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Tue, 21 Dec 2021 21:04:32 +0000
Subject: [PATCH] Use new enums in mqtt tests (#62520)

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

diff --git a/tests/components/mqtt/test_sensor.py b/tests/components/mqtt/test_sensor.py
index 680bafb3b2b..106845dbd51 100644
--- a/tests/components/mqtt/test_sensor.py
+++ b/tests/components/mqtt/test_sensor.py
@@ -83,27 +83,27 @@ async def test_setting_sensor_value_via_mqtt_message(hass, mqtt_mock):
 @pytest.mark.parametrize(
     "device_class,native_value,state_value,log",
     [
-        (sensor.DEVICE_CLASS_DATE, "2021-11-18", "2021-11-18", False),
-        (sensor.DEVICE_CLASS_DATE, "invalid", STATE_UNKNOWN, True),
+        (sensor.SensorDeviceClass.DATE, "2021-11-18", "2021-11-18", False),
+        (sensor.SensorDeviceClass.DATE, "invalid", STATE_UNKNOWN, True),
         (
-            sensor.DEVICE_CLASS_TIMESTAMP,
+            sensor.SensorDeviceClass.TIMESTAMP,
             "2021-11-18T20:25:00+00:00",
             "2021-11-18T20:25:00+00:00",
             False,
         ),
         (
-            sensor.DEVICE_CLASS_TIMESTAMP,
+            sensor.SensorDeviceClass.TIMESTAMP,
             "2021-11-18 20:25:00+00:00",
             "2021-11-18T20:25:00+00:00",
             False,
         ),
         (
-            sensor.DEVICE_CLASS_TIMESTAMP,
+            sensor.SensorDeviceClass.TIMESTAMP,
             "2021-11-18 20:25:00+01:00",
             "2021-11-18T19:25:00+00:00",
             False,
         ),
-        (sensor.DEVICE_CLASS_TIMESTAMP, "invalid", STATE_UNKNOWN, True),
+        (sensor.SensorDeviceClass.TIMESTAMP, "invalid", STATE_UNKNOWN, True),
     ],
 )
 async def test_setting_sensor_native_value_handling_via_mqtt_message(