jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use new enums in huisbaasje tests (#62714)

Browse Source
This commit is contained in:
Dave T 2021-12-23 22:44:09 +00:00 committed by GitHub
parent bbebf311b1
commit 759481688b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
tests/components/huisbaasje/test_sensor.py
View File

@ -3,7 +3,11 @@ from unittest.mock import patch
from homeassistant.components import huisbaasje from homeassistant.components import huisbaasje
from homeassistant.components.huisbaasje.const import FLOW_CUBIC_METERS_PER_HOUR 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 ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ATTR_ICON, ATTR_ICON,
@ -11,9 +15,6 @@ from homeassistant.const import (
CONF_ID, CONF_ID,
CONF_PASSWORD, CONF_PASSWORD,
CONF_USERNAME, CONF_USERNAME,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_GAS,
DEVICE_CLASS_POWER,
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
POWER_WATT, POWER_WATT,
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
@ -58,7 +59,9 @@ async def test_setup_entry(hass: HomeAssistant):
# Assert data is loaded # Assert data is loaded
current_power = hass.states.get("sensor.huisbaasje_current_power") current_power = hass.states.get("sensor.huisbaasje_current_power")
assert current_power.state == "1012.0" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
current_power.attributes.get(ATTR_STATE_CLASS) 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") current_power_in = hass.states.get("sensor.huisbaasje_current_power_in_peak")
assert current_power_in.state == "1012.0" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
current_power_in.attributes.get(ATTR_STATE_CLASS) 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.state == "unknown"
assert ( 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 current_power_in_low.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
assert ( assert (
@ -94,7 +101,10 @@ async def test_setup_entry(hass: HomeAssistant):
current_power_out = hass.states.get("sensor.huisbaasje_current_power_out_peak") current_power_out = hass.states.get("sensor.huisbaasje_current_power_out_peak")
assert current_power_out.state == "unknown" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
current_power_out.attributes.get(ATTR_STATE_CLASS) 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.state == "unknown"
assert ( assert (
current_power_out_low.attributes.get(ATTR_DEVICE_CLASS) 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 current_power_out_low.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
assert ( 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.state == "2.67"
assert ( assert (
energy_consumption_peak_today.attributes.get(ATTR_DEVICE_CLASS) energy_consumption_peak_today.attributes.get(ATTR_DEVICE_CLASS)
== DEVICE_CLASS_ENERGY == SensorDeviceClass.ENERGY
) )
assert ( assert (
energy_consumption_peak_today.attributes.get(ATTR_ICON) 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.state == "0.627"
assert ( assert (
energy_consumption_off_peak_today.attributes.get(ATTR_DEVICE_CLASS) energy_consumption_off_peak_today.attributes.get(ATTR_DEVICE_CLASS)
== DEVICE_CLASS_ENERGY == SensorDeviceClass.ENERGY
) )
assert ( assert (
energy_consumption_off_peak_today.attributes.get(ATTR_ICON) 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.state == "1.512"
assert ( assert (
energy_production_peak_today.attributes.get(ATTR_DEVICE_CLASS) energy_production_peak_today.attributes.get(ATTR_DEVICE_CLASS)
== DEVICE_CLASS_ENERGY == SensorDeviceClass.ENERGY
) )
assert ( assert (
energy_production_peak_today.attributes.get(ATTR_ICON) 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.state == "1.093"
assert ( assert (
energy_production_off_peak_today.attributes.get(ATTR_DEVICE_CLASS) energy_production_off_peak_today.attributes.get(ATTR_DEVICE_CLASS)
== DEVICE_CLASS_ENERGY == SensorDeviceClass.ENERGY
) )
assert ( assert (
energy_production_off_peak_today.attributes.get(ATTR_ICON) 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") energy_today = hass.states.get("sensor.huisbaasje_energy_today")
assert energy_today.state == "3.3" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
energy_today.attributes.get(ATTR_STATE_CLASS) 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") energy_this_week = hass.states.get("sensor.huisbaasje_energy_this_week")
assert energy_this_week.state == "17.5" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
energy_this_week.attributes.get(ATTR_STATE_CLASS) 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") energy_this_month = hass.states.get("sensor.huisbaasje_energy_this_month")
assert energy_this_month.state == "103.3" assert energy_this_month.state == "103.3"
assert ( 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 energy_this_month.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
assert ( assert (
@ -246,7 +262,10 @@ async def test_setup_entry(hass: HomeAssistant):
energy_this_year = hass.states.get("sensor.huisbaasje_energy_this_year") energy_this_year = hass.states.get("sensor.huisbaasje_energy_this_year")
assert energy_this_year.state == "673.0" 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_ICON) == "mdi:lightning-bolt"
assert ( assert (
energy_this_year.attributes.get(ATTR_STATE_CLASS) 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") gas_today = hass.states.get("sensor.huisbaasje_gas_today")
assert gas_today.state == "1.1" 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_ICON) == "mdi:counter"
assert ( assert (
gas_today.attributes.get(ATTR_STATE_CLASS) 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") gas_this_week = hass.states.get("sensor.huisbaasje_gas_this_week")
assert gas_this_week.state == "5.6" 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_ICON) == "mdi:counter"
assert ( assert (
gas_this_week.attributes.get(ATTR_STATE_CLASS) 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") gas_this_month = hass.states.get("sensor.huisbaasje_gas_this_month")
assert gas_this_month.state == "39.1" 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_ICON) == "mdi:counter"
assert ( assert (
gas_this_month.attributes.get(ATTR_STATE_CLASS) 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") gas_this_year = hass.states.get("sensor.huisbaasje_gas_this_year")
assert gas_this_year.state == "116.7" 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_ICON) == "mdi:counter"
assert ( assert (
gas_this_year.attributes.get(ATTR_STATE_CLASS) gas_this_year.attributes.get(ATTR_STATE_CLASS)