Use new enums in nzbget tests (#62524)

pull/62538/head
Dave T 2021-12-21 20:51:10 +00:00 committed by GitHub
parent cc6228f5c9
commit 8166f37830
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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():