From 759481688bce80ebb4bfb5c3ba0185e9f5c0cfb6 Mon Sep 17 00:00:00 2001 From: Dave T <17680170+davet2001@users.noreply.github.com> Date: Thu, 23 Dec 2021 22:44:09 +0000 Subject: [PATCH] Use new enums in huisbaasje tests (#62714) --- tests/components/huisbaasje/test_sensor.py | 61 ++++++++++++++-------- 1 file changed, 40 insertions(+), 21 deletions(-) diff --git a/tests/components/huisbaasje/test_sensor.py b/tests/components/huisbaasje/test_sensor.py index 171c6bd25c0..4418127a0d1 100644 --- a/tests/components/huisbaasje/test_sensor.py +++ b/tests/components/huisbaasje/test_sensor.py @@ -3,7 +3,11 @@ from unittest.mock import patch from homeassistant.components import huisbaasje from homeassistant.components.huisbaasje.const import FLOW_CUBIC_METERS_PER_HOUR -from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass +from homeassistant.components.sensor import ( + ATTR_STATE_CLASS, + SensorDeviceClass, + SensorStateClass, +) from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_ICON, @@ -11,9 +15,6 @@ from homeassistant.const import ( CONF_ID, CONF_PASSWORD, CONF_USERNAME, - DEVICE_CLASS_ENERGY, - DEVICE_CLASS_GAS, - DEVICE_CLASS_POWER, ENERGY_KILO_WATT_HOUR, POWER_WATT, VOLUME_CUBIC_METERS, @@ -58,7 +59,9 @@ async def test_setup_entry(hass: HomeAssistant): # Assert data is loaded current_power = hass.states.get("sensor.huisbaasje_current_power") assert current_power.state == "1012.0" - assert current_power.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER + assert ( + current_power.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER + ) assert current_power.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( current_power.attributes.get(ATTR_STATE_CLASS) @@ -68,7 +71,10 @@ async def test_setup_entry(hass: HomeAssistant): current_power_in = hass.states.get("sensor.huisbaasje_current_power_in_peak") assert current_power_in.state == "1012.0" - assert current_power_in.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER + assert ( + current_power_in.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.POWER + ) assert current_power_in.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( current_power_in.attributes.get(ATTR_STATE_CLASS) @@ -81,7 +87,8 @@ async def test_setup_entry(hass: HomeAssistant): ) assert current_power_in_low.state == "unknown" assert ( - current_power_in_low.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER + current_power_in_low.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.POWER ) assert current_power_in_low.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( @@ -94,7 +101,10 @@ async def test_setup_entry(hass: HomeAssistant): current_power_out = hass.states.get("sensor.huisbaasje_current_power_out_peak") assert current_power_out.state == "unknown" - assert current_power_out.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER + assert ( + current_power_out.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.POWER + ) assert current_power_out.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( current_power_out.attributes.get(ATTR_STATE_CLASS) @@ -108,7 +118,7 @@ async def test_setup_entry(hass: HomeAssistant): assert current_power_out_low.state == "unknown" assert ( current_power_out_low.attributes.get(ATTR_DEVICE_CLASS) - == DEVICE_CLASS_POWER + == SensorDeviceClass.POWER ) assert current_power_out_low.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( @@ -125,7 +135,7 @@ async def test_setup_entry(hass: HomeAssistant): assert energy_consumption_peak_today.state == "2.67" assert ( energy_consumption_peak_today.attributes.get(ATTR_DEVICE_CLASS) - == DEVICE_CLASS_ENERGY + == SensorDeviceClass.ENERGY ) assert ( energy_consumption_peak_today.attributes.get(ATTR_ICON) @@ -146,7 +156,7 @@ async def test_setup_entry(hass: HomeAssistant): assert energy_consumption_off_peak_today.state == "0.627" assert ( energy_consumption_off_peak_today.attributes.get(ATTR_DEVICE_CLASS) - == DEVICE_CLASS_ENERGY + == SensorDeviceClass.ENERGY ) assert ( energy_consumption_off_peak_today.attributes.get(ATTR_ICON) @@ -167,7 +177,7 @@ async def test_setup_entry(hass: HomeAssistant): assert energy_production_peak_today.state == "1.512" assert ( energy_production_peak_today.attributes.get(ATTR_DEVICE_CLASS) - == DEVICE_CLASS_ENERGY + == SensorDeviceClass.ENERGY ) assert ( energy_production_peak_today.attributes.get(ATTR_ICON) @@ -188,7 +198,7 @@ async def test_setup_entry(hass: HomeAssistant): assert energy_production_off_peak_today.state == "1.093" assert ( energy_production_off_peak_today.attributes.get(ATTR_DEVICE_CLASS) - == DEVICE_CLASS_ENERGY + == SensorDeviceClass.ENERGY ) assert ( energy_production_off_peak_today.attributes.get(ATTR_ICON) @@ -205,7 +215,9 @@ async def test_setup_entry(hass: HomeAssistant): energy_today = hass.states.get("sensor.huisbaasje_energy_today") assert energy_today.state == "3.3" - assert energy_today.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY + assert ( + energy_today.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY + ) assert energy_today.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( energy_today.attributes.get(ATTR_STATE_CLASS) @@ -218,7 +230,10 @@ async def test_setup_entry(hass: HomeAssistant): energy_this_week = hass.states.get("sensor.huisbaasje_energy_this_week") assert energy_this_week.state == "17.5" - assert energy_this_week.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY + assert ( + energy_this_week.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.ENERGY + ) assert energy_this_week.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( energy_this_week.attributes.get(ATTR_STATE_CLASS) @@ -232,7 +247,8 @@ async def test_setup_entry(hass: HomeAssistant): energy_this_month = hass.states.get("sensor.huisbaasje_energy_this_month") assert energy_this_month.state == "103.3" assert ( - energy_this_month.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY + energy_this_month.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.ENERGY ) assert energy_this_month.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( @@ -246,7 +262,10 @@ async def test_setup_entry(hass: HomeAssistant): energy_this_year = hass.states.get("sensor.huisbaasje_energy_this_year") assert energy_this_year.state == "673.0" - assert energy_this_year.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY + assert ( + energy_this_year.attributes.get(ATTR_DEVICE_CLASS) + == SensorDeviceClass.ENERGY + ) assert energy_this_year.attributes.get(ATTR_ICON) == "mdi:lightning-bolt" assert ( energy_this_year.attributes.get(ATTR_STATE_CLASS) @@ -271,7 +290,7 @@ async def test_setup_entry(hass: HomeAssistant): gas_today = hass.states.get("sensor.huisbaasje_gas_today") assert gas_today.state == "1.1" - assert gas_today.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_GAS + assert gas_today.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.GAS assert gas_today.attributes.get(ATTR_ICON) == "mdi:counter" assert ( gas_today.attributes.get(ATTR_STATE_CLASS) @@ -281,7 +300,7 @@ async def test_setup_entry(hass: HomeAssistant): gas_this_week = hass.states.get("sensor.huisbaasje_gas_this_week") assert gas_this_week.state == "5.6" - assert gas_this_week.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_GAS + assert gas_this_week.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.GAS assert gas_this_week.attributes.get(ATTR_ICON) == "mdi:counter" assert ( gas_this_week.attributes.get(ATTR_STATE_CLASS) @@ -294,7 +313,7 @@ async def test_setup_entry(hass: HomeAssistant): gas_this_month = hass.states.get("sensor.huisbaasje_gas_this_month") assert gas_this_month.state == "39.1" - assert gas_this_month.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_GAS + assert gas_this_month.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.GAS assert gas_this_month.attributes.get(ATTR_ICON) == "mdi:counter" assert ( gas_this_month.attributes.get(ATTR_STATE_CLASS) @@ -307,7 +326,7 @@ async def test_setup_entry(hass: HomeAssistant): gas_this_year = hass.states.get("sensor.huisbaasje_gas_this_year") assert gas_this_year.state == "116.7" - assert gas_this_year.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_GAS + assert gas_this_year.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.GAS assert gas_this_year.attributes.get(ATTR_ICON) == "mdi:counter" assert ( gas_this_year.attributes.get(ATTR_STATE_CLASS)