From 8166f37830ff43e4788e64ce09e50a9ddd082d81 Mon Sep 17 00:00:00 2001 From: Dave T <17680170+davet2001@users.noreply.github.com> Date: Tue, 21 Dec 2021 20:51:10 +0000 Subject: [PATCH] Use new enums in nzbget tests (#62524) --- tests/components/nzbget/test_sensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/components/nzbget/test_sensor.py b/tests/components/nzbget/test_sensor.py index d47e67c96c6..290414ab0ff 100644 --- a/tests/components/nzbget/test_sensor.py +++ b/tests/components/nzbget/test_sensor.py @@ -2,11 +2,11 @@ from datetime import timedelta from unittest.mock import patch +from homeassistant.components.sensor import SensorDeviceClass from homeassistant.const import ( ATTR_UNIT_OF_MEASUREMENT, DATA_MEGABYTES, DATA_RATE_MEGABYTES_PER_SECOND, - DEVICE_CLASS_TIMESTAMP, ) from homeassistant.helpers import entity_registry as er from homeassistant.util import dt as dt_util @@ -39,7 +39,7 @@ async def test_sensors(hass, nzbget_api) -> None: "post_processing_jobs": ("PostJobCount", "2", "Jobs", None), "post_processing_paused": ("PostPaused", "False", None, None), "queue_size": ("RemainingSizeMB", "512", DATA_MEGABYTES, None), - "uptime": ("UpTimeSec", uptime.isoformat(), None, DEVICE_CLASS_TIMESTAMP), + "uptime": ("UpTimeSec", uptime.isoformat(), None, SensorDeviceClass.TIMESTAMP), } for (sensor_id, data) in sensors.items():