Use DeviceClass Enums in brother tests (#62110)
This commit is contained in:
parent
0dbd948867
commit
84dad5d678
1 changed files with 19 additions and 19 deletions
|
@ -8,14 +8,14 @@ from homeassistant.components.brother.sensor import UNIT_PAGES
|
|||
from homeassistant.components.sensor import (
|
||||
ATTR_STATE_CLASS,
|
||||
DOMAIN as SENSOR_DOMAIN,
|
||||
STATE_CLASS_MEASUREMENT,
|
||||
SensorDeviceClass,
|
||||
SensorStateClass,
|
||||
)
|
||||
from homeassistant.const import (
|
||||
ATTR_DEVICE_CLASS,
|
||||
ATTR_ENTITY_ID,
|
||||
ATTR_ICON,
|
||||
ATTR_UNIT_OF_MEASUREMENT,
|
||||
DEVICE_CLASS_TIMESTAMP,
|
||||
PERCENTAGE,
|
||||
STATE_UNAVAILABLE,
|
||||
)
|
||||
|
@ -67,7 +67,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "75"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_black_toner_remaining")
|
||||
assert entry
|
||||
|
@ -78,7 +78,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "10"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_cyan_toner_remaining")
|
||||
assert entry
|
||||
|
@ -89,7 +89,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "8"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_magenta_toner_remaining")
|
||||
assert entry
|
||||
|
@ -100,7 +100,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "2"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_yellow_toner_remaining")
|
||||
assert entry
|
||||
|
@ -113,7 +113,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_COUNTER) == 986
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "92"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_drum_remaining_life")
|
||||
assert entry
|
||||
|
@ -126,7 +126,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_COUNTER) == 1611
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "92"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_black_drum_remaining_life")
|
||||
assert entry
|
||||
|
@ -139,7 +139,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_COUNTER) == 1611
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "92"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_cyan_drum_remaining_life")
|
||||
assert entry
|
||||
|
@ -152,7 +152,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_COUNTER) == 1611
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "92"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_magenta_drum_remaining_life")
|
||||
assert entry
|
||||
|
@ -165,7 +165,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_COUNTER) == 1611
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "92"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_yellow_drum_remaining_life")
|
||||
assert entry
|
||||
|
@ -176,7 +176,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:water-outline"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "97"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_fuser_remaining_life")
|
||||
assert entry
|
||||
|
@ -187,7 +187,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:current-ac"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "97"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_belt_unit_remaining_life")
|
||||
assert entry
|
||||
|
@ -198,7 +198,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
||||
assert state.state == "98"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_pf_kit_1_remaining_life")
|
||||
assert entry
|
||||
|
@ -209,7 +209,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
|
||||
assert state.state == "986"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_page_counter")
|
||||
assert entry
|
||||
|
@ -220,7 +220,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
|
||||
assert state.state == "538"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_duplex_unit_pages_counter")
|
||||
assert entry
|
||||
|
@ -231,7 +231,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
|
||||
assert state.state == "709"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_b_w_counter")
|
||||
assert entry
|
||||
|
@ -242,7 +242,7 @@ async def test_sensors(hass):
|
|||
assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
|
||||
assert state.state == "902"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
||||
|
||||
entry = registry.async_get("sensor.hl_l2340dw_color_counter")
|
||||
assert entry
|
||||
|
@ -252,7 +252,7 @@ async def test_sensors(hass):
|
|||
assert state
|
||||
assert state.attributes.get(ATTR_ICON) is None
|
||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None
|
||||
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_TIMESTAMP
|
||||
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TIMESTAMP
|
||||
assert state.state == "2019-09-24T12:14:56+00:00"
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue