jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-12 15:57:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use new enums in p1_monitor tests (#62548)

Browse Source
This commit is contained in:
Dave T 2021-12-22 13:28:22 +00:00 committed by GitHub
parent 2f8e65a9b0
commit eb10ff47df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/components/p1_monitor/test_sensor.py
View File

@ -4,8 +4,8 @@ import pytest
from homeassistant.components.p1_monitor.const import DOMAIN from homeassistant.components.p1_monitor.const import DOMAIN
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
ATTR_STATE_CLASS, ATTR_STATE_CLASS,
STATE_CLASS_MEASUREMENT, SensorDeviceClass,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass,
) )
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
@ -13,10 +13,6 @@ from homeassistant.const import (
ATTR_ICON, ATTR_ICON,
ATTR_UNIT_OF_MEASUREMENT, ATTR_UNIT_OF_MEASUREMENT,
CURRENCY_EURO, CURRENCY_EURO,
DEVICE_CLASS_CURRENT,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
DEVICE_CLASS_VOLTAGE,
ELECTRIC_CURRENT_AMPERE, ELECTRIC_CURRENT_AMPERE,
ELECTRIC_POTENTIAL_VOLT, ELECTRIC_POTENTIAL_VOLT,
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
@ -44,9 +40,9 @@ async def test_smartmeter(
assert entry.unique_id == f"{entry_id}_smartmeter_power_consumption" assert entry.unique_id == f"{entry_id}_smartmeter_power_consumption"
assert state.state == "877" assert state.state == "877"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Power Consumption" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Power Consumption"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
assert ATTR_ICON not in state.attributes assert ATTR_ICON not in state.attributes
state = hass.states.get("sensor.monitor_energy_consumption_high") state = hass.states.get("sensor.monitor_energy_consumption_high")
@ -58,9 +54,9 @@ async def test_smartmeter(
assert ( assert (
state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Consumption - High Tariff" state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Consumption - High Tariff"
) )
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_TOTAL_INCREASING assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.TOTAL_INCREASING
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
assert ATTR_ICON not in state.attributes assert ATTR_ICON not in state.attributes
state = hass.states.get("sensor.monitor_energy_tariff_period") state = hass.states.get("sensor.monitor_energy_tariff_period")
@ -101,9 +97,9 @@ async def test_phases(
assert entry.unique_id == f"{entry_id}_phases_voltage_phase_l1" assert entry.unique_id == f"{entry_id}_phases_voltage_phase_l1"
assert state.state == "233.6" assert state.state == "233.6"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Voltage Phase L1" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Voltage Phase L1"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_POTENTIAL_VOLT assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_POTENTIAL_VOLT
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_VOLTAGE assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.VOLTAGE
assert ATTR_ICON not in state.attributes assert ATTR_ICON not in state.attributes
state = hass.states.get("sensor.monitor_current_phase_l1") state = hass.states.get("sensor.monitor_current_phase_l1")
@ -113,9 +109,9 @@ async def test_phases(
assert entry.unique_id == f"{entry_id}_phases_current_phase_l1" assert entry.unique_id == f"{entry_id}_phases_current_phase_l1"
assert state.state == "1.6" assert state.state == "1.6"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Current Phase L1" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Current Phase L1"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_CURRENT assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.CURRENT
assert ATTR_ICON not in state.attributes assert ATTR_ICON not in state.attributes
state = hass.states.get("sensor.monitor_power_consumed_phase_l1") state = hass.states.get("sensor.monitor_power_consumed_phase_l1")
@ -125,9 +121,9 @@ async def test_phases(
assert entry.unique_id == f"{entry_id}_phases_power_consumed_phase_l1" assert entry.unique_id == f"{entry_id}_phases_power_consumed_phase_l1"
assert state.state == "315" assert state.state == "315"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Power Consumed Phase L1" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Power Consumed Phase L1"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
assert ATTR_ICON not in state.attributes assert ATTR_ICON not in state.attributes
assert entry.device_id assert entry.device_id
@ -157,7 +153,7 @@ async def test_settings(
assert entry.unique_id == f"{entry_id}_settings_energy_consumption_price_low" assert entry.unique_id == f"{entry_id}_settings_energy_consumption_price_low"
assert state.state == "0.20522" assert state.state == "0.20522"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Consumption Price - Low" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Consumption Price - Low"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
== f"{CURRENCY_EURO}/{ENERGY_KILO_WATT_HOUR}" == f"{CURRENCY_EURO}/{ENERGY_KILO_WATT_HOUR}"
@ -170,7 +166,7 @@ async def test_settings(
assert entry.unique_id == f"{entry_id}_settings_energy_production_price_low" assert entry.unique_id == f"{entry_id}_settings_energy_production_price_low"
assert state.state == "0.20522" assert state.state == "0.20522"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Production Price - Low" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Energy Production Price - Low"
assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
== f"{CURRENCY_EURO}/{ENERGY_KILO_WATT_HOUR}" == f"{CURRENCY_EURO}/{ENERGY_KILO_WATT_HOUR}"